Searched refs:AllToAllParticipantData (Results 1 – 1 of 1) sorted by relevance
159 struct AllToAllParticipantData : xla::ParticipantData { struct160 AllToAllParticipantData(const xla::RendezvousKey& rendezvous_key_p, in AllToAllParticipantData() function320 : public xla::Rendezvous<AllToAllParticipantData, std::nullptr_t> {323 : xla::Rendezvous<AllToAllParticipantData, std::nullptr_t>(k) {} in CpuAllToAllRendezvous()327 const AllToAllParticipantData& /*participant*/) override { in RunCollectiveOp() argument341 const AllToAllParticipantData& p = participants_[pos]; in RunCollectiveOp()361 for (const AllToAllParticipantData& sender : participants_) { in RunCollectiveOp()369 AllToAllParticipantData& receiver = participants_[participant_num]; in RunCollectiveOp()635 AllToAllParticipantData participant(rendezvous_key, device_ordinal, in __xla_cpu_runtime_AllToAll()