Searched refs:LockedNcclClique (Results 1 – 3 of 3) sorted by relevance
112 class LockedNcclClique {114 LockedNcclClique(NcclClique& clique, std::unique_ptr<absl::MutexLock> lock,116 LockedNcclClique(LockedNcclClique&&);117 ~LockedNcclClique();150 StatusOr<LockedNcclClique> AcquireNcclClique(
216 : public Rendezvous<NcclCliqueParticipantData, LockedNcclClique> {227 StatusOr<LockedNcclClique> RunCollectiveOp( in RunCollectiveOp()244 return LockedNcclClique(*clique, std::move(clique_lock), counter_); in RunCollectiveOp()288 LockedNcclClique::LockedNcclClique(NcclClique& clique, in LockedNcclClique() function in xla::gpu::LockedNcclClique293 LockedNcclClique::LockedNcclClique(LockedNcclClique&& other) in LockedNcclClique() function in xla::gpu::LockedNcclClique298 LockedNcclClique::~LockedNcclClique() { in ~LockedNcclClique()329 StatusOr<LockedNcclClique> AcquireNcclClique( in AcquireNcclClique()
96 LockedNcclClique locked_clique, in ExecuteOnStream()