torch-mlir/lib/Conversion
Xida Ren (Cedar) dd967eb199
[ONNX] Support onnx.LSTM (#2969)
This PR only performs a lit test. In lieu of an e2e test, https://github.com/nod-ai/SHARK-TestSuite/pull/142 makede sure that the lowering works & the numbers check out.

Co-authored-by: Xida Ren <xida.ren.dev@gmail.com>
2024-04-08 12:23:33 -07:00
..
TorchConversionToMLProgram Clang format refresh (#2812) 2024-01-29 12:59:33 -05:00
TorchOnnxToTorch [ONNX] Support onnx.LSTM (#2969) 2024-04-08 12:23:33 -07:00
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