Searched refs:CpuAllToAllRendezvous (Results 1 – 1 of 1) sorted by relevance
319 class CpuAllToAllRendezvous class322 explicit CpuAllToAllRendezvous(const xla::RendezvousKey& k) in CpuAllToAllRendezvous() function in __anon95669c100311::CpuAllToAllRendezvous580 xla::RefcountingHashMap<xla::RendezvousKey, CpuAllToAllRendezvous>&583 *new xla::RefcountingHashMap<xla::RendezvousKey, CpuAllToAllRendezvous>; in GlobalAllToAllRendezvousMap()648 return absl::make_unique<CpuAllToAllRendezvous>(k); in __xla_cpu_runtime_AllToAll()650 TF_CHECK_OK(CpuAllToAllRendezvous::SubmitParticipant( in __xla_cpu_runtime_AllToAll()