Lines Matching refs:allIvs
78 getInputAndOutputIndices(OpBuilder &b, Location loc, ArrayRef<Value> allIvs, in getInputAndOutputIndices() argument
84 makeCanonicalAffineApplies(b, loc, maps[0], allIvs), in getInputAndOutputIndices()
85 makeCanonicalAffineApplies(b, loc, maps[2], allIvs)}; in getInputAndOutputIndices()
120 ArrayRef<Value> allIvs, in emitScalarImplementation() argument
127 auto allIvsPlusDims = SmallVector<Value>(allIvs.begin(), allIvs.end()); in emitScalarImplementation()
170 SmallVector<Value> allIvs; in replaceIndexOpsByInductionVariables() local
174 allIvs.append(parallelOp.getInductionVars().begin(), in replaceIndexOpsByInductionVariables()
178 allIvs.push_back(forOp.getInductionVar()); in replaceIndexOpsByInductionVariables()
181 allIvs.push_back(affineForOp.getInductionVar()); in replaceIndexOpsByInductionVariables()
185 assert(linalgOp.getNumLoops() == allIvs.size() && in replaceIndexOpsByInductionVariables()
192 rewriter.replaceOp(indexOp, allIvs[indexOp.dim()]); in replaceIndexOpsByInductionVariables()
214 SmallVector<Value> allIvs; in linalgOpToLoopsImpl() local
221 allIvs.append(ivs.begin(), ivs.end()); in linalgOpToLoopsImpl()
222 emitScalarImplementation<LoadOpTy, StoreOpTy>(b, loc, allIvs, linalgOp); in linalgOpToLoopsImpl()
228 for (Value iv : allIvs) { in linalgOpToLoopsImpl()