Searched refs:_all_gather_tensor (Results 1 – 1 of 1) sorted by relevance
1732 def _all_gather_tensor(value, axis): function1823 ys = [_all_gather_tensor(t, axis=axis) for t in nest.flatten(value)]