torch-mlir/python/torch_mlir
penguin_wwy a5d3b546f8
[FxImporter] Fix embedding bag (#3387)
2024-05-29 14:46:21 +08:00
..
_mlir_libs [ci] Upgrade to new runners and disable unsupported jobs. (#2818) 2024-01-27 18:35:45 -08:00
dialects Upstream the ONNX importer. (#2636) 2023-12-12 19:02:51 -08:00
extras [FxImporter] Fix embedding bag (#3387) 2024-05-29 14:46:21 +08:00
tools/import_onnx [ONNX] Enables data propogation for onnx shape inference (#3280) 2024-05-08 09:29:23 -07:00
compiler_utils.py [NFC reformat] Applies pre-commit formatting to Python files. (#3244) 2024-04-27 14:16:31 -07:00
fx.py [Pipeline] Use dedicated simplification pipeline for TorchDynamo frontend (#3376) 2024-05-22 05:23:18 -07:00