torch-mlir/projects/pt1/python/torch_mlir
Branko Trifkovic 1c4b9d6a0e
Implement lowering of torch.aten.hstack (#3563)
2024-09-11 16:41:47 +05:30
..
_torch_mlir_custom_op_example [NFC reformat] Applies pre-commit formatting to Python files. (#3244) 2024-04-27 14:16:31 -07:00
csrc [NFC reformat] Applies pre-commit formatting to Python files. (#3244) 2024-04-27 14:16:31 -07:00
jit_ir_importer Implement lowering of torch.aten.hstack (#3563) 2024-09-11 16:41:47 +05:30
_dynamo_fx_importer.py [NFC] Update black version (#3256) 2024-04-29 11:06:01 +08:00
_version.py [NFC reformat] Applies pre-commit formatting to Python files. (#3244) 2024-04-27 14:16:31 -07:00
dynamo.py [FxImporter] Add aten._scaled_dot_product_flash_attention_for_cpu to default decomposition table (#3456) 2024-06-14 10:52:09 +08:00
torchscript.py Decompose aten.fmod into aten.mul,sub,div etc. (#3689) 2024-09-09 09:00:11 -07:00