Searched refs:reduction_comp (Results 1 – 2 of 2) sorted by relevance
361 const HloComputation* reduction_comp);
1736 const HloComputation* reduction_comp) { in ParseReductionComputation() argument1737 if (reduction_comp->num_parameters() != 2) { in ParseReductionComputation()1740 auto root = reduction_comp->root_instruction(); in ParseReductionComputation()1745 reduction_comp->parameter_instruction(0)) || in ParseReductionComputation()1747 reduction_comp->parameter_instruction(1))) { in ParseReductionComputation()