Lines Matching defs:insertSliceOp

2795   LogicalResult matchAndRewrite(InsertOpTy insertSliceOp,
2797 SmallVector<OpFoldResult> mixedOffsets(insertSliceOp.getMixedOffsets());
2798 SmallVector<OpFoldResult> mixedSizes(insertSliceOp.getMixedSizes());
2799 SmallVector<OpFoldResult> mixedStrides(insertSliceOp.getMixedStrides());
2809 insertSliceOp.getSourceType().getRank(), insertSliceOp.getDestType(),
2811 Value toInsert = insertSliceOp.getSource();
2812 if (sourceType != insertSliceOp.getSourceType()) {
2818 rewriter.setInsertionPoint(insertSliceOp->getParentOp());
2819 toInsert = rewriter.create<tensor::CastOp>(insertSliceOp.getLoc(),
2823 insertSliceOp, toInsert, insertSliceOp.getDest(), mixedOffsets,
2853 LogicalResult matchAndRewrite(InsertOpTy insertSliceOp,
2855 if (llvm::any_of(insertSliceOp.getOperands(), [](Value operand) {
2867 getSourceOfCastOp(insertSliceOp.getSource());
2869 getSourceOfCastOp(insertSliceOp.getDest());
2874 (sourceCastSource ? *sourceCastSource : insertSliceOp.getSource());
2875 auto dst = (destCastSource ? *destCastSource : insertSliceOp.getDest());
2884 SmallVector<int64_t> staticSizes(insertSliceOp.getStaticSizes());
2894 SmallVector<OpFoldResult> mixedSizes(insertSliceOp.getMixedSizes());
2904 if (verifyInsertSliceOp(srcType, dstType, insertSliceOp.getStaticOffsets(),
2905 staticSizes, insertSliceOp.getStaticStrides()) !=
2910 insertSliceOp.getLoc(), src, dst, insertSliceOp.getMixedOffsets(),
2911 mixedSizes, insertSliceOp.getMixedStrides());
2916 if (!isParallelInsert && dst.getType() != insertSliceOp.getDestType()) {
2917 replacement = rewriter.create<tensor::CastOp>(insertSliceOp.getLoc(),
2918 insertSliceOp.getDestType(),
2921 rewriter.replaceOp(insertSliceOp, replacement->getResults());
2952 LogicalResult matchAndRewrite(InsertOpTy insertSliceOp,
2954 RankedTensorType srcType = insertSliceOp.getSourceType();
2955 if (srcType.getRank() != insertSliceOp.getDestType().getRank())
2960 getConstantIntValue(insertSliceOp.getMixedSizes()[i])) {
2987 rewriter.setInsertionPoint(insertSliceOp->getParentOp());
2989 insertSliceOp.getLoc(), newSrcType, insertSliceOp.getSource());
2991 insertSliceOp, cast, insertSliceOp.getDest(),
2992 insertSliceOp.getMixedOffsets(), insertSliceOp.getMixedSizes(),
2993 insertSliceOp.getMixedStrides());