torch-mlir/lib
Rob Suderman f97cd4893f
[torch] Improve shape inference for dynamic shapes (#3091)
Shapes can be processed as tensors to represent the set of dimensions.
As reshapes take a list of scalars this can result in a single dynamic
dimension blocking the adjacent static dimensions.

This pass attempts to de-couple tensor computations related to shapes
and propagate values to better support lowering scalar tensor
computations.
2024-04-02 16:19:57 -07:00
..
CAPI Clang format refresh (#2812) 2024-01-29 12:59:33 -05:00
Conversion [torch] Improve shape inference for dynamic shapes (#3091) 2024-04-02 16:19:57 -07:00
Dialect [torch] Improve shape inference for dynamic shapes (#3091) 2024-04-02 16:19:57 -07:00
RefBackend [TorchToLinalg] Lower aten.cat to tensor.concat (#2650) 2023-12-15 15:45:32 -05:00
CMakeLists.txt [Stablehlo] lowering chlo to stablehlo in torch-to-stablehlo pipeline (#3037) 2024-03-19 21:18:54 +08:00
InitAll.cpp [Stablehlo] lowering chlo to stablehlo in torch-to-stablehlo pipeline (#3037) 2024-03-19 21:18:54 +08:00