Searched refs:batch_all_reduce (Results 1 – 3 of 3) sorted by relevance
3249 def batch_all_reduce(strategy, *value_flat): function3262 self.merge_call(batch_all_reduce, args=flattened_value))3266 ys = self.merge_call(batch_all_reduce, args=xs)
388 def batch_all_reduce( member in CollectiveReplicaLauncher
1157 dense_results = launcher.batch_all_reduce(packs, options)