Lines Matching defs:stt
76 SparseTensorType stt, Value tensor,
79 assert(stt.hasEncoding());
87 stt.isIdentity() ? lvl : stt.getDimToLvl().getDimPosition(lvl);
88 const Size sz = stt.getDynamicDimSize(dim);
103 SparseTensorType stt, Value tensor,
105 const Size sz = stt.getDynamicDimSize(dim);
108 if (stt.hasEncoding())
114 static void fillDimSizes(OpBuilder &builder, Location loc, SparseTensorType stt,
116 const Dimension dimRank = stt.getDimRank();
120 out.push_back(createOrFoldDimCall(builder, loc, stt, tensor, d));
127 SparseTensorType stt,
130 fillDimSizes(builder, loc, stt, tensor, out);
145 SparseTensorType stt) {
147 lvlTypes.reserve(stt.getLvlRank());
148 for (const auto lt : stt.getEncoding().getLvlTypes())
192 NewCallParams &genBuffers(SparseTensorType stt,
195 assert(dimSizesValues.size() == static_cast<size_t>(stt.getDimRank()));
197 params[kParamLvlTypes] = genLvlTypesBuffer(builder, loc, stt);
204 builder, loc, stt, dimSizesValues, params[kParamDimSizes],
207 const auto enc = stt.getEncoding();
211 constantPrimaryTypeEncoding(builder, loc, stt.getElementType());
258 SparseTensorType stt, Value ptr) {
259 auto eltTp = stt.getElementType();
268 SparseTensorType stt, Value ptr, Level l) {
269 Type posTp = stt.getPosType();
280 SparseTensorType stt, Value ptr, Level l) {
281 Type crdTp = stt.getCrdType();
292 SparseTensorType stt, Value ptr,
294 Type crdTp = stt.getCrdType();
327 const auto stt = getSparseTensorType(op.getSource());
329 if (!stt.hasEncoding())
383 const auto stt = getSparseTensorType(op);
384 if (!stt.hasEncoding())
389 Value reader = genReader(rewriter, loc, stt, adaptor.getOperands()[0],
393 .genBuffers(stt, dimSizesValues, dimSizesBuffer)
412 const auto stt = getSparseTensorType(op);
413 if (!stt.hasEncoding())
419 const Dimension dimRank = stt.getDimRank();
425 stt.isDynamicDim(d)
432 .genBuffers(stt, dimSizesValues)
446 const auto stt = getSparseTensorType(op);
447 if (!stt.hasEncoding())
450 const Dimension dimRank = stt.getDimRank();
456 dimSizesValues.push_back(stt.isDynamicDim(d)
463 .genBuffers(stt, dimSizesValues)
519 auto stt = getSparseTensorType(op.getTensor());
520 auto poss = genPositionsCall(rewriter, op.getLoc(), stt,
536 auto stt = getSparseTensorType(op.getTensor());
537 auto crds = genCoordinatesCall(rewriter, loc, stt, adaptor.getTensor(),
557 auto stt = getSparseTensorType(op.getTensor());
559 rewriter, loc, stt, adaptor.getTensor(), stt.getAoSCOOStart());
576 auto stt = getSparseTensorType(op.getTensor());
577 auto vals = genValuesCall(rewriter, op.getLoc(), stt, adaptor.getTensor());
592 auto stt = getSparseTensorType(op.getTensor());
593 auto vals = genValuesCall(rewriter, op.getLoc(), stt, adaptor.getTensor());
630 const auto stt = getSparseTensorType(op.getDest());
633 if (!stt.hasEncoding())
636 assert(stt.isIdentity() && "Run reinterpret-map before conversion.");
637 const auto elemTp = stt.getElementType();
638 const Level lvlRank = stt.getLvlRank();
723 const auto stt = getSparseTensorType(op.getTensor());
724 const Type elemTp = stt.getElementType();
725 const Level lvlRank = stt.getLvlRank();
784 auto stt = getSparseTensorType(op.getTensor());
788 const Level lvlRank = stt.getLvlRank();
791 if (!stt.isUniqueLvl(l) &&
792 (stt.isCompressedLvl(l) || stt.isLooseCompressedLvl(l))) {
800 if (stt.isWithPos(l)) {
802 genPositionsCall(rewriter, loc, stt, adaptor.getTensor(), l);
808 if (stt.isWithCrd(l)) {
810 genCoordinatesCall(rewriter, loc, stt, adaptor.getTensor(), l);
820 assert(!stt.isUniqueLvl(cooStartLvl) &&
821 (stt.isCompressedLvl(cooStartLvl) ||
822 stt.isLooseCompressedLvl(cooStartLvl)));
824 auto poss = genPositionsCall(rewriter, loc, stt, adaptor.getTensor(),
834 auto crds0 = genCoordinatesCall(rewriter, loc, stt, adaptor.getTensor(),
836 auto crds1 = genCoordinatesCall(rewriter, loc, stt, adaptor.getTensor(),
861 auto vals = genValuesCall(rewriter, loc, stt, adaptor.getTensor());