Searched defs:AllToAllParticipantData (Results 1 – 1 of 1) sorted by relevance
197 struct AllToAllParticipantData : xla::ParticipantData { struct198 AllToAllParticipantData(const xla::RendezvousKey& rendezvous_key_p, in AllToAllParticipantData() function204 int64_t device_ordinal;205 se::Stream* stream;206 std::vector<se::DeviceMemoryBase> source_buffers;207 std::vector<se::DeviceMemoryBase> destination_buffers;208 xla::GlobalDeviceId device_id;212 std::vector<xla::GlobalDeviceId> devices_to_copy_to;214 std::string ToString() const override { in ToString()391 const AllToAllParticipantData& /*participant*/) override { in RunCollectiveOp()