Lines Matching defs:allIvs
87 getInputAndOutputIndices(OpBuilder &b, Location loc, ArrayRef<Value> allIvs,
93 makeCanonicalAffineApplies(b, loc, maps[0], allIvs),
94 makeCanonicalAffineApplies(b, loc, maps[2], allIvs)};
129 ArrayRef<Value> allIvs,
136 auto allIvsPlusDims = SmallVector<Value>(allIvs);
183 SmallVector<Value> allIvs;
187 allIvs.append(parallelOp.getInductionVars());
190 allIvs.push_back(forOp.getInductionVar());
193 allIvs.push_back(affineForOp.getInductionVar());
197 assert(linalgOp.getNumLoops() == allIvs.size() &&
204 rewriter.replaceOp(indexOp, allIvs[indexOp.getDim()]);
226 SmallVector<Value> allIvs;
233 allIvs.append(ivs.begin(), ivs.end());
234 emitScalarImplementation<LoadOpTy, StoreOpTy>(b, loc, allIvs, linalgOp);
240 for (Value iv : allIvs) {