Lines Matching defs:dstTp
187 ShapedType dstTp, ValueRange srcs,
189 auto dstShape = dstTp.getShape();
885 const auto dstTp = tryGetSparseTensorType(op.getResult());
886 if (!srcTp || !dstTp)
889 if (!srcTp->hasEncoding() || !dstTp->hasEncoding() ||
890 !dstTp->hasStaticDimShape())
896 for (Dimension d : dstTp->getDimShape())
903 dstTp->withoutDimToLvl(),
904 !srcTp->isAllOrdered() || !srcTp->isIdentity() || !dstTp->isIdentity());
950 for (Dimension i = 0; i < dstTp->getDimRank(); i++)
963 if (bufferTp != *dstTp) {
964 auto dstRTT = dstTp->getRankedTensorType();
985 const auto dstTp = getSparseTensorType(op.getResult());
986 if (!srcTp.hasEncoding() || !dstTp.hasEncoding())
995 if (dstTp.hasStaticDimShape()) {
996 for (Dimension d : dstTp.getDimShape())
999 ArrayRef<Size> dstShape = dstTp.getDimShape();
1011 dstTp.withoutDimToLvl(),
1012 !srcTp.isAllOrdered() || !srcTp.isIdentity() || !dstTp.isIdentity());
1048 if (bufferTp != dstTp) {
1049 auto dstRTT = dstTp.getRankedTensorType();
1189 const auto dstTp = getSparseTensorType(op);
1192 concatSizesFromInputs(rewriter, sizes, loc, dstTp, op.getInputs(), conDim);
1198 // %tmp = bufferization.alloc_tensor dstTp
1207 TensorLike dstBuf(rewriter, loc, dstTp.getRankedTensorType(), sizes);
1225 if (!dstTp.isAllDense()) {
1256 Value ret = dstBuf.finalize(rewriter, loc, dstTp.getRankedTensorType());
1473 RankedTensorType dstTp = stt.getRankedTensorType();
1478 if (!stt.isPermutation()) { // demap coo, demap dstTp
1481 dstTp = getSparseTensorType(convert).withEncoding(enc.withoutDimToLvl());
1483 convert = rewriter.create<ConvertOp>(loc, dstTp, convert);