Lines Matching refs:writeOp
242 LogicalResult matchAndRewrite(vector::TransferWriteOp writeOp, in matchAndRewrite()
245 if (writeOp.getTransferRank() == 0) in matchAndRewrite()
248 if (writeOp.getMask()) in matchAndRewrite()
250 auto targetShape = getTargetShape(options, writeOp); in matchAndRewrite()
253 auto sourceVectorType = writeOp.getVectorType(); in matchAndRewrite()
255 Location loc = writeOp.getLoc(); in matchAndRewrite()
257 SmallVector<Value, 4> originalIndices(writeOp.getIndices().begin(), in matchAndRewrite()
258 writeOp.getIndices().end()); in matchAndRewrite()
261 getUnrollOrder(originalSize.size(), writeOp, options); in matchAndRewrite()
269 loc, writeOp.getVector(), elementOffsets, *targetShape, strides); in matchAndRewrite()
272 writeOp.getPermutationMap(), loc, rewriter); in matchAndRewrite()
274 loc, slicedVector, resultTensor ? resultTensor : writeOp.getSource(), in matchAndRewrite()
275 indices, writeOp.getPermutationMapAttr(), writeOp.getInBoundsAttr()); in matchAndRewrite()
281 rewriter.replaceOp(writeOp, resultTensor); in matchAndRewrite()
283 rewriter.eraseOp(writeOp); in matchAndRewrite()