Searched defs:dimSizes (Results 1 – 6 of 6) sorted by relevance
| /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()
|
| /llvm-project/mlir/lib/ExecutionEngine/ |
| H A D | SparseTensorRuntime.cpp | 171 const index_type *dimSizes = MEMREF_GET_PAYLOAD(dimSizesRef); in _mlir_ciface_newSparseTensor() local
383 auto *dimSizes = const_cast<uint64_t *>(reader.getDimSizes()); in _mlir_ciface_getSparseTensorReaderDimSizes() local
424 index_type *dimSizes = MEMREF_GET_PAYLOAD(dimSizesRef); in MLIR_SPARSETENSOR_FOREVERY_V_O() 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/Dialect/Vector/Transforms/ |
| H A D | LowerVectorMask.cpp | 108 auto dimSizes = op.getMaskDimSizes(); in matchAndRewrite() local
|
| H A D | VectorDropLeadUnitDim.cpp | 553 ArrayRef<int64_t> dimSizes = mask.getMaskDimSizes(); in matchAndRewrite() local
|
| /llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/ |
| H A D | SparseTensorRewriting.cpp | 1508 Value dimSizes = genAlloca(rewriter, loc, dimRank, indexTp); matchAndRewrite() local
|