torch-mlir/lib/Conversion/TorchOnnxToTorch
giacs-epic 99848265c3
[onnx] Relax constraints on input tensors in `onnx.STFT` conversion to torch dialect (#3676)
- When the signal tensor is real, onnx allows its shape to be
`[batch][length]` as well as `[batch][length][1]`.
- Onnx also allows to specify `frame_length` together with `window` (not
empty), given that it matches the window size.
- Adding checks on signal and result shapes.
2024-09-23 12:09:29 +05:30
..
CMakeLists.txt Support onnx.GRU and onnx.RNN (#3447) 2024-07-10 14:04:17 -04:00
DefaultDomainAtoF.cpp [ONNX] Fix issue with absent value in onnx.ConstantOfShape (#3713) 2024-09-17 16:01:01 -05:00
DefaultDomainGtoP.cpp Fix onnx.Gather lowering with dynamic shapes (#3675) 2024-08-29 17:02:16 -07:00
DefaultDomainQtoZ.cpp [onnx] Relax constraints on input tensors in `onnx.STFT` conversion to torch dialect (#3676) 2024-09-23 12:09:29 +05:30
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