Searched refs:batch_all_gather (Results 1 – 1 of 1) sorted by relevance
3431 def batch_all_gather(strategy, *value_flat): function3438 ys = self.merge_call(batch_all_gather, args=xs)