Lines Matching refs:readOp
3155 static Value foldRAW(TransferReadOp readOp) { in foldRAW() argument
3156 if (!readOp.getShapedType().isa<RankedTensorType>()) in foldRAW()
3158 auto defWrite = readOp.getSource().getDefiningOp<vector::TransferWriteOp>(); in foldRAW()
3160 if (checkSameValueRAW(defWrite, readOp)) in foldRAW()
3164 cast<VectorTransferOpInterface>(readOp.getOperation()))) in foldRAW()
3314 LogicalResult matchAndRewrite(TransferReadOp readOp, in matchAndRewrite()
3316 if (readOp.hasOutOfBoundsDim() || in matchAndRewrite()
3317 !readOp.getShapedType().isa<RankedTensorType>()) in matchAndRewrite()
3319 auto defWrite = readOp.getSource().getDefiningOp<vector::TransferWriteOp>(); in matchAndRewrite()
3323 SmallVector<int64_t> readDims = readOp.getTransferChunkAccessed(); in matchAndRewrite()
3325 if (readOp.getIndices() == defWrite.getIndices() && in matchAndRewrite()
3326 readOp.getMask() == defWrite.getMask()) { in matchAndRewrite()
3339 AffineMap readMap = compressUnusedDims(readOp.getPermutationMap()); in matchAndRewrite()
3349 Location loc = readOp.getLoc(); in matchAndRewrite()
3352 ArrayRef<int64_t> destShape = readOp.getVectorType().getShape(); in matchAndRewrite()
3360 rewriter.replaceOpWithNewOp<vector::TransposeOp>(readOp, vec, in matchAndRewrite()