Searched defs:input_tensor_list (Results 1 – 5 of 5) sorted by relevance
| /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
305 def forward(ctx, op, group, tensor, *input_tensor_list):
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| H A D | xrt_state_ops.h | 74 const xrt::XLATupleNode& tuple_node, const OpInputList& input_tensor_list, in ParseTupleNode()
126 const OpInputList& input_tensor_list, in ParseTupleTree()
|
| /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/ |
| H A D | distributed_c10d.py | 3481 output_tensor_lists, input_tensor_list, group=None, async_op=False argument
4007 def all_to_all(output_tensor_list, input_tensor_list, group=None, async_op=False): argument
|
| /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
|