Searched refs:CpuAllReduceRendezvous (Results 1 – 1 of 1) sorted by relevance
422 class CpuAllReduceRendezvous class425 explicit CpuAllReduceRendezvous(const xla::RendezvousKey& k) in CpuAllReduceRendezvous() function in __anon95669c100311::CpuAllReduceRendezvous566 xla::RefcountingHashMap<xla::RendezvousKey, CpuAllReduceRendezvous>&569 *new xla::RefcountingHashMap<xla::RendezvousKey, CpuAllReduceRendezvous>; in GlobalAllReduceRendezvousMap()697 return absl::make_unique<CpuAllReduceRendezvous>(k); in __xla_cpu_runtime_AllReduce()700 TF_CHECK_OK(CpuAllReduceRendezvous::SubmitParticipant( in __xla_cpu_runtime_AllReduce()