Searched refs:dimSizes (Results 1 – 3 of 3) sorted by relevance
144 : dimSizes(dimSizes) { in SparseTensorCOO()231 permsz[perm[r]] = dimSizes[r]; in newSparseTensorCOO()268 : dimSizes(dimSizes), rev(getRank()), in SparseTensorStorageBase()296 return dimSizes[d]; in getDimSize()369 const std::vector<uint64_t> dimSizes; member in __anon2f09016c0111::SparseTensorStorageBase803 permsz[t] = dimSizes[s]; in SparseTensorEnumeratorBase()914 : dimSizes(dimSizes), dimTypes(sparsity), nnz(getRank()) { in SparseTensorNNZ()997 const uint64_t sz = dimSizes[d]; in forallIndices()1403 auto &dimSizes = coo->getDimSizes(); in outSparseTensor() local1412 file << dimSizes[r] << " "; in outSparseTensor()[all …]
263 SmallVector<int64_t> dimSizes($_op.getPermutationMap().getNumDims(), 0);271 dimSizes[dim.cast<AffineDimExpr>().getPosition()] = size;273 return dimSizes;
725 auto dimSizes = op.getMaskDimSizes(); in matchAndRewrite() local729 assert(dimSizes.size() == 1 && in matchAndRewrite()731 bool value = dimSizes[0].cast<IntegerAttr>().getInt() == 1; in matchAndRewrite()748 dimSizes[0].cast<IntegerAttr>().getInt()); in matchAndRewrite()765 newDimSizes.push_back(dimSizes[r].cast<IntegerAttr>().getInt()); in matchAndRewrite()