Searched refs:MakeReduction (Results 1 – 1 of 1) sorted by relevance
86 HloComputation* MakeReduction(const HloOpcode type, HloModule* module) { in MakeReduction() function116 HloComputation* sum = MakeReduction(HloOpcode::kAdd, module.get()); in TEST_F()160 HloComputation* sum = MakeReduction(HloOpcode::kAdd, module.get()); in TEST_F()161 HloComputation* min = MakeReduction(HloOpcode::kMinimum, module.get()); in TEST_F()162 HloComputation* max = MakeReduction(HloOpcode::kMaximum, module.get()); in TEST_F()163 HloComputation* sum_2 = MakeReduction(HloOpcode::kAdd, module.get()); in TEST_F()184 HloComputation* sum = MakeReduction(HloOpcode::kAdd, module.get()); in TEST_F()216 HloComputation* reduction = MakeReduction(HloOpcode::kAdd, module.get()); in TEST_F()243 HloComputation* reduction = MakeReduction(HloOpcode::kAdd, module.get()); in TEST_F()