103 assertPermutedSizesMatchShape(const std::vector<uint64_t> &dimSizes,  in assertPermutedSizesMatchShape()  argument
107   assert(rank == dimSizes.size() && "Rank mismatch");  in assertPermutedSizesMatchShape()
109     assert((shape[r] == 0 || shape[r] == dimSizes[perm[r]]) &&  in assertPermutedSizesMatchShape()
143   SparseTensorCOO(const std::vector<uint64_t> &dimSizes, uint64_t capacity)  in SparseTensorCOO()
144       : dimSizes(dimSizes) {  in SparseTensorCOO()
159       assert(ind[r] < dimSizes[r] && "Index is too large for the dimension");  in add()
195   uint64_t getRank() const { return dimSizes.size(); }  in getRank()
198   const std::vector<uint64_t> &getDimSizes() const { return dimSizes; }  in getDimSizes()
225                                                 const uint64_t *dimSizes,  in newSparseTensorCOO()
230       assert(dimSizes[r] > 0 && "Dimension size zero has trivial storage");  in newSparseTensorCOO()
231       permsz[perm[r]] = dimSizes[r];  in newSparseTensorCOO()
237   const std::vector<uint64_t> dimSizes; // per-dimension sizes  member
266   SparseTensorStorageBase(const std::vector<uint64_t> &dimSizes,  in SparseTensorStorageBase()  argument
268       : dimSizes(dimSizes), rev(getRank()),  in SparseTensorStorageBase()
275       assert(dimSizes[r] > 0 && "Dimension size zero has trivial storage");  in SparseTensorStorageBase()
288   uint64_t getRank() const { return dimSizes.size(); }  in getRank()
291   const std::vector<uint64_t> &getDimSizes() const { return dimSizes; }  in getDimSizes()
296     return dimSizes[d];  in getDimSize()
369   const std::vector<uint64_t> dimSizes;  member in __anon2f09016c0111::SparseTensorStorageBase
394   SparseTensorStorage(const std::vector<uint64_t> &dimSizes,  in SparseTensorStorage()  argument
396       : SparseTensorStorageBase(dimSizes, perm, sparsity), pointers(getRank()),  in SparseTensorStorage()
405   SparseTensorStorage(const std::vector<uint64_t> &dimSizes,  in SparseTensorStorage()  argument
408       : SparseTensorStorage(dimSizes, perm, sparsity) {  in SparseTensorStorage()
451   SparseTensorStorage(const std::vector<uint64_t> &dimSizes,
799     const auto &dimSizes = src.getDimSizes(); // in source storage-order  in SparseTensorEnumeratorBase()  local
803       permsz[t] = dimSizes[s];  in SparseTensorEnumeratorBase()
912   SparseTensorNNZ(const std::vector<uint64_t> &dimSizes,  in SparseTensorNNZ()  argument
914       : dimSizes(dimSizes), dimTypes(sparsity), nnz(getRank()) {  in SparseTensorNNZ()
915     assert(dimSizes.size() == dimTypes.size() && "Rank mismatch");  in SparseTensorNNZ()
938       sz = checkedMul(sz, dimSizes[r]);  in SparseTensorNNZ()
947   uint64_t getRank() const { return dimSizes.size(); }  in getRank()
955     assert(enumerator.permutedSizes() == dimSizes && "Tensor size mismatch");  in initialize()
985       parentPos = parentPos * dimSizes[r] + ind[r];  in add()
997       const uint64_t sz = dimSizes[d];  in forallIndices()
1005   const std::vector<uint64_t> &dimSizes;  member in __anon2f09016c0111::SparseTensorNNZ
1012     const std::vector<uint64_t> &dimSizes, const uint64_t *perm,  in SparseTensorStorage()  argument
1014     : SparseTensorStorage(dimSizes, perm, sparsity) {  in SparseTensorStorage()
1403   auto &dimSizes = coo->getDimSizes();  in outSparseTensor()  local
1412     file << dimSizes[r] << " ";  in outSparseTensor()
1413   file << dimSizes[rank - 1] << std::endl;  in outSparseTensor()
1845   const uint64_t *dimSizes = stfile.getDimSizes();  in readSparseTensorShape()  local
1847   out->assign(dimSizes, dimSizes + rank);  in readSparseTensorShape()