Searched defs:reduce_computation (Results 1 – 7 of 7) sorted by relevance
116 HloComputation* reduce_computation; in TEST_F() local
198 auto reduce_computation = in TEST_F() local
912 HloComputation* reduce_computation, in CreateAllReduce()1130 HloComputation* reduce_computation) { in CreateReduce()1140 HloComputation* reduce_computation) { in CreateReduce()1151 const Window& window, HloComputation* reduce_computation) { in CreateReduceWindow()
306 HloComputation* reduce_computation; in MakeReduceHlo() local
555 HloComputation* reduce_computation, in HloAllReduceInstruction()784 HloComputation* reduce_computation) in HloReduceInstruction()2077 const Window& window, HloComputation* reduce_computation) in HloReduceWindowInstruction()
1109 optional<HloComputation*> reduce_computation; in ParseInstructionRhs() local1258 optional<HloComputation*> reduce_computation; in ParseInstructionRhs() local
979 auto reduce_computation = in TEST_F() local