torch-mlir/lib/Conversion
Ramiro Leal-Cavazos 93f6d8e776
[LINALG] Add 0-rank case for `aten.permute` (#940)
The function `AffineMap::inferFromExprList` does not work if the first
vector of expressions is empty, because it uses these expressions to
obtain the context. This prevented `aten.permute` from working for
inputs of 0-rank. This commit adds support for 0-rank inputs.
2022-06-14 12:50:46 -07:00
..
TorchToLinalg [LINALG] Add 0-rank case for `aten.permute` (#940) 2022-06-14 12:50:46 -07:00
TorchToSCF [TORCH][MLIR]Add lowering for control flow operations. 2022-04-29 16:25:58 +05:30
TorchToStd Add E2E support for AtenAllBoolOp (#874) 2022-06-01 18:20:25 -07:00
TorchToTMTensor llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -07:00
TorchToTosa [tosa] Support for AtenAvgPool2d op 2022-05-27 07:56:37 +05:30
Utils [MLIR][TORCH] Add decomposition of aten.adaptive_avg_pool2d op 2022-05-27 07:56:37 +05:30
CMakeLists.txt [MLIR][TORCH] Add TorchToTMTensor pass 2022-03-08 22:52:34 +05:30
PassDetail.h llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -07:00
Passes.cpp [MLIR][TORCH] Add TorchToTMTensor pass 2022-03-08 22:52:34 +05:30