Searched defs:AllToAllParticipantData (Results 1 – 1 of 1) sorted by relevance
159 struct AllToAllParticipantData : xla::ParticipantData { struct160 AllToAllParticipantData(const xla::RendezvousKey& rendezvous_key_p, in AllToAllParticipantData() argument164 std::vector<se::DeviceMemoryBase> source_buffers;165 std::vector<se::DeviceMemoryBase> destination_buffers;166 int replica_id;170 std::vector<int> replica_ids_to_copy_to;172 std::string ToString() const override { in ToString()327 const AllToAllParticipantData& /*participant*/) override { in RunCollectiveOp()