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.cc342 node_states[node].nccl_manager.AddToAllReduce( in RunMultiNodeAllReduceTest()
484 NcclManager::instance()->AddToAllReduce( in TYPED_TEST()
551 NcclManager::instance()->AddToAllReduce( in TYPED_TEST()
701 NcclManager::instance()->AddToAllReduce(std::move(participant), in TYPED_TEST()
737 NcclManager::instance()->AddToAllReduce( in TYPED_TEST()
767 NcclManager::instance()->AddToAllReduce(std::move(participant), in TYPED_TEST()
886 nodes[node].nccl_manager.AddToAllReduce( in TYPED_TEST()
Dnccl_manager.h165 void AddToAllReduce(std::unique_ptr<Participant> participant,
Dcollective_communicator.cc144 nccl_manager_.AddToAllReduce(std::move(participant), context, in Enqueue()
Dnccl_manager.cc432 void NcclManager::AddToAllReduce(std::unique_ptr<Participant> participant, in AddToAllReduce() function in tensorflow::NcclManager
/external/tensorflow/tensorflow/core/kernels/
Dnccl_ops.cc113 NcclManager::instance()->AddToAllReduce( in ComputeAsync()