Searched refs:_all_reduce_sum_fn (Results 1 – 1 of 1) sorted by relevance
46 _all_reduce_sum_fn, args=(filtered_grads_and_vars,))150 def _all_reduce_sum_fn(distribution, grads_and_vars): function