Searched refs:run_all_reduce (Results 1 – 3 of 3) sorted by relevance
254 def run_all_reduce(group_key, instance_key, merge_op): function272 results = run_all_reduce(group_key=10, instance_key=20, merge_op=merge_op)
459 def run_all_reduce(): function478 run_all_reduce()
361 def run_all_reduce(group_size, group_key): function378 for reduced_tensor in run_all_reduce(group_size, group_key):