Searched refs:AddParticipant (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager.cc | 346 AddParticipant(std::move(participant), context, kAllReduce, reduction_op); in AddToAllReduce()
351 AddParticipant(std::move(participant), context, kAllGather, in AddToAllGather()
358 AddParticipant(std::move(participant), context, kBroadcast, in AddBroadcastSend()
364 AddParticipant(std::move(participant), context, kBroadcast, in AddBroadcastRecv()
371 AddParticipant(std::move(participant), context, kReduce, reduction_op); in AddReduceSend()
377 AddParticipant(std::move(participant), context, kReduce, reduction_op); in AddReduceRecv()
395 void NcclManager::AddParticipant(std::unique_ptr<Participant> participant, in AddParticipant() function in tensorflow::NcclManager
|
| D | nccl_manager.h | 196 void AddParticipant(std::unique_ptr<Participant> participant,
|