torch-mlir/lib/Conversion
Aart Bik 46a25d7241
[torch-mlir][sparse] preserve sparsity during lowering torch to linalg (#2809)
This preserves sparsity at the most obvious places of lowering TORCH
tensors to MLIR RankedTensorType tensors. Other places are marked for
audit. With some initial lowering tests.
2024-01-26 10:54:59 -08:00
..
TorchConversionToMLProgram Re-organize project structure to separate PyTorch dependencies from core project. (#2542) 2023-11-02 19:45:55 -07:00
TorchOnnxToTorch [MLIR][TORCH] Add support for dynamic shape for Onnx.Transpose op (#2803) 2024-01-26 09:46:54 -08:00
TorchToArith Fix for unused variable failure when trying to bump torch-mlir in IREE (#2560) 2023-11-08 15:55:41 -08:00
TorchToLinalg [torch-mlir][sparse] preserve sparsity during lowering torch to linalg (#2809) 2024-01-26 10:54:59 -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]fix CumsumInputDtypeInt32Module_basic on stablehlo backend. (#2797) 2024-01-25 10:44:08 +08:00
TorchToTMTensor Re-organize project structure to separate PyTorch dependencies from core project. (#2542) 2023-11-02 19:45:55 -07:00
TorchToTensor [onnx] Lowering for `onnx.shape` to `torch` and `tensor` (#2648) 2023-12-15 11:37:49 -08:00
TorchToTosa Do not try to legalize transposed convolution (#2721) 2024-01-22 10:57:56 -08:00
Utils [Torch Dialect]Support aten.cosine_similarity (#2364) 2023-11-08 15:28:30 +08:00
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 [onnx] Lowering for `onnx.shape` to `torch` and `tensor` (#2648) 2023-12-15 11:37:49 -08:00