Searched refs:AddBroadcastRecv (Results 1 – 5 of 5) sorted by relevance
176 void AddBroadcastRecv(std::unique_ptr<Participant> participant,
156 nccl_manager_.AddBroadcastRecv(std::move(participant), context); in Enqueue()
402 node_states[node].nccl_manager.AddBroadcastRecv( in RunMultiNodeBroadcastTest()794 NcclManager::instance()->AddBroadcastRecv(std::move(participant), in TYPED_TEST()849 NcclManager::instance()->AddBroadcastRecv(std::move(participant), in TYPED_TEST()
451 void NcclManager::AddBroadcastRecv(std::unique_ptr<Participant> participant, in AddBroadcastRecv() function in tensorflow::NcclManager
250 NcclManager::instance()->AddBroadcastRecv( in ComputeAsync()