Lines Matching refs:extractStridedSliceOp
1409 auto extractStridedSliceOp = in foldExtractFromExtractStrided() local
1411 if (!extractStridedSliceOp) in foldExtractFromExtractStrided()
1414 if (extractStridedSliceOp.hasNonUnitStrides()) in foldExtractFromExtractStrided()
1419 extractVector<int64_t>(extractStridedSliceOp.getOffsets()); in foldExtractFromExtractStrided()
1423 extractStridedSliceOp.getType().getDimSize(lastOffset) != in foldExtractFromExtractStrided()
1424 extractStridedSliceOp.getVectorType().getDimSize(lastOffset)) in foldExtractFromExtractStrided()
1434 extractStridedSliceOp.getVectorType().getRank() - sliceOffsets.size()) in foldExtractFromExtractStrided()
1440 extractOp.getVectorMutable().assign(extractStridedSliceOp.getVector()); in foldExtractFromExtractStrided()
2264 auto extractStridedSliceOp = in matchAndRewrite() local
2268 if (!extractStridedSliceOp) in matchAndRewrite()
2271 if (extractStridedSliceOp.getOperand() != insertStridedSliceOp.getDest()) in matchAndRewrite()
2275 if (extractStridedSliceOp.getStrides() != in matchAndRewrite()
2277 extractStridedSliceOp.getOffsets() != insertStridedSliceOp.getOffsets()) in matchAndRewrite()
2622 LogicalResult matchAndRewrite(ExtractStridedSliceOp extractStridedSliceOp, in matchAndRewrite() argument
2626 auto *defOp = extractStridedSliceOp.getVector().getDefiningOp(); in matchAndRewrite()
2631 if (extractStridedSliceOp.hasNonUnitStrides()) in matchAndRewrite()
2638 populateFromInt64AttrArray(extractStridedSliceOp.getOffsets(), in matchAndRewrite()
2641 populateFromInt64AttrArray(extractStridedSliceOp.getSizes(), sliceSizes); in matchAndRewrite()
2663 extractStridedSliceOp, extractStridedSliceOp.getResult().getType(), in matchAndRewrite()
2675 LogicalResult matchAndRewrite(ExtractStridedSliceOp extractStridedSliceOp, in matchAndRewrite() argument
2680 extractStridedSliceOp.getVector().getDefiningOp<arith::ConstantOp>(); in matchAndRewrite()
2686 auto newAttr = DenseElementsAttr::get(extractStridedSliceOp.getType(), in matchAndRewrite()
2688 rewriter.replaceOpWithNewOp<arith::ConstantOp>(extractStridedSliceOp, in matchAndRewrite()