Searched defs:output_tensor_lists (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/test/distributed/ |
| H A D | test_c10d_common.py | 1561 def allgather(self, output_tensor_lists, input_tensor_list, opts=None): argument
|
| /aosp_15_r20/external/pytorch/torch/distributed/ |
| H A D | distributed_c10d.py | 3481 output_tensor_lists, input_tensor_list, group=None, async_op=False argument
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/ |
| H A D | distributed_test.py | 3614 self, output_tensor_lists, input_tensors, expected_tensors, group_id argument
|