torch-mlir/lib
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
..
CAPI Clang format refresh (#2812) 2024-01-29 12:59:33 -05:00
Conversion [torch] Fix aten.squeeze lowering to use result shape (#3106) 2024-04-04 09:43:12 -07:00
Dialect build: manually update PyTorch version (#3094) 2024-04-03 10:48:37 +05:30
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