Searched defs:AllToAllParticipantData (Results 1 – 1 of 1) sorted by relevance
167 struct AllToAllParticipantData : xla::ParticipantData { struct168 AllToAllParticipantData(const xla::RendezvousKey& rendezvous_key_p, in AllToAllParticipantData() function174 xla::int64 device_ordinal;175 se::Stream* stream;176 std::vector<se::DeviceMemoryBase> source_buffers;177 std::vector<se::DeviceMemoryBase> destination_buffers;178 int replica_id;182 std::vector<int> replica_ids_to_copy_to;184 std::string ToString() const override { in ToString()357 const AllToAllParticipantData& /*participant*/) override { in RunCollectiveOp()