torch-mlir/lib/Conversion/TorchOnnxToTorch
jinchen c9ed993603
Support NMS op lowering (#3871)
TODO: support multiple batches and classes
2024-11-26 16:49:56 -08:00
..
CMakeLists.txt Support onnx.GRU and onnx.RNN (#3447) 2024-07-10 14:04:17 -04:00
DefaultDomainAtoF.cpp [ONNX] Remove kernel shape and weight shape equivalence check from Onnx.Conv lowering (#3869) 2024-11-15 10:36:41 +05:30
DefaultDomainGtoP.cpp Support NMS op lowering (#3871) 2024-11-26 16:49:56 -08:00
DefaultDomainQtoZ.cpp [ONNX] rework some reduction op lowerings (#3870) 2024-11-14 16:25:28 +00:00
OnnxRecurrentLayerOpExpanders.cpp onnx.LSTM - bidirectional, layout attr (#3771) 2024-10-08 11:29:49 -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 Support onnx.GRU and onnx.RNN (#3447) 2024-07-10 14:04:17 -04:00