Lines Matching refs:subViewOp
45 memref::SubViewOp subViewOp, ValueRange indices, in resolveSourceIndices() argument
47 SmallVector<OpFoldResult> mixedOffsets = subViewOp.getMixedOffsets(); in resolveSourceIndices()
48 SmallVector<OpFoldResult> mixedSizes = subViewOp.getMixedSizes(); in resolveSourceIndices()
49 SmallVector<OpFoldResult> mixedStrides = subViewOp.getMixedStrides(); in resolveSourceIndices()
55 llvm::SmallBitVector unusedDims = subViewOp.getDroppedDims(); in resolveSourceIndices()
56 for (auto dim : llvm::seq<unsigned>(0, subViewOp.getSourceType().getRank())) { in resolveSourceIndices()
86 Location loc = subViewOp.getLoc(); in resolveSourceIndices()
111 memref::SubViewOp subViewOp, in getPermutationMapAttr() argument
113 llvm::SmallBitVector unusedDims = subViewOp.getDroppedDims(); in getPermutationMapAttr()
115 int64_t sourceRank = subViewOp.getSourceType().getRank(); in getPermutationMapAttr()
141 void replaceOp(OpTy loadOp, memref::SubViewOp subViewOp,
156 void replaceOp(OpTy storeOp, memref::SubViewOp subViewOp,
163 LoadOpTy loadOp, memref::SubViewOp subViewOp, ArrayRef<Value> sourceIndices, in replaceOp() argument
165 rewriter.replaceOpWithNewOp<LoadOpTy>(loadOp, subViewOp.getSource(), in replaceOp()
171 vector::TransferReadOp transferReadOp, memref::SubViewOp subViewOp, in replaceOp() argument
177 transferReadOp, transferReadOp.getVectorType(), subViewOp.getSource(), in replaceOp()
179 getPermutationMapAttr(rewriter.getContext(), subViewOp, in replaceOp()
187 StoreOpTy storeOp, memref::SubViewOp subViewOp, in replaceOp() argument
190 subViewOp.getSource(), sourceIndices); in replaceOp()
195 vector::TransferWriteOp transferWriteOp, memref::SubViewOp subViewOp, in replaceOp() argument
201 transferWriteOp, transferWriteOp.getVector(), subViewOp.getSource(), in replaceOp()
203 getPermutationMapAttr(rewriter.getContext(), subViewOp, in replaceOp()
213 auto subViewOp = in matchAndRewrite() local
215 if (!subViewOp) in matchAndRewrite()
219 if (failed(resolveSourceIndices(loadOp.getLoc(), rewriter, subViewOp, in matchAndRewrite()
223 replaceOp(loadOp, subViewOp, sourceIndices, rewriter); in matchAndRewrite()
231 auto subViewOp = in matchAndRewrite() local
233 if (!subViewOp) in matchAndRewrite()
237 if (failed(resolveSourceIndices(storeOp.getLoc(), rewriter, subViewOp, in matchAndRewrite()
241 replaceOp(storeOp, subViewOp, sourceIndices, rewriter); in matchAndRewrite()