Lines Matching refs:insertSliceOp

1596   LogicalResult matchAndRewrite(InsertOpTy insertSliceOp,  in matchAndRewrite()  argument
1599 if (llvm::none_of(insertSliceOp.getOperands(), [](Value operand) { in matchAndRewrite()
1607 SmallVector<OpFoldResult> mixedOffsets(insertSliceOp.getMixedOffsets()); in matchAndRewrite()
1608 SmallVector<OpFoldResult> mixedSizes(insertSliceOp.getMixedSizes()); in matchAndRewrite()
1609 SmallVector<OpFoldResult> mixedStrides(insertSliceOp.getMixedStrides()); in matchAndRewrite()
1616 insertSliceOp.getSourceType().getRank(), insertSliceOp.getDestType(), in matchAndRewrite()
1618 Value toInsert = insertSliceOp.getSource(); in matchAndRewrite()
1619 if (sourceType != insertSliceOp.getSourceType()) { in matchAndRewrite()
1625 rewriter.setInsertionPoint(insertSliceOp->getParentOp()); in matchAndRewrite()
1626 toInsert = rewriter.create<tensor::CastOp>(insertSliceOp.getLoc(), in matchAndRewrite()
1630 insertSliceOp, toInsert, insertSliceOp.getDest(), mixedOffsets, in matchAndRewrite()
1660 LogicalResult matchAndRewrite(InsertOpTy insertSliceOp, in matchAndRewrite()
1662 if (llvm::any_of(insertSliceOp.getOperands(), [](Value operand) { in matchAndRewrite()
1674 getSourceOfCastOp(insertSliceOp.getSource()); in matchAndRewrite()
1675 Optional<Value> destCastSource = getSourceOfCastOp(insertSliceOp.getDest()); in matchAndRewrite()
1680 (sourceCastSource ? *sourceCastSource : insertSliceOp.getSource()); in matchAndRewrite()
1681 auto dst = (destCastSource ? *destCastSource : insertSliceOp.getDest()); in matchAndRewrite()
1684 if (verifyInsertSliceOp(srcType, dstType, insertSliceOp.getStaticOffsets(), in matchAndRewrite()
1685 insertSliceOp.getStaticSizes(), in matchAndRewrite()
1686 insertSliceOp.getStaticStrides()) != in matchAndRewrite()
1691 insertSliceOp.getLoc(), src, dst, insertSliceOp.getMixedOffsets(), in matchAndRewrite()
1692 insertSliceOp.getMixedSizes(), insertSliceOp.getMixedStrides()); in matchAndRewrite()
1697 if (!isParallelInsert && dst.getType() != insertSliceOp.getDestType()) { in matchAndRewrite()
1698 replacement = rewriter.create<tensor::CastOp>(insertSliceOp.getLoc(), in matchAndRewrite()
1699 insertSliceOp.getDestType(), in matchAndRewrite()
1702 rewriter.replaceOp(insertSliceOp, replacement->getResults()); in matchAndRewrite()
1733 LogicalResult matchAndRewrite(InsertOpTy insertSliceOp, in matchAndRewrite()
1735 RankedTensorType srcType = insertSliceOp.getSourceType(); in matchAndRewrite()
1736 if (srcType.getRank() != insertSliceOp.getDestType().getRank()) in matchAndRewrite()
1742 getConstantIntValue(insertSliceOp.getMixedSizes()[i])) in matchAndRewrite()
1763 rewriter.setInsertionPoint(insertSliceOp->getParentOp()); in matchAndRewrite()
1765 insertSliceOp.getLoc(), newSrcType, insertSliceOp.getSource()); in matchAndRewrite()
1767 insertSliceOp, cast, insertSliceOp.getDest(), in matchAndRewrite()
1768 insertSliceOp.getMixedOffsets(), insertSliceOp.getMixedSizes(), in matchAndRewrite()
1769 insertSliceOp.getMixedStrides()); in matchAndRewrite()
2302 LogicalResult matchAndRewrite(ParallelInsertSliceOp insertSliceOp, in matchAndRewrite() argument
2305 if (llvm::none_of(insertSliceOp.getOperands(), [](Value operand) { in matchAndRewrite()
2313 SmallVector<OpFoldResult> mixedOffsets(insertSliceOp.getMixedOffsets()); in matchAndRewrite()
2314 SmallVector<OpFoldResult> mixedSizes(insertSliceOp.getMixedSizes()); in matchAndRewrite()
2315 SmallVector<OpFoldResult> mixedStrides(insertSliceOp.getMixedStrides()); in matchAndRewrite()
2323 insertSliceOp.getSourceType().getRank(), in matchAndRewrite()
2324 insertSliceOp.getDestType(), mixedOffsets, mixedSizes, in matchAndRewrite()
2326 Value toInsert = insertSliceOp.getSource(); in matchAndRewrite()
2327 if (sourceType != insertSliceOp.getSourceType()) { in matchAndRewrite()
2329 rewriter.setInsertionPoint(insertSliceOp->getParentOp()); in matchAndRewrite()
2330 toInsert = rewriter.create<tensor::CastOp>(insertSliceOp.getLoc(), in matchAndRewrite()
2334 insertSliceOp, toInsert, insertSliceOp.getDest(), mixedOffsets, in matchAndRewrite()