Searched defs:dimToLvl (Results 1 – 7 of 7) sorted by relevance
| /llvm-project/mlir/test/CAPI/ |
| H A D | sparse_tensor.c | 37 MlirAffineMap dimToLvl = in testRoundtripEncoding() local
|
| /llvm-project/mlir/include/mlir/Dialect/SparseTensor/IR/ |
| H A D | SparseTensorType.h | 75 SparseTensorType withDimToLvl(AffineMap dimToLvl) const { in withDimToLvl() argument
369 const AffineMap dimToLvl; global() variable
[all...] |
| /llvm-project/mlir/lib/CAPI/Dialect/ |
| H A D | SparseTensor.cpp | 49 mlirSparseTensorEncodingAttrGet(MlirContext ctx,intptr_t lvlRank,MlirSparseTensorLevelType const * lvlTypes,MlirAffineMap dimToLvl,MlirAffineMap lvlToDim,int posWidth,int crdWidth,MlirAttribute explicitVal,MlirAttribute implicitVal) mlirSparseTensorEncodingAttrGet() argument
|
| /llvm-project/mlir/lib/Bindings/Python/ |
| H A D | DialectSparseTensor.cpp | 46 __anon055f89890102(py::object cls, std::vector<MlirSparseTensorLevelType> lvlTypes, std::optional<MlirAffineMap> dimToLvl, std::optional<MlirAffineMap> lvlToDim, int posWidth, int crdWidth, std::optional<MlirAttribute> explicitVal, std::optional<MlirAttribute> implicitVal, MlirContext context) populateDialectSparseTensorSubmodule() argument
|
| /llvm-project/mlir/lib/Dialect/SparseTensor/IR/ |
| H A D | SparseTensorDialect.cpp | 445 const auto dimToLvl = getDimToLvl(); in getDimRank() local
578 AffineMap dimToLvl = {}; in parse() local
785 AffineMap dimToLvl, AffineMa in verify() argument
1044 inferLvlToDim(AffineMap dimToLvl,MLIRContext * context) inferLvlToDim() argument
1059 inverseBlockSparsity(AffineMap dimToLvl,MLIRContext * context) inverseBlockSparsity() argument
1112 getBlockSize(AffineMap dimToLvl) getBlockSize() argument
1129 isBlockSparsity(AffineMap dimToLvl) isBlockSparsity() argument
1186 if (const auto dimToLvl = enc.getDimToLvl()) toDim() local
[all...] |
| /llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/ |
| H A D | SparseReinterpretMap.cpp | 563 auto dimToLvl = AffineMap::getPermutationMap(perm, linalgOp.getContext()); in resolveCycle() local
|
| /llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/Utils/ |
| H A D | CodegenUtils.cpp | 655 assert(lvlRank == dimToLvl.getNumResults()); in genMapBuffers() local
|