Searched defs:CollectivePermuteParticipantData (Results 1 – 1 of 1) sorted by relevance
137 struct CollectivePermuteParticipantData : xla::ParticipantData { struct138 CollectivePermuteParticipantData(const xla::RendezvousKey& rendezvous_key_p, in CollectivePermuteParticipantData() function143 int replica_id;144 se::DeviceMemoryBase source_data;145 se::DeviceMemoryBase destination_data;146 xla::int64 byte_size;147 std::vector<int> replica_ids_to_copy_to;149 std::string ToString() const override { in ToString()388 const CollectivePermuteParticipantData& /*participant*/) override { in RunCollectiveOp()