Lines Matching refs:sliceOp
363 if (auto sliceOp = dyn_cast_or_null<tensor::ExtractSliceOp>(definingOp)) { in fold() local
366 if (sliceOp.getType().getRank() == sliceOp.getSourceType().getRank() && in fold()
367 sliceOp.isDynamicSize(unsignedIndex)) { in fold()
368 return {sliceOp.getDynamicSize(unsignedIndex)}; in fold()
1192 LogicalResult matchAndRewrite(ExtractSliceOp sliceOp, in matchAndRewrite() argument
1195 if (llvm::any_of(sliceOp.getOperands(), [](Value operand) { in matchAndRewrite()
1200 auto castOp = sliceOp.getSource().getDefiningOp<tensor::CastOp>(); in matchAndRewrite()
1210 sliceOp.getType().getRank(), sliceOp.getSourceType(), in matchAndRewrite()
1211 sliceOp.getMixedOffsets(), sliceOp.getMixedSizes(), in matchAndRewrite()
1212 sliceOp.getMixedStrides()); in matchAndRewrite()
1214 sliceOp.getLoc(), resultType, castOp.getSource(), sliceOp.getOffsets(), in matchAndRewrite()
1215 sliceOp.getSizes(), sliceOp.getStrides(), sliceOp.getStaticOffsets(), in matchAndRewrite()
1216 sliceOp.getStaticSizes(), sliceOp.getStaticStrides()); in matchAndRewrite()
1217 rewriter.replaceOpWithNewOp<tensor::CastOp>(sliceOp, sliceOp.getType(), in matchAndRewrite()