mirror of https://github.com/llvm/torch-mlir
1a2a9e066f
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> |
||
|---|---|---|
| .. | ||
| TorchToLinalg | ||
| TorchToSCF | ||
| TorchToStd | ||
| TorchToTMTensor | ||
| TorchToTosa | ||
| Utils | ||
| CMakeLists.txt | ||
| PassDetail.h | ||
| Passes.cpp | ||