HomeSort by relevance Sort by last modified time
    Searched refs:num_local_devices (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/core/nccl/
nccl_manager.cc 137 num_local_devices(num_local_devices_in),
148 const int num_local_devices; // devices local to this node member in struct:tensorflow::NcclManager::Collective
166 // eligible for running with <available_participants> == num_local_devices.
230 for (i = 0; i < collective->num_local_devices; ++i) {
236 if (i == collective->num_local_devices) {
269 std::vector<CommunicatorMember> members(collective->num_local_devices);
270 std::vector<int> devices(collective->num_local_devices);
271 for (int i = 0; i < collective->num_local_devices; ++i) {
301 std::vector<ncclComm_t> nccl_comms(collective->num_local_devices);
314 for (int i = 0; i < collective->num_local_devices; ++i)
    [all...]
nccl_manager.h 117 Context(const string& collective_key, int num_local_devices,
120 num_local_devices(num_local_devices),
128 int num_local_devices; member in struct:tensorflow::NcclManager::Context
  /external/tensorflow/tensorflow/core/kernels/
collective_nccl_reducer.cc 136 const int num_local_devices = col_params_->instance.num_devices_per_task.at( local
151 << " num local devices " << num_local_devices
157 {nccl_collective_key, num_local_devices, num_global_devices,
  /external/tensorflow/tensorflow/core/distributed_runtime/
master.cc 451 int num_local_devices = 0;
454 if (num_local_devices == 0) {
458 num_local_devices++;

Completed in 393 milliseconds