Searched defs:reduce_inst (Results 1 – 1 of 1) sorted by relevance
2643 HloInstruction* unnested_hlo, HloInstruction* reduce_inst, int reduce_idx, in EmitPrologueForOneReduction()2716 HloInstruction* reduce_inst = output_instructions[i]; in EmitPrologueForReduction() local