Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dcollective_ops_utils.h75 struct RendezvousKey { struct
81 explicit RendezvousKey(const RunId& run_id, in RendezvousKey() argument
92 friend H AbslHashValue(H h, const RendezvousKey& k) { in AbslHashValue() argument
109 "RendezvousKey{run_id=%s, global_devices=[%s], " in ToString() argument
115 RunId run_id;
116 std::vector<GlobalDeviceId> global_devices;
117 int num_local_participants;
118 CollectiveOpKind collective_op_kind;
142 ParticipantData(const RendezvousKey& rendezvous_key, int64 device_ordinal, in ParticipantData() argument
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcollective_permute_thunk.cc68 struct RendezvousKey { struct
69 RunId run_id;
70 int num_participants; // int, not int64, to match BlockingCounter's counter.
72 string ToString() const { in ToString()