Searched refs:ParticipantData (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | collective_permute_thunk.cc | 91 struct ParticipantData { struct
120 ParticipantData participant);
136 std::map<int64, ParticipantData> participants_ TF_GUARDED_BY(mu_);
160 ParticipantData participant) { in SubmitParticipant()
195 const ParticipantData& src_participant = kv.second; in SubmitParticipant()
197 const ParticipantData& dest_participant = in SubmitParticipant()
258 ParticipantData participant{replica_id, params.stream, src_addr, dest_addr, in ExecuteOnStream()
|
| D | nccl_utils.cc | 210 struct NcclCliqueParticipantData : public ParticipantData {
211 using ParticipantData::ParticipantData;
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | collective_ops_utils.h | 141 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
161 : ParticipantData(rendezvous_key_p, device_ordinal_p, stream_p) {} in AllReduceParticipantData()
208 std::enable_if_t<std::is_base_of<ParticipantData, I>::value>>
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_runtime.cc | 137 struct CollectivePermuteParticipantData : xla::ParticipantData {
141 : ParticipantData(rendezvous_key_p, device_ordinal_p, stream_p) {} in CollectivePermuteParticipantData()
159 struct AllToAllParticipantData : xla::ParticipantData {
162 : ParticipantData(rendezvous_key_p, device_ordinal_p, stream_p) {} in AllToAllParticipantData()
|