Searched defs:all_reduce (Results 1 – 5 of 5) sorted by relevance
24 def all_reduce(t, group_size, group_key, instance_key, merge_op, final_op, function
348 const HloInstruction* all_reduce) in NcclAllReduceThunk()
324 auto all_reduce = pair.ar; in RewriteGraph() local
211 HloInstruction* all_reduce = in TEST_F() local
1475 def all_reduce(self, reduce_op, value): member in ReplicaContext