Searched refs:AddParticipant (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager.cc | 435 AddParticipant(std::move(participant), context, kAllReduce, reduction_op); in AddToAllReduce()
440 AddParticipant(std::move(participant), context, kAllGather, in AddToAllGather()
447 AddParticipant(std::move(participant), context, kBroadcast, in AddBroadcastSend()
453 AddParticipant(std::move(participant), context, kBroadcast, in AddBroadcastRecv()
460 AddParticipant(std::move(participant), context, kReduce, reduction_op); in AddReduceSend()
467 AddParticipant(std::move(participant), context, kReduce, reduction_op); in AddReduceRecv()
489 void NcclManager::AddParticipant(std::unique_ptr<Participant> participant, in AddParticipant() function in tensorflow::NcclManager
|
| D | nccl_manager.h | 226 void AddParticipant(std::unique_ptr<Participant> participant,
|