Searched refs:dimTypes (Results 1 – 3 of 3) sorted by relevance
269 dimTypes(sparsity, sparsity + getRank()) { in SparseTensorStorageBase()276 assert((dimTypes[r] == DimLevelType::kDense || in SparseTensorStorageBase()277 dimTypes[r] == DimLevelType::kCompressed) && in SparseTensorStorageBase()309 return (dimTypes[d] == DimLevelType::kCompressed); in isCompressedDim()371 const std::vector<DimLevelType> dimTypes; member in __anon2f09016c0111::SparseTensorStorageBase914 : dimSizes(dimSizes), dimTypes(sparsity), nnz(getRank()) { in SparseTensorNNZ()915 assert(dimSizes.size() == dimTypes.size() && "Rank mismatch"); in SparseTensorNNZ()920 switch (dimTypes[r]) { in SparseTensorNNZ()969 assert(dimTypes[stopDim] == DimLevelType::kCompressed && in forallIndices()983 if (dimTypes[r] == DimLevelType::kCompressed) in add()[all …]
35 ArrayRef<SparseTensorEncodingAttr::DimLevelType> dimTypes = in isSparseTensor() local37 for (auto dimType : dimTypes) in isSparseTensor()
368 ArrayRef<SparseTensorEncodingAttr::DimLevelType> dimTypes) { in canUseDirectConversion() argument370 for (uint64_t rank = dimTypes.size(), r = 0; r < rank; r++) { in canUseDirectConversion()371 switch (dimTypes[r]) { in canUseDirectConversion()