Searched defs:ParticipantData (Results 1 – 2 of 2) sorted by relevance
141 struct ParticipantData { struct142 ParticipantData(const RendezvousKey& rendezvous_key, int64 device_ordinal, in ParticipantData() argument148 virtual ~ParticipantData() {} in ~ParticipantData() argument158 struct AllReduceParticipantData : ParticipantData { argument
91 struct ParticipantData { struct92 int64 replica_id;93 se::Stream* stream;95 se::DeviceMemoryBase src;96 se::DeviceMemoryBase dest;99 std::vector<int64> dest_replicas;