Searched defs:output_tensor_list (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/ |
| H A D | multi_threaded_pg.py | 330 def alltoall(self, output_tensor_list, input_tensor_list, opts=AllToAllOptions()): argument
399 …def allgather_into_tensor_coalesced(self, output_tensor_list, input_tensor_list, opts=AllgatherOpt… argument
|
| /aosp_15_r20/external/pytorch/torch/distributed/nn/ |
| H A D | functional.py | 158 def all_to_all(output_tensor_list, input_tensor_list, group=group.WORLD): argument
|
| /aosp_15_r20/external/pytorch/test/distributed/ |
| H A D | test_c10d_common.py | 1601 def reduce_scatter(self, output_tensor_list, input_tensor_lists, opts=None): argument
|
| /aosp_15_r20/external/pytorch/torch/distributed/ |
| H A D | distributed_c10d.py | 4007 def all_to_all(output_tensor_list, input_tensor_list, group=None, async_op=False): argument
|