Searched defs:_all_gather_base (Results 1 – 2 of 2) sorted by relevance
122 def _all_gather_base(output_tensor, input_tensor, group=group.WORLD): function
3449 def _all_gather_base(output_tensor, input_tensor, group=None, async_op=False): function