Searched refs:AddToAllReduce (Results 1 – 5 of 5) sorted by relevance
276 NcclManager::instance()->AddToAllReduce( in TYPED_TEST()345 NcclManager::instance()->AddToAllReduce( in TYPED_TEST()443 nccl_managers[node].AddToAllReduce( in TYPED_TEST()480 NcclManager::instance()->AddToAllReduce(std::move(participant), in TYPED_TEST()514 NcclManager::instance()->AddToAllReduce( in TYPED_TEST()543 NcclManager::instance()->AddToAllReduce(std::move(participant), in TYPED_TEST()
143 void AddToAllReduce(std::unique_ptr<Participant> participant,
343 void NcclManager::AddToAllReduce(std::unique_ptr<Participant> participant, in AddToAllReduce() function in tensorflow::NcclManager
155 NcclManager::instance()->AddToAllReduce( in Run()
110 NcclManager::instance()->AddToAllReduce( in ComputeAsync()