Lines Matching refs:sliceOp
50 LogicalResult matchAndRewrite(tensor::ExtractSliceOp sliceOp, in matchAndRewrite()
52 Value source = sliceOp.getSource(); in matchAndRewrite()
55 return rewriter.notifyMatchFailure(sliceOp, in matchAndRewrite()
62 return rewriter.notifyMatchFailure(sliceOp, in matchAndRewrite()
67 return rewriter.notifyMatchFailure(sliceOp, in matchAndRewrite()
72 return rewriter.notifyMatchFailure(sliceOp, in matchAndRewrite()
76 if (!sliceOp.hasUnitStride()) in matchAndRewrite()
77 return rewriter.notifyMatchFailure(sliceOp, "expected unit stride"); in matchAndRewrite()
79 if (sliceOp.getType().getRank() != sliceOp.getSourceType().getRank()) { in matchAndRewrite()
80 return rewriter.notifyMatchFailure(sliceOp, "expected no rank reduction"); in matchAndRewrite()
87 sliceOp, "expected a projected permutation for output"); in matchAndRewrite()
101 auto sliceLoc = sliceOp.getLoc(); in matchAndRewrite()
103 rewriter, sliceLoc, sliceOp.getMixedOffsets()); in matchAndRewrite()
105 sliceOp.getMixedSizes()); in matchAndRewrite()
133 rewriter.replaceOp(sliceOp, newOp->getResults()); in matchAndRewrite()