Searched refs:batch_all_reduce (Results 1 – 1 of 1) sorted by relevance
1501 def batch_all_reduce(strategy, *value_flat): function1509 ys = self.merge_call(batch_all_reduce, args=xs)1516 value, self.merge_call(batch_all_reduce, args=nest.flatten(value)))