torch-mlir/lib/Dialect/Torch
lingzhiz1998 5bc59ce1fa
[TorchToLinalg] Support lowering MaxPool3dWithIndices (#3652)
Support torch.MaxPool3dWithIndices lowering to linalg backend.
2024-08-27 14:14:25 -05:00
..
IR [torch] Unpacking sometimes misses shape inference (#3609) 2024-08-08 16:17:31 -07:00
Transforms [TorchToLinalg] Support lowering MaxPool3dWithIndices (#3652) 2024-08-27 14:14:25 -05:00
Utils [TorchToLinalg] Fix Quantized Convolution Accumulator Type (#3459) 2024-06-20 13:54:20 -07:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00