Home
last modified time | relevance | path

Searched defs:dimRank (Results 1 – 11 of 11) sorted by relevance

/llvm-project/mlir/include/mlir/ExecutionEngine/SparseTensor/
H A DStorage.h200 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 DMapRef.h92 const uint64_t dimRank; variable
H A DCOO.h105 const uint64_t dimRank = getRank(); in add() local
/llvm-project/mlir/lib/ExecutionEngine/SparseTensor/
H A DStorage.cpp28 uint64_t dimRank, const uint64_t *dimSizes, uint64_t lvlRank, in SparseTensorStorageBase()
/llvm-project/mlir/lib/ExecutionEngine/
H A DSparseTensorRuntime.cpp166 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 DCodegenUtils.cpp417 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 DSparseTensorDialect.cpp611 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 DSparseTensorConversion.cpp116 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 DSparseTensorRewriting.cpp1030 const Dimension dimRank = srcTp.getDimRank(); matchAndRewrite() local
1506 const Dimension dimRank = srcTp.getDimRank(); matchAndRewrite() local
H A DSparseTensorCodegen.cpp166 const Dimension dimRank = stt.getDimRank(); createDimSizes() local
/llvm-project/mlir/lib/Dialect/SparseTensor/IR/Detail/
H A DVar.h215 constexpr Ranks(unsigned symRank, unsigned dimRank, unsigned lvlRank) in Ranks()