Lines Matching defs:reductionIndex

184 /// value `initValue`, and the `reductionIndex`-th reduction combiner carried
188            scf::ReduceOp reduce, int64_t reductionIndex, Attribute initValue) {
190   Type type = reduce.getOperands()[reductionIndex].getType();
197                       {reduce.getOperands()[reductionIndex].getLoc()});
204       &reduce.getReductions()[reductionIndex].front().back();
210   builder.inlineRegionBefore(reduce.getReductions()[reductionIndex],
222                                             int64_t reductionIndex) {
225   Location reduceOperandLoc = reduce.getOperands()[reductionIndex].getLoc();
246                                                 int64_t reductionIndex) {
258   assert(llvm::hasSingleElement(reduce.getReductions()[reductionIndex]) &&
262   Type type = reduce.getOperands()[reductionIndex].getType();
263   Block &reduction = reduce.getReductions()[reductionIndex].front();
266         createDecl(builder, symbolTable, reduce, reductionIndex,
269                         reductionIndex);
273         createDecl(builder, symbolTable, reduce, reductionIndex,
276                         reductionIndex);
280         createDecl(builder, symbolTable, reduce, reductionIndex,
283                         reductionIndex);
287         createDecl(builder, symbolTable, reduce, reductionIndex,
290                         reductionIndex);
294         builder, symbolTable, reduce, reductionIndex,
298                         reductionIndex);
305     return createDecl(builder, symbolTable, reduce, reductionIndex,
309     return createDecl(builder, symbolTable, reduce, reductionIndex,
321     return createDecl(builder, symbolTable, reduce, reductionIndex,
331         createDecl(builder, symbolTable, reduce, reductionIndex,
335                         decl, reduce, reductionIndex);
344         createDecl(builder, symbolTable, reduce, reductionIndex,
348         decl, reduce, reductionIndex);