torch-mlir/lib/Conversion
Vivek Khandelwal 1d6e4c3d77
[MLIR][TORCH] Add OnnxToTorch lowering for Einsum op (#3117)
Signed-Off By: Vivek Khandelwal <vivekkhandelwal1424@gmail.com>
2024-04-08 22:38:01 +05:30
..
TorchConversionToMLProgram Clang format refresh (#2812) 2024-01-29 12:59:33 -05:00
TorchOnnxToTorch [MLIR][TORCH] Add OnnxToTorch lowering for Einsum op (#3117) 2024-04-08 22:38:01 +05:30
TorchToArith [MLIR][TORCH]Add support lowing aten.Int.bool to arith (#3083) 2024-04-01 10:05:08 -07:00
TorchToLinalg [torch] Fix aten.squeeze lowering to use result shape (#3106) 2024-04-04 09:43:12 -07:00
TorchToSCF Fix SCF Forloop fails to convert to linalg when a tensor argument is supplied to the loop block (#3040) 2024-03-20 11:04:02 -07:00
TorchToStablehlo [Torch] Support Aten__And__ScalarOp (#3114) 2024-04-08 20:24:17 +08:00
TorchToTMTensor [torch] Rework lowering to tm_tensor.scatter to stop serialization (#2940) 2024-02-27 11:46:57 -08:00
TorchToTensor [torch] Improve shape inference for `torch-to-linalg` path for reshapes (#3055) 2024-03-26 12:41:40 -07:00
TorchToTosa Dynamic size support for flatten (#3005) 2024-03-19 15:19:29 -07:00
Utils [MLIR][TORCH] Fix OnnxToLinalg lowering issue for sub and sum op (#2954) 2024-02-29 21:48:46 +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