Searched defs:gather_list (Results 1 – 1 of 1) sorted by relevance
3558 def _validate_output_list_for_rank(my_rank, dst, gather_list): argument3572 def gather(tensor, gather_list=None, dst=0, group=None, async_op=False): argument