Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/nccl/
Dnccl_manager.cc174 communicator_key(communicator_key_in) { in Collective()
195 const string communicator_key; member
288 if (collective->communicator_key.empty()) { in GetCommunicator()
330 if (collective->communicator_key.size() != NCCL_UNIQUE_ID_BYTES) { in GetCommunicator()
333 collective->communicator_key.size()); in GetCommunicator()
339 if (comm->key == collective->communicator_key) { in GetCommunicator()
398 StringToNcclUniqueId(collective->communicator_key, &nccl_id); in GetCommunicator()
427 new Communicator(std::move(members), collective->communicator_key)); in GetCommunicator()
511 context.communicator_key); in AddParticipant()
519 collective->communicator_key.empty()) { in AddParticipant()
[all …]
Dnccl_manager.h135 int num_global_devices, const string& communicator_key, in Context()
140 communicator_key(communicator_key), in Context()
158 const string& communicator_key; member
Dnccl_manager_test.cc319 const string communicator_key = in RunMultiNodeAllReduceTest() local
329 &node_states, &communicator_key, &collective_key, in RunMultiNodeAllReduceTest()
345 communicator_key, /*source_rank=*/-1}, in RunMultiNodeAllReduceTest()
370 const string communicator_key = in RunMultiNodeBroadcastTest() local
381 &collective_key, &communicator_key, &test_case]() { in RunMultiNodeBroadcastTest()
400 communicator_key, src_global_rank}); in RunMultiNodeBroadcastTest()
405 communicator_key, src_global_rank}); in RunMultiNodeBroadcastTest()
636 const string communicator_key = in TEST() local
638 EXPECT_EQ(communicator_key.size(), NCCL_UNIQUE_ID_BYTES); in TEST()
877 const string& communicator_key) { in TYPED_TEST() argument
[all …]
Dcollective_communicator.cc108 col_params->group.runtime_details.communicator_key, in Enqueue()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_param_resolver_distributed.cc159 gr->group.runtime_details.communicator_key); in CompleteGroupAsync()
254 gr->group.runtime_details.communicator_key = resp.communicator_key(); in UpdateGroupCache()
265 << absl::CEscape(resp.communicator_key()); in UpdateGroupCache()
273 if (previous_gr->group.runtime_details.communicator_key != in UpdateGroupCache()
274 resp.communicator_key()) { in UpdateGroupCache()
278 " gives communicator_key=", absl::CEscape(resp.communicator_key()), in UpdateGroupCache()
280 absl::CEscape(previous_gr->group.runtime_details.communicator_key)); in UpdateGroupCache()
Dcollective_param_resolver_distributed_test.cc268 EXPECT_EQ(cp_[dev0]->group.runtime_details.communicator_key, in ValidateCollectiveParams()
269 cp_[device_name]->group.runtime_details.communicator_key); in ValidateCollectiveParams()
374 col_group_runtime_details->communicator_key = "mock-communicator-key"; in InitializeCollectiveGroupRuntimeDetails()
/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl.cc72 col_group_runtime_details->communicator_key = in InitializeCollectiveGroupRuntimeDetails()
/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_param_resolver_local.h112 string communicator_key TF_GUARDED_BY(mu);
/external/tensorflow/tensorflow/core/framework/
Dcollective.cc53 absl::CEscape(communicator_key), "}"); in ToString()
Dcollective.h55 string communicator_key; // for communicator-based techniques e.g. NCCL member
/external/tensorflow/tensorflow/core/protobuf/
Dworker.proto560 bytes communicator_key = 7; field