Searched refs:CollectivePermuteParticipantData (Results 1 – 1 of 1) sorted by relevance
137 struct CollectivePermuteParticipantData : xla::ParticipantData { struct138 CollectivePermuteParticipantData(const xla::RendezvousKey& rendezvous_key_p, in CollectivePermuteParticipantData() argument381 : public xla::Rendezvous<CollectivePermuteParticipantData, std::nullptr_t> {384 : xla::Rendezvous<CollectivePermuteParticipantData, std::nullptr_t>(k) {} in CpuCollectivePermuteRendezvous()388 const CollectivePermuteParticipantData& /*participant*/) override { in RunCollectiveOp() argument745 CollectivePermuteParticipantData participant(rendezvous_key, device_ordinal, in __xla_cpu_runtime_CollectivePermute()