Lines Matching refs:getDimSizes
198 const std::vector<uint64_t> &getDimSizes() const { return dimSizes; } in getDimSizes() function
291 const std::vector<uint64_t> &getDimSizes() const { return dimSizes; } in getDimSizes() function in __anon2f09016c0111::SparseTensorStorageBase
426 sz = checkedMul(sz, getDimSizes()[r]); in SparseTensorStorage()
432 assert(coo->getDimSizes() == getDimSizes() && "Tensor size mismatch"); in SparseTensorStorage()
556 const auto &coosz = coo->getDimSizes(); in newSparseTensor()
657 return parentSz * getDimSizes()[d]; in assembledSize()
704 const uint64_t sz = getDimSizes()[d]; in finalizeSegment()
799 const auto &dimSizes = src.getDimSizes(); // in source storage-order in SparseTensorEnumeratorBase()
886 const uint64_t sz = src.getDimSizes()[d]; in forallElements()
1019 SparseTensorNNZ nnz(getDimSizes(), getDimTypes()); in SparseTensorStorage()
1070 parentPos = parentPos * getDimSizes()[r] + ind[r]; in SparseTensorStorage()
1207 const uint64_t *getDimSizes() const { return idata + 2; } in getDimSizes() function in __anon2f09016c0111::SparseTensorFile
1376 auto *coo = SparseTensorCOO<V>::newSparseTensorCOO(rank, stfile.getDimSizes(), in openSparseTensorCOO()
1403 auto &dimSizes = coo->getDimSizes(); in outSparseTensor()
1480 shape[i] = coo->getDimSizes()[i]; in fromMLIRSparseTensor()
1845 const uint64_t *dimSizes = stfile.getDimSizes(); in readSparseTensorShape()