torch-mlir/lib/Conversion/TorchOnnxToTorch
Vivek Khandelwal 6abc7371c8
[MLIR][TORCH] Fix OnnxToLinalg lowering issue for Squeeze and Unsqueeze op (#2991)
This commit also cleans up the OnnxToTorch lowering for the Squeeze and
Unsqueeze op and adds the support for handling edge cases.

Signed-Off By: Vivek Khandelwal <vivekkhandelwal1424@gmail.com>
2024-04-22 08:52:42 +00:00
..
CMakeLists.txt [ONNX] Support onnx.LSTM (#2969) 2024-04-08 12:23:33 -07:00
DefaultDomainAtoF.cpp [onnx] Support for `onnx.EyeLike` via torch lowering (#2994) 2024-04-15 09:23:26 -07:00
DefaultDomainGtoP.cpp [onnx] Fix `onnx.Not` for non-bool inputs (#3187) 2024-04-19 11:32:24 -07:00
DefaultDomainQtoZ.cpp [MLIR][TORCH] Fix OnnxToLinalg lowering issue for Squeeze and Unsqueeze op (#2991) 2024-04-22 08:52:42 +00:00
OnnxLstmExpander.cpp [ONNX] Support onnx.LSTM (#2969) 2024-04-08 12:23:33 -07:00
PassDetail.h Initial TorchOnnxToTorch conversion pipeline. (#2585) 2023-11-21 21:02:55 -08:00
Passes.cpp Initial TorchOnnxToTorch conversion pipeline. (#2585) 2023-11-21 21:02:55 -08:00
Patterns.cpp Initial TorchOnnxToTorch conversion pipeline. (#2585) 2023-11-21 21:02:55 -08:00
TorchOnnxToTorch.cpp Initial TorchOnnxToTorch conversion pipeline. (#2585) 2023-11-21 21:02:55 -08:00
Utils.cpp [MLIR][TORCH] Fix Onnx.ReduceSum lowering for failing e2e tests (#3095) 2024-04-03 09:57:19 +05:30