Searched refs:srcStt (Results 1 – 4 of 4) sorted by relevance
1402 SparseTensorType srcStt = getSparseTensorType(getSource()); in needsExtraSort() 1410 if (srcStt.isAllOrdered() && dstStt.isAllOrdered() && in needsExtraSort() 1411 srcStt.hasSameDimToLvl(dstStt)) { in needsExtraSort() 1544 auto srcStt = getSparseTensorType(source); in build() 1545 SmallVector<int64_t> srcLvlShape = srcStt.getLvlShape();1549 RankedTensorType::get(dstDimShape, srcStt.getElementType(), dstEnc); in verify() 1554 auto srcStt = getSparseTensorType(getSource()); in verify() 1556 ArrayRef<LevelType> srcLvlTps = srcStt.getLvlTypes(); in verify() 1566 if (srcStt.getPosWidth() != dstStt.getPosWidth() || in verify() 1567 srcStt in verify() 1396 SparseTensorType srcStt = getSparseTensorType(getSource()); needsExtraSort() local 1537 auto srcStt = getSparseTensorType(source); build() local 1547 auto srcStt = getSparseTensorType(getSource()); verify() local 2002 SparseTensorType srcStt = getSparseTensorType(getInputCoo()); verify() local [all...]
714 auto srcStt = getSparseTensorType(op.getTensor()); in rewriteOp() local728 blockArgTps.push_back(srcStt.getElementType()); in rewriteOp()741 ValueRange dimCrds = srcStt.translateCrds(rewriter, loc, lvlCrds, in rewriteOp()744 body->getArguments().take_front(srcStt.getDimRank()), dimCrds); in rewriteOp()745 body->eraseArguments(0, srcStt.getDimRank()); in rewriteOp()
660 SparseTensorType srcStt = getSparseTensorType(op.getInputCoo());664 assert(dstStt.isAllOrdered() && !srcStt.isAllOrdered() && in matchAndRewrite() 665 dstStt.isCOOType() && srcStt.isCOOType()); in matchAndRewrite() 666 assert(dstStt.hasSameDimToLvl(srcStt)); in matchAndRewrite() 676 assert(dstStt.hasSameDimToLvl(srcStt)); in matchAndRewrite() 679 auto id = AffineMap::getMultiDimIdentityMap(srcStt.getLvlRank(), ctx); in matchAndRewrite() 668 SparseTensorType srcStt = getSparseTensorType(op.getInputCoo()); matchAndRewrite() local
1282 SparseTensorType srcStt = getSparseTensorType(op.getSource()); in matchAndRewrite() 1295 bool skipZeroCheck = srcStt.hasEncoding() || fromSparseConst; in matchAndRewrite() 1280 SparseTensorType srcStt = getSparseTensorType(op.getSource()); matchAndRewrite() local