Lines Matching refs:CastOp

263     auto castOp = operand.get().getDefiningOp<memref::CastOp>();  in foldMemRefCast()
264 if (castOp && memref::CastOp::canFoldIntoConsumerOp(castOp)) { in foldMemRefCast()
549 rewriter.replaceOpWithNewOp<tensor::CastOp>(padOp, oldResultType, in matchAndRewrite()
1155 if (!tensor::CastOp::areCastCompatible(returnedArg.getType(), in matchAndRewrite()
1158 returnedArg = rewriter.create<tensor::CastOp>( in matchAndRewrite()
1292 rewriter.replaceOpWithNewOp<tensor::CastOp>(op, op.getType(), newOp); in matchAndRewrite()
1342 rewriter.replaceOpWithNewOp<tensor::CastOp>( in matchAndRewrite()
1385 : public OpRewritePattern<tensor::CastOp> {
1386 using OpRewritePattern<tensor::CastOp>::OpRewritePattern;
1388 LogicalResult matchAndRewrite(tensor::CastOp castOp, in matchAndRewrite()
1660 auto castOp = opOperand->get().getDefiningOp<tensor::CastOp>(); in matchAndRewrite()
1672 auto tensorCastOp = opOperand->get().getDefiningOp<tensor::CastOp>(); in matchAndRewrite()
1679 auto tensorCastOp = opOperand->get().getDefiningOp<tensor::CastOp>(); in matchAndRewrite()
1694 replacements.push_back(rewriter.create<tensor::CastOp>( in matchAndRewrite()
1708 struct FoldTensorCastConsumerOp : public OpRewritePattern<tensor::CastOp> {
1709 using OpRewritePattern<tensor::CastOp>::OpRewritePattern;
1711 LogicalResult matchAndRewrite(tensor::CastOp castOp, in matchAndRewrite()
1740 rewriter.create<tensor::CastOp>(loc, resultType, outOperand->get()); in matchAndRewrite()
1752 Value castBack = rewriter.create<tensor::CastOp>( in matchAndRewrite()
1780 if (auto castOp = dyn_cast<tensor::CastOp>(parentOp)) { in populateMap()
1844 Value newOperand = rewriter.create<tensor::CastOp>(loc, resultType, src); in createNewOperandWithStaticSizes()
1911 ? rewriter.create<tensor::CastOp>(loc, oldType, newResult) in matchAndRewrite()