Searched defs:reduce_computation (Results 1 – 9 of 9) sorted by relevance
83 HloComputation* reduce_computation; in TEST_F() local
116 HloComputation* reduce_computation; in TEST_F() local
80 HloComputation* reduce_computation; in TEST_F() local
330 HloComputation* reduce_computation; in MakeReduceHlo() local356 HloComputation* reduce_computation; in MakeReduceHlo() local
198 auto reduce_computation = in TEST_F() local
1054 HloComputation* reduce_computation, in CreateAllReduce()1289 HloComputation* reduce_computation) { in CreateReduce()1299 HloComputation* reduce_computation) { in CreateReduce()1310 const Window& window, HloComputation* reduce_computation) { in CreateReduceWindow()1318 HloComputation* reduce_computation) { in CreateReduceWindow()
667 HloComputation* reduce_computation, in HloAllReduceInstruction()904 HloComputation* reduce_computation) in HloReduceInstruction()2232 const Window& window, HloComputation* reduce_computation) in HloReduceWindowInstruction()2240 HloComputation* reduce_computation) in HloReduceWindowInstruction()
1571 optional<HloComputation*> reduce_computation; in ParseInstructionRhs() local1816 optional<HloComputation*> reduce_computation; in ParseInstructionRhs() local
972 auto reduce_computation = in TEST_F() local