Searched defs:patch_reduce_scatter (Results 1 – 3 of 3) sorted by relevance
68 def patch_reduce_scatter(new_reduce_scatter): function
144 def patch_reduce_scatter(new_reduce_scatter, full_precision_param_dtype): function
964 def patch_reduce_scatter(new_reduce_scatter_tensor: Callable): function