nccl_manager.cc | 83 struct NcclManager::Communicator { 85 explicit Communicator(std::vector<CommunicatorMember> members, 153 Communicator* communicator = nullptr; member in struct:tensorflow::NcclManager::Collective 173 // Guarded by the mutex of the containing Communicator. 196 NcclManager::Communicator** communicator) { 207 // For single-node collectives, we identify a communicator uniquely by the 210 // communicator for GPUs 0, 1, and 2. 213 // the same as getting the communicator connecting the devices in th [all...] |