Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/nccl/
Dnccl_manager.cc169 communicator_key(communicator_key_in) { in Collective()
190 const string communicator_key; member
276 if (collective->communicator_key.empty()) { in GetCommunicator()
318 if (collective->communicator_key.size() != NCCL_UNIQUE_ID_BYTES) { in GetCommunicator()
321 collective->communicator_key.size()); in GetCommunicator()
327 if (comm->key == collective->communicator_key) { in GetCommunicator()
386 StringToNcclUniqueId(collective->communicator_key, &nccl_id); in GetCommunicator()
415 new Communicator(std::move(members), collective->communicator_key)); in GetCommunicator()
495 context.num_global_devices, context.communicator_key); in AddParticipant()
503 collective->communicator_key.empty()) { in AddParticipant()
[all …]
Dnccl_manager.h146 int num_global_devices, const string& communicator_key, in Context()
151 communicator_key(communicator_key), in Context()
169 const string& communicator_key; member
Dnccl_manager_test.cc312 const string communicator_key = in RunMultiNodeAllReduceTest() local
322 &node_states, &communicator_key, &collective_key, in RunMultiNodeAllReduceTest()
338 communicator_key, /*source_rank=*/-1}, in RunMultiNodeAllReduceTest()
363 const string communicator_key = in RunMultiNodeBroadcastTest() local
374 &collective_key, &communicator_key, &test_case]() { in RunMultiNodeBroadcastTest()
393 communicator_key, src_global_rank}); in RunMultiNodeBroadcastTest()
398 communicator_key, src_global_rank}); in RunMultiNodeBroadcastTest()
629 const string communicator_key = in TEST() local
631 EXPECT_EQ(communicator_key.size(), NCCL_UNIQUE_ID_BYTES); in TEST()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_param_resolver_distributed.cc144 gr->group.runtime_details.communicator_key); in CompleteGroupAsync()
245 gr->group.runtime_details.communicator_key = resp.communicator_key(); in UpdateGroupCache()
247 << absl::CEscape(gr->group.runtime_details.communicator_key); in UpdateGroupCache()
255 << absl::CEscape(gr->group.runtime_details.communicator_key); in UpdateGroupCache()
259 if (previous_gr->group.runtime_details.communicator_key != in UpdateGroupCache()
260 gr->group.runtime_details.communicator_key) { in UpdateGroupCache()
264 absl::CEscape(gr->group.runtime_details.communicator_key), in UpdateGroupCache()
266 absl::CEscape(previous_gr->group.runtime_details.communicator_key)); in UpdateGroupCache()
Dcollective_param_resolver_distributed_test.cc273 EXPECT_EQ(cp_[0].group.runtime_details.communicator_key, in ValidateCollectiveParams()
274 cp_[idx].group.runtime_details.communicator_key); in ValidateCollectiveParams()
335 col_group_runtime_details->communicator_key = "mock-communicator-key"; in InitializeCollectiveGroupRuntimeDetails()
/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_broadcaster.cc50 col_params_->group.runtime_details.communicator_key, in Run()
56 col_params_->group.runtime_details.communicator_key, in Run()
Dcollective_nccl.cc71 col_group_runtime_details->communicator_key = in InitializeCollectiveGroupRuntimeDetails()
Dcollective_nccl_gatherer.cc47 col_params_->group.runtime_details.communicator_key, in Run()
Dcollective_nccl_reducer.cc149 col_params_->group.runtime_details.communicator_key, /*source_rank=*/-1}, in Run()
/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_param_resolver_local.h138 string communicator_key GUARDED_BY(out_mu);
/external/tensorflow/tensorflow/core/framework/
Dcollective.cc53 absl::CEscape(communicator_key), "}"); in ToString()
Dcollective.h53 string communicator_key; // for communicator-based techniques e.g. NCCL member
/external/tensorflow/tensorflow/core/protobuf/
Dworker.proto549 bytes communicator_key = 7; field