Lines Matching full:indices
43 ArrayRef<Value> indices) { in rebuildLoadOp() argument
45 return rewriter.create<memref::LoadOp>(loc, srcMemRef, indices, in rebuildLoadOp()
73 ArrayRef<Value> indices) { in rebuildStoreOp() argument
76 srcMemRef, indices, in rebuildStoreOp()
105 ArrayRef<Value> indices) { in rebuildLdMatrixOp() argument
108 loc, ldMatrixOp.getResult().getType(), srcMemRef, indices, in rebuildLdMatrixOp()
133 ArrayRef<Value> indices) { in rebuildTransferReadOp() argument
136 loc, transferReadOp.getResult().getType(), srcMemRef, indices, in rebuildTransferReadOp()
151 ArrayRef<Value> indices) { in rebuildTransferWriteOp() argument
154 loc, transferWriteOp.getValue(), srcMemRef, indices, in rebuildTransferWriteOp()
190 SmallVector<OpFoldResult> indices = in getGenericOpViewSizeForEachDim() local
197 for (auto [srcSize, indice] : llvm::zip(srcSizes, indices)) { in getGenericOpViewSizeForEachDim()
204 /// Rewrite a store/load-like op so that all its indices are zeros.
220 /// LoadStoreLikeOp that reads from srcMemRef[indices].
226 Value /*srcMemRef*/, ArrayRef<Value> /*indices*/),
250 // If our load already has only zeros as indices there is nothing in matchAndRewrite()
252 SmallVector<OpFoldResult> indices = in matchAndRewrite() local
254 if (std::all_of(indices.begin(), indices.end(), in matchAndRewrite()
270 // apply them properly to the input indices. in matchAndRewrite()
274 /*offsets=*/indices, in matchAndRewrite()