Searched defs:reduce_computation (Results 1 – 9 of 9) sorted by relevance
116 HloComputation* reduce_computation; in TEST_F() local
80 HloComputation* reduce_computation; in TEST_F() local
83 HloComputation* reduce_computation; in TEST_F() local
332 HloComputation* reduce_computation; in MakeReduceHlo() local358 HloComputation* reduce_computation; in MakeReduceHlo() local
1183 HloComputation* reduce_computation, in CreateAllReduce()1194 HloComputation* reduce_computation, in CreateReduceScatter()1206 HloComputation* reduce_computation, in CreateAllReduceStart()1466 HloComputation* reduce_computation) { in CreateReduce()1476 HloComputation* reduce_computation) { in CreateReduce()1487 const Window& window, HloComputation* reduce_computation) { in CreateReduceWindow()1495 HloComputation* reduce_computation) { in CreateReduceWindow()
198 auto reduce_computation = in TEST_F() local
674 HloComputation* reduce_computation, in HloAllReduceInstructionBase()735 HloComputation* reduce_computation, in HloReduceScatterInstruction()1004 HloComputation* reduce_computation) in HloReduceInstruction()2370 const Window& window, HloComputation* reduce_computation) in HloReduceWindowInstruction()2378 HloComputation* reduce_computation) in HloReduceWindowInstruction()
1708 optional<HloComputation*> reduce_computation; in ParseInstructionRhs() local1953 optional<HloComputation*> reduce_computation; in ParseInstructionRhs() local
974 auto reduce_computation = in TEST_F() local