torch-mlir/lib/Conversion/TorchOnnxToTorch
Xida Ren (Cedar) 33eef15e42
Support onnx.If (#2825)
This is probably a decent PR for learning about blocks and regions.

If you're here to learn about that, consider also looking at
lib/Conversion/TorchToSCF/TorchToSCF.cpp

While this doesn't include an e2e test, it is tested downstream in
https://github.com/nod-ai/SHARK-TestSuite/blob/main/e2eshark/onnx/operators/If/model.py

---------

Co-authored-by: Xida Ren <xida.ren.dev@gmail.com>
2024-04-30 18:36:40 +00:00
..
CMakeLists.txt [ONNX] Support onnx.LSTM (#2969) 2024-04-08 12:23:33 -07:00
DefaultDomainAtoF.cpp [MLIR][TORCH] Add OnnxToTorch support for BlackmanWindow function (#3181) 2024-04-30 12:21:27 -04:00
DefaultDomainGtoP.cpp Support onnx.If (#2825) 2024-04-30 18:36:40 +00:00
DefaultDomainQtoZ.cpp [MLIR][TORCH] Add OnnxToTorch support for BlackmanWindow function (#3181) 2024-04-30 12:21:27 -04: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 [onnx] Add `onnx-to-torch` lowering for random ops (#3193) 2024-04-22 22:28:07 +05:30