Lines Matching refs:insertSliceOp
48 tensor::InsertSliceOp insertSliceOp; member
106 if (write.insertSliceOp) in findMatchingTransferRead()
108 << *write.insertSliceOp.getOperation() << "\n"); in findMatchingTransferRead()
118 if (write.insertSliceOp) { in findMatchingTransferRead()
121 write.insertSliceOp.getSource().getType()) in findMatchingTransferRead()
125 << *sliceOp << " vs " << *write.insertSliceOp << "\n"); in findMatchingTransferRead()
126 if (!sameOffsetsSizesAndStrides(sliceOp, write.insertSliceOp)) in findMatchingTransferRead()
175 user == write.transferWriteOp || user == write.insertSliceOp) in tensorChunkAccessedByUnknownOp()
235 if (auto insertSliceOp = v.getDefiningOp<tensor::InsertSliceOp>()) { in getLoopInvariantTransferWriteOpDefining() local
238 insertSliceOp.getSource().getDefiningOp<vector::TransferWriteOp>(); in getLoopInvariantTransferWriteOpDefining()
243 auto bbArg = insertSliceOp.getDest().dyn_cast<BlockArgument>(); in getLoopInvariantTransferWriteOpDefining()
249 for (Value operand : insertSliceOp->getOperands().drop_front( in getLoopInvariantTransferWriteOpDefining()
254 return HoistableWrite{write, insertSliceOp}; in getLoopInvariantTransferWriteOpDefining()
266 assert(((read.extractSliceOp && write.insertSliceOp) || in hoistReadWrite()
267 (!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()
302 yieldOp->setOperand(initArgNumber, write.insertSliceOp.getDest()); in hoistReadWrite()
320 if (write.insertSliceOp) { in hoistReadWrite()
322 .replaceAllUsesWith(write.insertSliceOp.getResult()); in hoistReadWrite()
325 write.insertSliceOp.getDestMutable().assign( in hoistReadWrite()
364 if (write.insertSliceOp) in hoistRedundantVectorTransfersOnTensor()
366 << *write.insertSliceOp.getOperation() << "\n"); in hoistRedundantVectorTransfersOnTensor()