Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnccl_all_reduce_thunk.cc46 struct ParticipantData { struct
92 Status SubmitParticipant(ParticipantData participant);
116 Status DoAllReduce(ParticipantData data, ncclComm_t comm);
139 std::vector<ParticipantData> participants_ GUARDED_BY(mutex_);
148 Status GlobalRendezvousManager::SubmitParticipant(ParticipantData participant) { in SubmitParticipant()
264 for (ParticipantData& data : participants_) { in InitializeCommunicationChannels()
282 Status GlobalRendezvousManager::DoAllReduce(ParticipantData participant, in DoAllReduce()
319 ParticipantData participant; in ExecuteOnStream()