torch-mlir/lib/Conversion
Yuanqiang Liu 5bb1a65ec9
[Stablehlo] refactor reduction lowering and support aten.amin (#3383)
* implement detailed lowering template pattern
`ConvertAtenReduceAllDimsOp` and `ConvertAtenReduceKeepDimOp`
* support `aten.amin`'s lowering.
2024-05-23 20:40:20 +08:00
..
TorchConversionToMLProgram Clang format refresh (#2812) 2024-01-29 12:59:33 -05:00
TorchOnnxToTorch [Torch] Fix bugs for `Torch::AtenOneHotOp` (#3350) 2024-05-22 17:19:08 +00:00
TorchToArith [Torch] Fix bugs for `Torch::AtenOneHotOp` (#3350) 2024-05-22 17:19:08 +00:00
TorchToLinalg [MLIR] Fix 64-bit product during aten.view lowering (#3378) 2024-05-23 08:59:28 +05:30
TorchToSCF Fix deprecated uses of cast/dyn_cast/dyn_cast_or_null/isa (#3130) 2024-04-11 06:47:35 -07:00
TorchToStablehlo [Stablehlo] refactor reduction lowering and support aten.amin (#3383) 2024-05-23 20:40:20 +08:00
TorchToTMTensor [Torch Dialect] decompose all index_put-like op to aten.index_put.hacked_twin for stricter semantics (#3071) 2024-05-08 22:44:57 +08:00
TorchToTensor Fix deprecated uses of cast/dyn_cast/dyn_cast_or_null/isa (#3243) 2024-04-27 14:00:56 -07:00
TorchToTosa [NFC] Standardize the std::is_same competime expression (#3321) 2024-05-10 17:07:37 +08:00
Utils [linalg] Fix bug for conversion of complex dtype (#3269) 2024-05-01 12:06:53 +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