Searched defs:reduce_computation (Results 1 – 7 of 7) sorted by relevance
101 HloComputation* reduce_computation; in TEST_F() local
196 auto reduce_computation = in TEST_F() local
259 HloComputation* reduce_computation; in MakeReduceHlo() local
834 HloComputation* reduce_computation, in CreateAllReduce()1035 HloComputation* reduce_computation) { in CreateReduce()1045 HloComputation* reduce_computation) { in CreateReduce()1056 const Window& window, HloComputation* reduce_computation) { in CreateReduceWindow()
520 HloComputation* reduce_computation, in HloAllReduceInstruction()730 HloComputation* reduce_computation) in HloReduceInstruction()1954 const Window& window, HloComputation* reduce_computation) in HloReduceWindowInstruction()
1047 optional<HloComputation*> reduce_computation; in ParseInstructionRhs() local1196 optional<HloComputation*> reduce_computation; in ParseInstructionRhs() local
800 auto reduce_computation = in TEST_F() local