Searched refs:_tensor_to_object (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/torch/distributed/ |
| D | distributed_c10d.py | 2639 def _tensor_to_object(tensor, tensor_size, group): function
2740 object_list[i] = _tensor_to_object(tensor, tensor_size, group)
2851 object_gather_list[i] = _tensor_to_object(tensor, tensor_size, group)
3039 object_list[i] = _tensor_to_object(obj_view, obj_size, group)
3154 object_list[i] = _tensor_to_object(obj_view, obj_size, group)
3269 scatter_object_output_list[0] = _tensor_to_object(
|