Lines Matching refs:toMemref
89 ToMemrefOp toMemref) { in foldToMemrefToTensorPair() argument
90 auto memrefToTensor = toMemref.getTensor().getDefiningOp<ToTensorOp>(); in foldToMemrefToTensorPair()
95 Type destType = toMemref.getType(); in foldToMemrefToTensorPair()
99 rewriter.replaceOp(toMemref, memrefToTensor.getMemref()); in foldToMemrefToTensorPair()
114 rewriter.replaceOp(toMemref, *replacement); in foldToMemrefToTensorPair()
127 rewriter.replaceOpWithNewOp<memref::CastOp>(toMemref, destType, in foldToMemrefToTensorPair()
533 if (auto toMemref = getMemref().getDefiningOp<ToMemrefOp>()) in fold() local
536 if (toMemref->getBlock() == this->getOperation()->getBlock() && in fold()
537 toMemref->getNextNode() == this->getOperation()) in fold()
538 return toMemref.getTensor(); in fold()
581 LogicalResult matchAndRewrite(ToMemrefOp toMemref, in matchAndRewrite()
584 toMemref.getOperand().getDefiningOp<tensor::CastOp>(); in matchAndRewrite()
593 Value memref = rewriter.create<ToMemrefOp>(toMemref.getLoc(), memrefType, in matchAndRewrite()
595 rewriter.replaceOpWithNewOp<memref::CastOp>(toMemref, toMemref.getType(), in matchAndRewrite()
606 LogicalResult matchAndRewrite(ToMemrefOp toMemref, in matchAndRewrite()
608 return foldToMemrefToTensorPair(rewriter, toMemref); in matchAndRewrite()
619 auto toMemref = load.getMemref().getDefiningOp<ToMemrefOp>(); in matchAndRewrite() local
620 if (!toMemref) in matchAndRewrite()
623 rewriter.replaceOpWithNewOp<tensor::ExtractOp>(load, toMemref.getTensor(), in matchAndRewrite()