Searched refs:_object_to_tensor (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/torch/distributed/ |
| D | distributed_c10d.py | 2620 def _object_to_tensor(obj, device, group): function
2711 input_tensor, local_size = _object_to_tensor(obj, current_device, group)
2811 input_tensor, local_size = _object_to_tensor(obj, current_device, group)
2930 *[_object_to_tensor(obj, current_device, group) for obj in object_list]
3120 *[_object_to_tensor(obj, current_device, group) for obj in object_list]
3232 _object_to_tensor(obj, pg_device, group)
|