torch-mlir/lib/Conversion
Rob Suderman 9d9a05366e
[torch] Fix aten.squeeze lowering to use result shape (#3106)
Squeezes can be ambiguous without the output shape information. For
instance (1, 1, 256) squeezed can be either (1, 256) or (256). We need
to check the resulting shape to know what the shape should look like.
2024-04-04 09:43:12 -07:00
..
TorchConversionToMLProgram Clang format refresh (#2812) 2024-01-29 12:59:33 -05:00
TorchOnnxToTorch [MLIR][TORCH] Fix Onnx.TopK lowering (#3103) 2024-04-03 22:12:48 +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 AtenDivTensorModeOp with static int input for linalg and stablehlo backend (#3088) 2024-04-02 17:28:53 +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