torch-mlir/lib/Conversion
Rob Suderman d51e80b648
[onnx] Fix onnx.gather lowering for rank-0 indices (#2973)
We assumed rank was atleast 1 however it can be rank-0, generating an
illegal pair of flatten / unflatten operations. Corrected this.
2024-03-04 08:25:19 -08:00
..
TorchConversionToMLProgram Clang format refresh (#2812) 2024-01-29 12:59:33 -05:00
TorchOnnxToTorch [onnx] Fix onnx.gather lowering for rank-0 indices (#2973) 2024-03-04 08:25:19 -08:00
TorchToArith [torch] Supporting `torch.aten.mul.float` lowering to `arith` (#2833) 2024-02-05 16:23:04 -08:00
TorchToLinalg [torch] Support `aten.view` rank-0 collapse (#2965) 2024-03-01 12:31:07 -08:00
TorchToSCF Re-organize project structure to separate PyTorch dependencies from core project. (#2542) 2023-11-02 19:45:55 -07:00
TorchToStablehlo [Stablehlo] add torch_to_stablehlo::getBackendTypeForScalarType (#2975) 2024-03-04 23:31:54 +08:00
TorchToTMTensor [torch] Rework lowering to tm_tensor.scatter to stop serialization (#2940) 2024-02-27 11:46:57 -08:00
TorchToTensor [torch] Add edgecase for aten.shape_to_tensor for rank-0 input (#2962) 2024-02-28 09:47:06 -08:00
TorchToTosa [tosa] Fix TOSA batch matmul lowering to correct transpose ordering (#2959) 2024-02-28 09:46:58 -08:00
Utils [MLIR][TORCH] Fix OnnxToLinalg lowering issue for sub and sum op (#2954) 2024-02-29 21:48:46 +05:30
CMakeLists.txt [onnx] Lowering for `onnx.shape` to `torch` and `tensor` (#2648) 2023-12-15 11:37:49 -08: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