torch-mlir/lib/Conversion
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
..
TorchConversionToMLProgram [NFC] Remove unused header files (#3386) 2024-05-30 14:30:36 +08:00
TorchOnnxToTorch [onnx] Relax constraints on input tensors in `onnx.STFT` conversion to torch dialect (#3676) 2024-09-23 12:09:29 +05:30
TorchToArith [TorchToArith] Add a lowering for `torch.add.float_int` (#3594) 2024-08-07 11:55:27 -05:00
TorchToLinalg Delete unnecessary linalg conversion for aten.fmod (#3707) 2024-09-13 09:39:58 -07:00
TorchToSCF [NFC] Change to *cast instead of .*cast variants (#3405) 2024-05-30 23:45:13 -07:00
TorchToStablehlo [Stablehlo] fix aten compare ops' promote rules (#3709) 2024-09-13 18:48:41 +08:00
TorchToTMTensor [torch] Fix attention on linalg for dynamic shapes (#3714) 2024-09-18 14:52:54 -05:00
TorchToTensor [NFC] Remove unused header files (#3386) 2024-05-30 14:30:36 +08:00
TorchToTosa [TOSA] Extend Torch to TOSA legalization coverage (#3718) 2024-09-20 14:33:55 -07:00
Utils Added support for integer to complex conversion (#3604) 2024-08-14 18:13:00 +05:30
CMakeLists.txt [torch] Improve shape inference for `torch-to-linalg` path for reshapes (#3055) 2024-03-26 12:41:40 -07:00
PassDetail.h Minor fixes for `ConvertTorchConversionToMLProgram`. (#1991) 2023-04-04 09:09:58 -07:00
Passes.cpp Clang format refresh (#2812) 2024-01-29 12:59:33 -05:00