torch-mlir/lib/Conversion/TorchOnnxToTorch
zjgarvey c19fc9ba47
[ONNX] Fixes Issue with Dynamic Dims in GlobalAveragePool -> Torch Conversion (#3053)
Two e2e tests (AdaptiveAveragePool1/2dUnitOutputSizeDynamic) were
failing due to numerics. This was as a result of passing -1 as the
kernel size in the lowering for the corresponding onnx op
GlobalAveragePool.
2024-03-28 09:43:09 -07:00
..
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 [ONNX] Fixes Issue with Dynamic Dims in GlobalAveragePool -> Torch Conversion (#3053) 2024-03-28 09:43:09 -07:00
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