Searched refs:collective_key (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager.h | 117 Context(const string& collective_key, int num_local_devices, in Context()
119 : collective_key(collective_key), in Context()
125 const string& collective_key; member
170 void SignalMultiNodeReady(const string& collective_key);
207 Collective* CheckReady(const string& collective_key, Collective* collective)
|
| D | nccl_manager.cc | 380 void NcclManager::SignalMultiNodeReady(const string& collective_key) { in SignalMultiNodeReady() argument
384 auto collective_it = collectives_.find(collective_key); in SignalMultiNodeReady()
388 to_run = CheckReady(collective_key, collective); in SignalMultiNodeReady()
403 auto collective_it = collectives_.find(context.collective_key); in AddParticipant()
410 collectives_.emplace(context.collective_key, in AddParticipant()
466 to_run = CheckReady(context.collective_key, collective); in AddParticipant()
472 NcclManager::Collective* NcclManager::CheckReady(const string& collective_key, in CheckReady() argument
480 auto collectives_it = collectives_.find(collective_key); in CheckReady()
|
| D | nccl_manager_test.cc | 415 const string collective_key = "allreduce"; in TYPED_TEST() local
432 &collective_key, reduction_op, &test_case] { in TYPED_TEST()
445 {collective_key, num_ranks_per_node, num_global_ranks, in TYPED_TEST()
453 nccl_managers[node].SignalMultiNodeReady(collective_key); in TYPED_TEST()
|