Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/nccl/
Dnccl_manager_test.cc276 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()
Dnccl_manager.h143 void AddToAllReduce(std::unique_ptr<Participant> participant,
Dnccl_manager.cc343 void NcclManager::AddToAllReduce(std::unique_ptr<Participant> participant, in AddToAllReduce() function in tensorflow::NcclManager
/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_reducer.cc155 NcclManager::instance()->AddToAllReduce( in Run()
Dnccl_ops.cc110 NcclManager::instance()->AddToAllReduce( in ComputeAsync()