Lines Matching refs:reshapeOp

500   LogicalResult matchAndRewrite(TensorReshapeOp reshapeOp,  in matchAndRewrite()
502     auto oldFill = reshapeOp.getSrc().template getDefiningOp<FillOp>();  in matchAndRewrite()
508         loc, reshapeOp.getResultType(), oldFill.output(),  in matchAndRewrite()
509         reshapeOp.getReassociation());  in matchAndRewrite()
510     rewriter.replaceOpWithNewOp<FillOp>(reshapeOp, ValueRange{oldFill.value()},  in matchAndRewrite()
1326   LogicalResult matchAndRewrite(TensorReshapeOp reshapeOp,  in matchAndRewrite()
1328     if (!reshapeOp.getSrc().template getDefiningOp<InitTensorOp>())  in matchAndRewrite()
1330     Location loc = reshapeOp.getLoc();  in matchAndRewrite()
1333         cast<ReifyRankedShapedTypeOpInterface>(reshapeOp.getOperation());  in matchAndRewrite()
1340         reshapeOp.getResultType().getElementType());  in matchAndRewrite()
1341     if (initTensor.getType() != reshapeOp.getResultType()) {  in matchAndRewrite()
1343           reshapeOp, reshapeOp.getResultType(), initTensor);  in matchAndRewrite()
1345       rewriter.replaceOp(reshapeOp, initTensor);  in matchAndRewrite()