llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/lib/Conversion/TorchOnnxToTorch
History
Gaurav Shukla 129a79417a
[MLIR][ONNX] Fix onnx.gather_nd implementation (#3070) 
The indices should be expanded before the torch.gather operation.

Signed-off-by: Gaurav Shukla <gaurav@amd.com>
 		2024-04-01 20:17:09 +05:30
..
CMakeLists.txt [MLIR][ONNX] Add OnnxToTorch support for Maxpool Op (#2695) 2024-01-12 14:54:38 -08:00
DefaultDomainAtoF.cpp [ONNX] onnx.DynamicQuantizeLinear to Torch (#3009) 2024-03-20 10:58:25 -07:00
DefaultDomainGtoP.cpp [MLIR][ONNX] Fix onnx.gather_nd implementation (#3070) 2024-04-01 20:17:09 +05:30
DefaultDomainQtoZ.cpp [torch] Improve shape inference for `torch-to-linalg` path for reshapes (#3055) 2024-03-26 12:41:40 -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] Convert `onnx.QLinearConv` to `torch` (#2851) 2024-02-05 16:09:41 -08:00