torch-mlir/lib/Conversion
zjgarvey 7f2a17e757
[ONNX] fix padding for `onnx.MaxPool` (#3611)
The saga of aligning onnx and torch padding conventions continues. 

```python
onnx_pads = [low_x, low_y, low_z, high_x, high_y, high_z]
torch_pads = [low_z, high_z, low_y, high_y, low_x, high_x]
```

So not only is the lexicographical ordering hierarchy swapped (low/high
x spatial-dim -> spatial-dim x low/high) but the ordering in the the
spatial-dim specification is also reversed.

This patch properly reverses the pad ordering (and actually uses the
`shuffledPadding` to pad).
2024-08-07 20:34:00 -07:00
..
TorchConversionToMLProgram [NFC] Remove unused header files (#3386) 2024-05-30 14:30:36 +08:00
TorchOnnxToTorch [ONNX] fix padding for `onnx.MaxPool` (#3611) 2024-08-07 20:34:00 -07:00
TorchToArith [TorchToArith] Add a lowering for `torch.add.float_int` (#3594) 2024-08-07 11:55:27 -05:00
TorchToLinalg [TorchToLinalg] Fix possible OOB access in Interpolate lowering (#3570) 2024-08-02 13:55:37 -05:00
TorchToSCF [NFC] Change to *cast instead of .*cast variants (#3405) 2024-05-30 23:45:13 -07:00
TorchToStablehlo [stablehlo] support dynamic-shaped index in stablehlo conversion for aten.index-like ops (#3322) 2024-08-01 10:41:09 +08:00
TorchToTMTensor [Torch] add support for aten.scatter_add (#3534) 2024-07-12 09:15:42 +08:00
TorchToTensor [NFC] Remove unused header files (#3386) 2024-05-30 14:30:36 +08:00
TorchToTosa [TOSA] Fix Tensor.hacked_twin to support diff size indexes (#3547) 2024-07-30 14:32:05 -07:00
Utils [LINALG] Added support for conversion from float to complex. (#3595) 2024-08-07 12:36:48 +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