Searched refs:CpuCollectivePermuteRendezvous (Results 1 – 1 of 1) sorted by relevance
380 class CpuCollectivePermuteRendezvous class383 explicit CpuCollectivePermuteRendezvous(const xla::RendezvousKey& k) in CpuCollectivePermuteRendezvous() function in __anon95669c100311::CpuCollectivePermuteRendezvous573 xla::RefcountingHashMap<xla::RendezvousKey, CpuCollectivePermuteRendezvous>&576 CpuCollectivePermuteRendezvous>; in GlobalCollectivePermuteRendezvousMap()754 return absl::make_unique<CpuCollectivePermuteRendezvous>(k); in __xla_cpu_runtime_CollectivePermute()757 CpuCollectivePermuteRendezvous::SubmitParticipant( in __xla_cpu_runtime_CollectivePermute()