torch-mlir/projects/pt1/python/torch_mlir
Yuanqiang Liu 8814d0ae64
[Torch] emit aten.dot and canonicalize it to aten.matmul (#3361)
* canonicalize `aten.dot` to `aten.matmul`
2024-05-18 22:45:14 +08:00
..
_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 [Torch] emit aten.dot and canonicalize it to aten.matmul (#3361) 2024-05-18 22:45:14 +08:00
_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 [NFC reformat] Applies pre-commit formatting to Python files. (#3244) 2024-04-27 14:16:31 -07:00
torchscript.py [Stablehlo] refactor amax, max, max.dim's lowering to stablehlo (#3348) 2024-05-16 00:05:19 +08:00