HomeSort by relevance Sort by last modified time
    Searched defs:global_rank (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/nccl/
nccl_manager.h 60 Tensor* output, int global_rank, DoneCallback done_callback)
67 global_rank(global_rank),
102 // `global_rank` is not required for single-node collectives.
103 const int global_rank; member in struct:tensorflow::NcclManager::Participant
  /external/tensorflow/tensorflow/core/common_runtime/
collective_param_resolver_local.cc 152 int global_rank; member in struct:tensorflow::__anon44803::DevRec
171 dr->global_rank = 0; // Will be populated later by EstablishGlobalRank.
311 it.second.global_rank = it.second.local_rank + next_rank;
474 new_device_names[dr.global_rank] =
476 new_task_names[dr.global_rank] =

Completed in 375 milliseconds