Searched refs:num_global_devices (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager.h | 118 int num_global_devices, const string& communicator_key) in Context()
121 num_global_devices(num_global_devices), in Context()
131 int num_global_devices; member
|
| D | nccl_manager.cc | 138 num_global_devices(num_global_devices_in), in Collective()
149 const int num_global_devices; // devices across all nodes member
228 if (comm->num_devices == collective->num_global_devices) { in GetCommunicator()
321 nccl_comms.data() + i, collective->num_global_devices, nccl_id, rank)); in GetCommunicator()
408 context.num_global_devices, context.communicator_key); in AddParticipant()
439 collective->num_global_devices != context.num_global_devices) { in AddParticipant()
443 collective->num_global_devices, " but now got ", in AddParticipant()
444 context.num_global_devices); in AddParticipant()
476 if (collective->num_global_devices == collective->num_local_devices || in CheckReady()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_reducer.cc | 135 const int num_global_devices = col_params_->group.group_size; in Run() local
152 << " num global devices " << num_global_devices << " device " in Run()
157 {nccl_collective_key, num_local_devices, num_global_devices, in Run()
|