Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnccl_all_reduce_thunk.cc116 Status DoAllReduce(ParticipantData data, ncclComm_t comm);
229 Status all_reduce_status = DoAllReduce(participant, comm); in SubmitParticipant()
282 Status GlobalRendezvousManager::DoAllReduce(ParticipantData participant, in DoAllReduce() function in xla::gpu::__anonfb4efd5a0111::GlobalRendezvousManager