Searched refs:_get_reduce_scatter_tensors (Results 1 – 1 of 1) sorted by relevance
837 padded_unsharded_grad, new_sharded_grad = _get_reduce_scatter_tensors(879 def _get_reduce_scatter_tensors( function