/llvm-project/mlir/include/mlir/ExecutionEngine/SparseTensor/ |
H A D | Storage.h | 200 SparseTensorStorage(uint64_t dimRank, const uint64_t *dimSizes, in SparseTensorStorage() 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() 602 uint64_t dimRank, const uint64_t *dimSizes, uint64_t lvlRank, in newFromBuffers() 618 uint64_t dimRank, const uint64_t *dimSizes, uint64_t lvlRank, in SparseTensorStorage() 672 uint64_t dimRank, const uint64_t *dimSizes, uint64_t lvlRank, in SparseTensorStorage()
|
H A D | MapRef.h | 92 const uint64_t dimRank; variable
|
H A D | COO.h | 105 const uint64_t dimRank = getRank(); in add() local
|
/llvm-project/mlir/lib/ExecutionEngine/SparseTensor/ |
H A D | Storage.cpp | 28 uint64_t dimRank, const uint64_t *dimSizes, uint64_t lvlRank, in SparseTensorStorageBase()
|
/llvm-project/mlir/lib/ExecutionEngine/ |
H A D | SparseTensorRuntime.cpp | 166 const uint64_t dimRank = MEMREF_GET_USIZE(dimSizesRef); in _mlir_ciface_newSparseTensor() local 373 const uint64_t dimRank = MEMREF_GET_USIZE(dimShapeRef); local
|
/llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/Utils/ |
H A D | CodegenUtils.cpp | 417 const Dimension dimRank = getSparseTensorType(src).getDimRank(); in sizesFromSrc() local 439 const Dimension dimRank = stt.getDimRank(); in foreachInSparseConstant() local 588 Dimension dimRank = stt.getDimRank(); genReader() local 634 const Dimension dimRank = stt.getDimRank(); genMapBuffers() local [all...] |
/llvm-project/mlir/lib/Dialect/SparseTensor/IR/ |
H A D | SparseTensorDialect.cpp | 611 const Dimension dimRank = dlm.getDimRank(); in parse() local 871 const Dimension dimRank = dimToLvl ? dimToLvl.getNumDims() : lvlRank; in verify() local 914 const Dimension dimRank = dimShape.size(); verifyEncoding() local 1380 for (Dimension d = 0, dimRank = tp1.getRank(); d < dimRank; d++) verify() local 1848 const Dimension dimRank = dstTp.getDimRank(); verify() local 1934 const Dimension dimRank = stt.getDimRank(); build() local 1957 const Dimension dimRank = t.getDimRank(); verify() local [all...] |
/llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/ |
H A D | SparseTensorConversion.cpp | 116 const Dimension dimRank = stt.getDimRank(); in fillDimSizes() local 419 const Dimension dimRank = stt.getDimRank(); in matchAndRewrite() local 450 const Dimension dimRank = stt.getDimRank(); in matchAndRewrite() local
|
H A D | SparseTensorRewriting.cpp | 1030 const Dimension dimRank = srcTp.getDimRank(); matchAndRewrite() local 1506 const Dimension dimRank = srcTp.getDimRank(); matchAndRewrite() local
|
H A D | SparseTensorCodegen.cpp | 166 const Dimension dimRank = stt.getDimRank(); createDimSizes() local
|
/llvm-project/mlir/lib/Dialect/SparseTensor/IR/Detail/ |
H A D | Var.h | 215 constexpr Ranks(unsigned symRank, unsigned dimRank, unsigned lvlRank) in Ranks()
|