Searched refs:_reduce_scatter_tensor_coalesced (Results 1 – 2 of 2) sorted by relevance
75 def _reduce_scatter_tensor_coalesced( function
6399 def _reduce_scatter_tensor_coalesced(inputs, reduce_op, group_size, group_name): function