Lines Matching defs:collapseShapeOp

57   LogicalResult matchAndRewrite(tensor::CollapseShapeOp collapseShapeOp,
60         collapseShapeOp.getSrc().getDefiningOp<tensor::ExtractSliceOp>();
70         collapseShapeOp.getSrcType(), collapseShapeOp.getResultType());
72       return rewriter.notifyMatchFailure(collapseShapeOp,
76         extractSliceOp.getLoc(), collapseShapeOp.getResultType(),
79     rewriter.replaceOp(collapseShapeOp, unPaddedExtractSlice);
91     auto collapseShapeOp =
93     if (!collapseShapeOp)
95     RankedTensorType srcType = collapseShapeOp.getSrcType();
109     rewriter.replaceOpWithNewOp<OpTy>(insertSliceOp, collapseShapeOp.getSrc(),