Home
last modified time | relevance | path

Searched refs:AddBroadcastSend (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/nccl/
Dnccl_manager.h174 void AddBroadcastSend(std::unique_ptr<Participant> participant,
Dcollective_communicator.cc154 nccl_manager_.AddBroadcastSend(std::move(participant), context); in Enqueue()
Dnccl_manager_test.cc397 node_states[node].nccl_manager.AddBroadcastSend( in RunMultiNodeBroadcastTest()
709 NcclManager::instance()->AddBroadcastSend( in TYPED_TEST()
821 NcclManager::instance()->AddBroadcastSend(std::move(participant), in TYPED_TEST()
Dnccl_manager.cc444 void NcclManager::AddBroadcastSend(std::unique_ptr<Participant> participant, in AddBroadcastSend() function in tensorflow::NcclManager
/external/tensorflow/tensorflow/core/kernels/
Dnccl_ops.cc214 NcclManager::instance()->AddBroadcastSend( in ComputeAsync()