Home
last modified time | relevance | path

Searched refs:LocalParticipant (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnccl_utils.cc140 const std::vector<LocalParticipant>& local_participants) { in LocalParticipantsToString()
142 for (const LocalParticipant& local_participant : local_participants) { in LocalParticipantsToString()
152 const std::vector<LocalParticipant>& local_participants, in CreateNcclClique()
219 const std::vector<LocalParticipant>& local_participants, in NcclCliqueRendezvous()
249 const std::vector<LocalParticipant>& local_participants_;
258 StatusOr<std::vector<LocalParticipant>> GetLocalParticipants( in GetLocalParticipants()
261 std::vector<LocalParticipant> local_participants; in GetLocalParticipants()
331 se::Stream* stream, const std::vector<LocalParticipant>& local_participants, in AcquireNcclClique()
Dnccl_utils.h103 struct LocalParticipant { struct
108 StatusOr<std::vector<LocalParticipant>> GetLocalParticipants( argument
152 se::Stream* stream, const std::vector<LocalParticipant>& local_participants,
Dnccl_collective_thunk.cc85 std::vector<LocalParticipant> local_participants, in ExecuteOnStream()