Home
last modified time | relevance | path

Searched refs:communicator_key (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_param_resolver_distributed.cc190 if (!cp->instance.communicator_key.empty()) { in CompleteInstanceAsync()
192 cp->instance.communicator_key); in CompleteInstanceAsync()
296 string communicator_key = resp.communicator_key(); in UpdateInstanceCache() local
298 auto continue_with_ir = [cp, irp, source_rank, communicator_key, in UpdateInstanceCache()
321 if (ir->communicator_key != communicator_key) { in UpdateInstanceCache()
322 if (!ir->communicator_key.empty()) { in UpdateInstanceCache()
326 " gives communicator_key with size =", communicator_key.size(), in UpdateInstanceCache()
328 ir->communicator_key.size()); in UpdateInstanceCache()
332 ir->communicator_key = communicator_key; in UpdateInstanceCache()
Dcollective_param_resolver_distributed_test.cc270 EXPECT_EQ(cp_[0].instance.communicator_key, in ValidateCollectiveParams()
271 cp_[idx].instance.communicator_key); in ValidateCollectiveParams()
/external/tensorflow/tensorflow/core/nccl/
Dnccl_manager.cc140 communicator_key(communicator_key_in), in Collective()
151 const string communicator_key; member
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()
309 StringToNcclUniqueId(collective->communicator_key, &nccl_id); in GetCommunicator()
338 new Communicator(std::move(members), collective->communicator_key)); in GetCommunicator()
408 context.num_global_devices, context.communicator_key); in AddParticipant()
418 !collective->communicator_key.empty()) { in AddParticipant()
422 collective->communicator_key.size()); in AddParticipant()
[all …]
Dnccl_manager.h118 int num_global_devices, const string& communicator_key) in Context()
122 communicator_key(communicator_key) {} in Context()
139 const string& communicator_key; member
Dnccl_manager_test.cc402 const string communicator_key = in TEST() local
404 EXPECT_EQ(communicator_key.size(), NCCL_UNIQUE_ID_BYTES); in TEST()
423 const string communicator_key = nccl_managers[0].GenerateCommunicatorKey(); in TYPED_TEST() local
431 auto node_fn = [this, node, &nccl_managers, &communicator_key, in TYPED_TEST()
446 communicator_key}, in TYPED_TEST()
/external/tensorflow/tensorflow/core/framework/
Dcollective.cc70 communicator_key = other.communicator_key; in operator =()
103 ", communicator_key=", str_util::CEscape(communicator_key), in ToString()
Dcollective.h97 string communicator_key; member
/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_param_resolver_local.cc702 cp->instance.communicator_key = irec->communicator_key; in CompleteInstanceFromInitializedIRec()
739 if (!ir->communicator_key.empty()) { in WaitForGroup()
743 str_util::CEscape(ir->communicator_key), in WaitForGroup()
747 ir->communicator_key = cp->instance.communicator_key; in WaitForGroup()
766 if (init_nccl && ir->communicator_key.empty()) { in WaitForGroup()
Dcollective_param_resolver_local.h138 string communicator_key GUARDED_BY(out_mu);
/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_reducer.cc55 col_params->instance.communicator_key = in InitializeInstanceBeforeGroupDiscovery()
158 col_params_->instance.communicator_key}, in Run()
/external/tensorflow/tensorflow/core/protobuf/
Dworker.proto538 bytes communicator_key = 3; field