Searched refs:reduce_instr (Results 1 – 1 of 1) sorted by relevance
2510 auto reduce_instr = Cast<HloReduceInstruction>(hlo); in MakeElementGenerator() local2512 for (const HloInstruction* instr : reduce_instr->inputs()) { in MakeElementGenerator()2517 for (const HloInstruction* instr : reduce_instr->init_values()) { in MakeElementGenerator()2520 return EmitElementalReduce(reduce_instr, std::move(input_generators), in MakeElementGenerator()