Searched defs:scatter_list (Results 1 – 2 of 2) sorted by relevance
379 def reduce_scatter(self, output_tensor, scatter_list, opts=ReduceScatterOptions()): argument
3629 def scatter(tensor, scatter_list=None, src=0, group=None, async_op=False): argument