Searched refs:num_global_devices (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager.h | 135 int num_global_devices, const string& communicator_key, in Context()
139 num_global_devices(num_global_devices), in Context()
150 int num_global_devices; member
|
| D | nccl_manager.cc | 172 num_global_devices(num_global_devices_in), in Collective()
193 const int num_global_devices; // devices across all nodes member
311 if (comm->num_devices == collective->num_global_devices) { in GetCommunicator()
410 nccl_comms.data() + i, collective->num_global_devices, nccl_id, rank)); in GetCommunicator()
510 context.num_local_devices, context.num_global_devices, in AddParticipant()
524 " and num_global_devices=", collective->num_global_devices, in AddParticipant()
542 collective->num_global_devices != context.num_global_devices) { in AddParticipant()
546 collective->num_global_devices, " but now got ", in AddParticipant()
547 context.num_global_devices); in AddParticipant()
603 if (collective->num_global_devices == collective->num_local_devices || in CheckReady()
|
| D | collective_communicator.cc | 73 const int num_global_devices = col_params->group.group_size; in Enqueue() local
107 nccl_collective_key, num_local_devices, num_global_devices, in Enqueue()
114 << " num global devices " << num_global_devices << " device " in Enqueue()
|