torch-mlir/lib/Dialect/Torch
Vivek Khandelwal fa4794dae2
[MLIR][TORCH] Add torch-onnx-to-torch-backend pipeline (#3801)
This commit adds the torch-onnx-to-torch-backend pipeline which
converts the Torch Onnx IR to Torch Backend IR.

This commit also moves the `ScalarizeShapes` pass from the
`torch-backend-to-linalg-on-tensors-backend-pipeline` to the
`torch-onnx-to-torch-backend` pipeline since the primary goal of
this pass is to scalarize the shapes in the IR coming from the
Onnx models.
2024-10-21 11:20:44 -05:00
..
IR [Torch] add fold logic for some ops (#3794) 2024-10-16 16:00:58 +08:00
Transforms [MLIR][TORCH] Add torch-onnx-to-torch-backend pipeline (#3801) 2024-10-21 11:20:44 -05:00
Utils Add Op for `torch.aten.unfold` (#3772) 2024-10-08 21:10:43 +00:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00