torch-mlir/lib/Conversion/TorchOnnxToTorch
Phaneesh Barwaria 9a6fe58a02
onnx.MelWeightMatrix Onnx to Torch to Linalg (#3659)
- This PR adds new (and equivalent) more tensorized impl of
MelWeightMatrix which lowers all the way to linalg.
- [Ref Pytorch
Impl](https://gist.github.com/PhaneeshB/4e6dfcded3007b1b686fbe28f07a67cd)
- Thanks to @rsuderman for pointing out the difficulties [earlier
impl](#3503) posed during lowering to linalg and also for providing a
better numpy impl 🙏
2024-08-22 08:55:03 -07:00
..
CMakeLists.txt Support onnx.GRU and onnx.RNN (#3447) 2024-07-10 14:04:17 -04:00
DefaultDomainAtoF.cpp [ONNX] Add training mode support for BatchNormalization op (#3597) 2024-08-14 10:46:38 +05:30
DefaultDomainGtoP.cpp onnx.MelWeightMatrix Onnx to Torch to Linalg (#3659) 2024-08-22 08:55:03 -07:00
DefaultDomainQtoZ.cpp [onnx] Fix onnx.ThresholdedRelu crash (#3638) 2024-08-16 09:23:38 -07:00
OnnxRecurrentLayerOpExpanders.cpp [onnx] Fix shortcircuit path (#3633) 2024-08-16 09:23:47 -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