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
jinchen 61e6312c87
Support select_last_index attribute of onnx argmax op (#3192) 
The tests listed in https://github.com/nod-ai/SHARK-Turbine/issues/635
all compiled, but having run issue of dtype mismatch of i/si.
 		2024-04-23 10:16:08 -07:00
..
CMakeLists.txt [ONNX] Support onnx.LSTM (#2969) 2024-04-08 12:23:33 -07:00
DefaultDomainAtoF.cpp Support select_last_index attribute of onnx argmax op (#3192) 2024-04-23 10:16:08 -07:00
DefaultDomainGtoP.cpp [onnx] Fix `onnx.Not` for non-bool inputs (#3187) 2024-04-19 11:32:24 -07:00
DefaultDomainQtoZ.cpp [torch] Add OnnxToTorch lowering for `onnx.ReduceL2` (#3175) 2024-04-23 02:03:05 -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