Home
last modified time | relevance | path

Searched refs:insertSliceOp (Results 1 – 7 of 7) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DHoisting.cpp48 tensor::InsertSliceOp insertSliceOp; member
106 if (write.insertSliceOp) in findMatchingTransferRead()
118 if (write.insertSliceOp) { in findMatchingTransferRead()
254 return HoistableWrite{write, insertSliceOp}; in getLoopInvariantTransferWriteOpDefining()
266 assert(((read.extractSliceOp && write.insertSliceOp) || in hoistReadWrite()
295 if (write.insertSliceOp) in hoistReadWrite()
296 write.insertSliceOp->moveAfter(forOp); in hoistReadWrite()
301 if (write.insertSliceOp) in hoistReadWrite()
320 if (write.insertSliceOp) { in hoistReadWrite()
325 write.insertSliceOp.getDestMutable().assign( in hoistReadWrite()
[all …]
H A DDropUnitDims.cpp495 LogicalResult matchAndRewrite(InsertOpTy insertSliceOp, in matchAndRewrite()
497 RankedTensorType sourceType = insertSliceOp.getSourceType(); in matchAndRewrite()
498 SmallVector<OpFoldResult> offsets = insertSliceOp.getMixedOffsets(); in matchAndRewrite()
499 SmallVector<OpFoldResult> sizes = insertSliceOp.getMixedSizes(); in matchAndRewrite()
500 SmallVector<OpFoldResult> strides = insertSliceOp.getMixedStrides(); in matchAndRewrite()
505 Location loc = insertSliceOp.getLoc(); in matchAndRewrite()
513 rewriter.setInsertionPoint(insertSliceOp->getParentOp()); in matchAndRewrite()
515 loc, insertSliceOp.getSource(), *reassociation); in matchAndRewrite()
518 insertSliceOp, reshapedSource, insertSliceOp.getDest(), in matchAndRewrite()
519 insertSliceOp.getMixedOffsets(), insertSliceOp.getMixedSizes(), in matchAndRewrite()
[all …]
H A DFusionOnTensors.cpp257 if (auto insertSliceOp = dyn_cast<tensor::InsertSliceOp>(op)) { in hasOtherUses() local
259 getBackwardSlice(insertSliceOp.getSource(), &backwardSlice, in hasOtherUses()
/llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/Transforms/
H A DAllocTensorElimination.cpp197 auto insertSliceOp = in insertSliceAnchoredAllocTensorEliminationStep() local
199 if (!insertSliceOp) in insertSliceAnchoredAllocTensorEliminationStep()
201 if (&operand != &insertSliceOp->getOpOperand(0) /*source*/) in insertSliceAnchoredAllocTensorEliminationStep()
205 neededValues.append(insertSliceOp.getOffsets().begin(), in insertSliceAnchoredAllocTensorEliminationStep()
206 insertSliceOp.getOffsets().end()); in insertSliceAnchoredAllocTensorEliminationStep()
207 neededValues.append(insertSliceOp.getSizes().begin(), in insertSliceAnchoredAllocTensorEliminationStep()
208 insertSliceOp.getSizes().end()); in insertSliceAnchoredAllocTensorEliminationStep()
209 neededValues.append(insertSliceOp.getStrides().begin(), in insertSliceAnchoredAllocTensorEliminationStep()
210 insertSliceOp.getStrides().end()); in insertSliceAnchoredAllocTensorEliminationStep()
211 neededValues.push_back(insertSliceOp.getDest()); in insertSliceAnchoredAllocTensorEliminationStep()
/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/Transforms/
H A DBufferizableOpInterfaceImpl.cpp558 OpTy insertSliceOp) { in areEquivalentExtractSliceOps() argument
559 if (!extractSliceOp || !insertSliceOp) in areEquivalentExtractSliceOps()
561 if (extractSliceOp != insertSliceOp && in areEquivalentExtractSliceOps()
575 OpTy insertSliceOp) { in hasMatchingExtractSliceOp() argument
596 if (auto insertSliceOp = dyn_cast<OpTy>(readingOp)) { in isNotConflictingInsertSliceLikeOp() local
605 if (uRead == &insertSliceOp->getOpOperand(1) /*dest*/ && in isNotConflictingInsertSliceLikeOp()
607 insertSliceOp)) in isNotConflictingInsertSliceLikeOp()
622 if (uRead == &insertSliceOp->getOpOperand(0) /*source*/ && in isNotConflictingInsertSliceLikeOp()
658 insertSliceOp)) in isNotConflictingInsertSliceLikeOp()
705 auto insertSliceOp = cast<tensor::InsertSliceOp>(op); in bufferize() local
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/
H A DTensorOps.cpp1616 insertSliceOp.getSourceType().getRank(), insertSliceOp.getDestType(), in matchAndRewrite()
1618 Value toInsert = insertSliceOp.getSource(); in matchAndRewrite()
1630 insertSliceOp, toInsert, insertSliceOp.getDest(), mixedOffsets, in matchAndRewrite()
1691 insertSliceOp.getLoc(), src, dst, insertSliceOp.getMixedOffsets(), in matchAndRewrite()
1692 insertSliceOp.getMixedSizes(), insertSliceOp.getMixedStrides()); 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()
2326 Value toInsert = insertSliceOp.getSource(); in matchAndRewrite()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/SCF/IR/
H A DSCF.cpp1270 auto insertSliceOp = dyn_cast<tensor::ParallelInsertSliceOp>(&op); in getYieldedTypes() local
1271 return insertSliceOp ? insertSliceOp.yieldedType() : Type(); in getYieldedTypes()