Home
last modified time | relevance | path

Searched defs:ParticipantData (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dcollective_ops_utils.h141 struct ParticipantData { struct
142 ParticipantData(const RendezvousKey& rendezvous_key, int64 device_ordinal, in ParticipantData() function
148 virtual ~ParticipantData() {} in ~ParticipantData() argument
158 struct AllReduceParticipantData : ParticipantData { argument
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcollective_permute_thunk.cc91 struct ParticipantData { struct
92 int64 replica_id;
93 se::Stream* stream;
95 se::DeviceMemoryBase src;
96 se::DeviceMemoryBase dest;
99 std::vector<int64> dest_replicas;