| /llvm-project/mlir/include/mlir/ExecutionEngine/SparseTensor/ |
| H A D | Storage.h | 201 uint64_t lvlRank, const uint64_t *lvlSizes, in SparseTensorStorage()
273 uint64_t lvlRank = getLvlRank(); in getCoordinatesBuffer() local
294 uint64_t lvlRank = getLvlRank(); in lexInsert() local
366 uint64_t lvlRank = getLvlRank(); in sortInPlace() local
455 const uint64_t lvlRank = getLvlRank(); in fromCOO() local
518 const uint64_t lvlRank = getLvlRank(); in endPath() local
532 const uint64_t lvlRank = getLvlRank(); in insPath() local
546 const uint64_t lvlRank = getLvlRank(); in lexDiff() local
581 uint64_t dimRank, const uint64_t *dimSizes, uint64_t lvlRank, in newEmpty()
591 uint64_t dimRank, const uint64_t *dimSizes, uint64_t lvlRank, in newFromCOO()
[all …]
|
| H A D | MapRef.h | 93 const uint64_t lvlRank; variable
|
| /llvm-project/mlir/lib/Bindings/Python/ |
| H A D | DialectSparseTensor.cpp | 76 const int lvlRank = mlirSparseTensorEncodingGetLvlRank(self); in populateDialectSparseTensorSubmodule() local
124 const int lvlRank = mlirSparseTensorEncodingGetLvlRank(self); in populateDialectSparseTensorSubmodule() local
131 const int lvlRank = mlirSparseTensorEncodingGetLvlRank(self); in populateDialectSparseTensorSubmodule() local
136 const int lvlRank = mlirSparseTensorEncodingGetLvlRank(self); in populateDialectSparseTensorSubmodule() local
[all...] |
| /llvm-project/mlir/lib/ExecutionEngine/SparseTensor/ |
| H A D | Storage.cpp | 20 static inline bool isAllDense(uint64_t lvlRank, const LevelType *lvlTypes) { in isAllDense()
28 uint64_t dimRank, const uint64_t *dimSizes, uint64_t lvlRank, in SparseTensorStorageBase()
|
| /llvm-project/mlir/test/CAPI/ |
| H A D | sparse_tensor.c | 46 int lvlRank = mlirSparseTensorEncodingGetLvlRank(originalAttr); in testRoundtripEncoding() local
|
| /llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/ |
| H A D | SparseTensorCodegen.cpp | 116 const Level lvlRank = stt.getLvlRank(); allocSchemeForRank() local
185 Level lvlRank = stt.getLvlRank(); createAllocFields() local
249 for (Level lvl = 0, lvlRank = stt.getLvlRank(); lvl < lvlRank; lvl++) { createAllocFields() local
285 const Level lvlRank = stt.getLvlRank(); genCompressed() local
374 const Level lvlRank = stt.getLvlRank(); genEndInsert() local
479 const Level lvlRank = stt.getLvlRank(); genImplementation() local
538 const Level lvlRank = stt.getLvlRank(); getMangledFuncName() local
1348 for (Level lvl = 0, lvlRank = stt.getLvlRank(); lvl < lvlRank; lvl++) { matchAndRewrite() local
1545 const Level lvlRank = dstTp.getLvlRank(); matchAndRewrite() local
[all...] |
| H A D | Sparsification.cpp | 208 const Level lvlRank = stt.getLvlRank(); in getNumNonTrivialIdxExpOnSparseLvls() local
256 const Level lvlRank = map.getNumResults(); in findSparseAnnotations() local
340 const Level lvlRank = stt.getLvlRank(); in genIndex() local
362 const Level lvlRank = stt.getLvlRank(); genSubscript() local
687 const Level lvlRank = stt.getLvlRank(); genInvariants() local
1027 const Level lvlRank = stt.getLvlRank(); getAllTidLvlsInLatPoints() local
1127 const Level lvlRank = enc.getLvlRank(); genConstantDenseAddressFromLevel() local
[all...] |
| H A D | SparseTensorConversion.cpp | 638 const Level lvlRank = stt.getLvlRank(); in matchAndRewrite() local
725 const Level lvlRank = stt.getLvlRank(); in matchAndRewrite() local
788 const Level lvlRank = stt.getLvlRank(); in matchAndRewrite() local
|
| H A D | SparseStorageSpecifierToLLVM.cpp | 29 const Level lvlRank = enc.getLvlRank(); in getSpecifierFields() local
|
| H A D | SparseReinterpretMap.cpp | 725 Level lvlRank = getSparseTensorType(adaptor.getTensor()).getLvlRank(); in rewriteOp() local
|
| H A D | SparseTensorRewriting.cpp | 1372 const Level lvlRank = stt.getLvlRank(); matchAndRewrite() local
|
| /llvm-project/mlir/lib/CAPI/Dialect/ |
| H A D | SparseTensor.cpp | 48 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/Dialect/SparseTensor/Transforms/Utils/ |
| H A D | CodegenEnv.cpp | 81 const Level lvlRank = linalgOp.getMatchingIndexingMap(&t).getNumResults(); in startEmit() local
|
| H A D | LoopEmitter.cpp | 158 Level lvlRank; in initialize() local
313 const Level lvlRank = stt.getLvlRank(); initializeLoopEmit() local
340 Level lvlRank = SparseTensorType(rtp).getLvlRank(); initSubSectIterator() local
[all...] |
| H A D | IterationGraphSorter.cpp | 209 const Level lvlRank = loop2LvlMap.getNumResults(); in addConstraints() local
|
| H A D | CodegenUtils.cpp | 635 const Level lvlRank = stt.getLvlRank(); genMapBuffers() local
|
| /llvm-project/mlir/include/mlir/Dialect/SparseTensor/IR/ |
| H A D | SparseTensorType.h | 368 const Level lvlRank; global() variable
[all...] |
| /llvm-project/mlir/lib/Dialect/SparseTensor/IR/Detail/ |
| H A D | Var.h | 215 constexpr Ranks(unsigned symRank, unsigned dimRank, unsigned lvlRank) in Ranks()
|
| /llvm-project/mlir/lib/ExecutionEngine/ |
| H A D | SparseTensorRuntime.cpp | 167 const uint64_t lvlRank = MEMREF_GET_USIZE(lvlSizesRef); in _mlir_ciface_newSparseTensor() local
|
| /llvm-project/mlir/lib/Dialect/SparseTensor/IR/ |
| H A D | SparseTensorDialect.cpp | 107 const Level lvlRank = enc.getLvlRank(); in foreachField() local
216 const Level lvlRank = enc.getLvlRank(); in getFieldIndexAndStride() local
607 const Level lvlRank = dlm.getLvlRank(); in parse() local
867 const Level lvlRank = lvlTypes.size(); in verify() local
1252 const Level lvlRank = enc.getLvlRank(); verifySparsifierGetterSetter() local
[all...] |