torch-mlir/lib
Vivek Khandelwal 4df96616db
[MLIR][TORCH] Modify Onnx.Reshape lowering for static shape cases (#2852)
This commit modifies the OnnxToTorch lowering of Onnx.Reshape op by
creating the result shape list for the aten.reshape using the result
shape values inferred from the op's result shape.

Signed-Off By: Vivek Khandelwal <vivekkhandelwal1424@gmail.com>
2024-02-07 17:44:07 -08:00
..
CAPI Clang format refresh (#2812) 2024-01-29 12:59:33 -05:00
Conversion [MLIR][TORCH] Modify Onnx.Reshape lowering for static shape cases (#2852) 2024-02-07 17:44:07 -08:00
Dialect [torch] Add `torch.aten.where.*` folders (#2886) 2024-02-07 19:43:31 -05:00
RefBackend [TorchToLinalg] Lower aten.cat to tensor.concat (#2650) 2023-12-15 15:45:32 -05:00
CMakeLists.txt Enable -Werror in lib/ and LTC. (#2841) 2024-01-30 23:33:21 -08:00
InitAll.cpp Initial TorchOnnxToTorch conversion pipeline. (#2585) 2023-11-21 21:02:55 -08:00