Lines Matching defs:toMemref
89 RewriterBase &rewriter, ToMemrefOp toMemref,
91 auto memrefToTensor = toMemref.getTensor().getDefiningOp<ToTensorOp>();
96 Type destType = toMemref.getType();
100 rewriter.replaceOp(toMemref, memrefToTensor.getMemref());
115 rewriter.replaceOp(toMemref, *replacement);
128 rewriter.replaceOpWithNewOp<memref::CastOp>(toMemref, destType,
743 if (auto toMemref = getMemref().getDefiningOp<ToMemrefOp>())
746 if (toMemref->getBlock() == this->getOperation()->getBlock() &&
747 toMemref->getNextNode() == this->getOperation())
748 return toMemref.getTensor();
791 LogicalResult matchAndRewrite(ToMemrefOp toMemref,
794 toMemref.getOperand().getDefiningOp<tensor::CastOp>();
803 Value memref = rewriter.create<ToMemrefOp>(toMemref.getLoc(), memrefType,
805 rewriter.replaceOpWithNewOp<memref::CastOp>(toMemref, toMemref.getType(),
816 LogicalResult matchAndRewrite(ToMemrefOp toMemref,
820 return foldToMemrefToTensorPair(rewriter, toMemref, options);
831 auto toMemref = load.getMemref().getDefiningOp<ToMemrefOp>();
832 if (!toMemref)
835 rewriter.replaceOpWithNewOp<tensor::ExtractOp>(load, toMemref.getTensor(),