torch-mlir/test/Conversion
zjgarvey 1201babb9f
[ONNX] rework some reduction op lowerings (#3870)
- Refactors more "onnx.ReduceXXX" patterns through helper function.
- Fixes bug with iterating unconditionally on `output_dim == 1` during
`dimList` inference.

This change results in passes for the following 11 models:

crossvit_15_240
crossvit_15_dagger_240
crossvit_15_dagger_408
crossvit_18_240
crossvit_18_dagger_240
crossvit_18_dagger_408
crossvit_9_240
crossvit_9_dagger_240
crossvit_base_240
crossvit_small_240
crossvit_tiny_240

---------

Co-authored-by: Vinayak Dev <104419489+vinayakdsci@users.noreply.github.com>
2024-11-14 16:25:28 +00:00
..
TorchConversionToMLProgram Bump LLVM and StableHLO (#2598) 2023-11-28 22:12:24 -08:00
TorchOnnxToTorch [ONNX] rework some reduction op lowerings (#3870) 2024-11-14 16:25:28 +00:00
TorchToArith [TorchToArith] add lowerings for some scalar bool binary ops (#3823) 2024-11-01 10:40:20 -05:00
TorchToLinalg OnnxToTorch bicubic interpolation (#3802) 2024-11-12 12:54:29 -06:00
TorchToSCF Bump llvm to f9031f00f2c9 (#3672) 2024-08-28 11:29:10 -07:00
TorchToStablehlo Bump llvm/llvm-project@b6603e1bf1 (#3697) 2024-09-10 08:57:15 -07:00
TorchToTensor [onnx] Lowering for `onnx.shape` to `torch` and `tensor` (#2648) 2023-12-15 11:37:49 -08:00
TorchToTosa Support default padding case for tosa::AvgPool in the presence of count_include_pad (#3868) 2024-11-12 13:48:20 -08:00