torch-mlir/lib/Dialect/TorchConversion/Transforms
Vivek Khandelwal 1a2a9e066f [MLIR][TORCH] Add TorchToTMTensor pass
This pass is added to lower ops, which can not be lowered
via the TorchToLinalg pass, such as `torch.bincount` op.
This pass also uses torch-mlir's TMTensor Dialect to lower the
complex ops.

Also add torch.bincount op lowering with the help of TMTensor dialect

Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
2022-03-08 22:52:34 +05:30
..
BackendTypeConversion.cpp [Torch][Linalg] Add basic support for RNG 2022-01-31 18:56:42 -05:00
CMakeLists.txt [MLIR][TORCH] Add TorchToTMTensor pass 2022-03-08 22:52:34 +05:30
PassDetail.h Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
Passes.cpp [MLIR][TORCH] Add TorchToTMTensor pass 2022-03-08 22:52:34 +05:30
VerifyInvariantsBeforeBackendLowering.cpp Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
VerifyLinalgOnTensorsBackendContract.cpp [MLIR][TORCH] Add TorchToTMTensor pass 2022-03-08 22:52:34 +05:30
VerifyTosaBackendContract.cpp * [tosa] Support for AtenPowTensorScalarOp with constant Scalar as input 2022-01-11 22:55:54 -05:00