Searched refs:collective (Results 1 – 25 of 172) sorted by relevance
1234567
195 Status NcclManager::GetCommunicator(NcclManager::Collective* collective, in GetCommunicator() argument198 std::sort(collective->participants.begin(), collective->participants.end(), in GetCommunicator()206 if (collective->single_node) { in GetCommunicator()228 if (comm->num_devices == collective->num_global_devices) { in GetCommunicator()230 for (i = 0; i < collective->num_local_devices; ++i) { in GetCommunicator()232 collective->participants[i]->executor) { in GetCommunicator()236 if (i == collective->num_local_devices) { in GetCommunicator()247 if (collective->communicator_key.size() != NCCL_UNIQUE_ID_BYTES) { in GetCommunicator()250 collective->communicator_key.size()); in GetCommunicator()256 if (comm->key == collective->communicator_key) { in GetCommunicator()[all …]
190 Status GetCommunicator(Collective* collective, Communicator** communicator);207 Collective* CheckReady(const string& collective_key, Collective* collective)211 void RunCollective(Collective* collective);
4 name: "collective.all_reduce"
4 name: "collective.broadcast_send"
4 name: "collective.broadcast_recv"
4 name: "collective.BroadcastRecv"
4 name: "collective.BroadcastSend"
4 name: "collective.AllReduce"
36 for different input data in each invocation. This collective
126 // If graph_def contains any collective ops this must be a positive515 // Supplies data about one collective op belonging to the instance identified543 // a common collective instance to coordinate using the same step_ids.
439 // If true, make collective op execution order sequential and deterministic440 // for potentially concurrent collective instances.494 // If non-zero, declares that this graph is going to use collective
24 a collective op to do all-reduce. This supports between-graph communication and
127 collective works based on the Program.
122 the right to control the distribution of derivative or collective works
59 considered to be a derivative work or collective work270 collective works based on the Program.675 collective works based on the Library.
47 collective whole. A work that constitutes a Collection will not be