Searched defs:CollectivePermuteParticipantData (Results 1 – 1 of 1) sorted by relevance
171 struct CollectivePermuteParticipantData : xla::ParticipantData { struct172 CollectivePermuteParticipantData(const xla::RendezvousKey& rendezvous_key_p, in CollectivePermuteParticipantData() function179 int64_t device_ordinal;180 se::Stream* stream;181 int replica_id;182 se::DeviceMemoryBase source_data;183 se::DeviceMemoryBase destination_data;184 int64_t byte_size;185 std::vector<int> replica_ids_to_copy_to;187 std::string ToString() const override { in ToString()[all …]