Lines Matching defs:stt

137 static RankedTensorType getBufferType(const SparseTensorType &stt,
139   return needTmpCOO ? stt.getCOOType(/*ordered=*/false)
140                     : stt.getRankedTensorType();
724     auto stt = getSparseTensorType(tensor);
732     printSizes(rewriter, loc, tensor, stt.getDimRank(), /*isDim=*/true);
734     printSizes(rewriter, loc, tensor, stt.getLvlRank(), /*isDim=*/false);
737     foreachFieldAndTypeInSparseTensor(stt, [&rewriter, &loc, &tensor,
738                                             &stt](Type, FieldIndex,
765         if (stt.getAoSCOOStart() == l)
1144     auto stt = tryGetSparseTensorType(op.getSource());
1145     if (!dim || !stt || !stt->hasEncoding())
1148     if (stt->isPermutation()) {
1150                                          toLvl(stt->getEncoding(), *dim));
1162     for (Level l = 0; l < stt->getLvlRank(); l++) {
1169     AffineExpr lvl2DimExp = stt->getLvlToDim().getResult(*dim);
1171         op.getLoc(), AffineMap::get(stt->getLvlRank(), 0, lvl2DimExp),
1373     const auto stt = getSparseTensorType(input);
1374     const Level lvlRank = stt.getLvlRank();
1385     const auto enc = stt.getEncoding();
1465     auto stt = getSparseTensorType(op.getResult());
1466     if (!stt.hasEncoding() || stt.getAoSCOOStart() == 0)
1473     RankedTensorType dstTp = stt.getRankedTensorType();
1474     RankedTensorType cooTp = stt.getCOOType(/*ordered=*/true);
1477     auto enc = stt.getEncoding();
1478     if (!stt.isPermutation()) { // demap coo, demap dstTp
1484     if (!stt.isPermutation()) // remap to original enc