Searched refs:getDimSizes (Results 1 – 1 of 1) sorted by relevance
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()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()1403 auto &dimSizes = coo->getDimSizes(); in outSparseTensor()1480 shape[i] = coo->getDimSizes()[i]; in fromMLIRSparseTensor()[all …]