Searched refs:NcclUniqueIdCallback (Results 1 – 6 of 6) sorted by relevance
53 using NcclUniqueIdCallback = variable73 NcclUniqueIdCallback nccl_unique_id_callback);74 const NcclUniqueIdCallback& nccl_unique_id_callback() const;78 NcclUniqueIdCallback nccl_unique_id_callback_;
42 NcclUniqueIdCallback nccl_unique_id_callback) { in set_nccl_unique_id_callback()47 const NcclUniqueIdCallback& GpuExecutableRunOptions::nccl_unique_id_callback() in nccl_unique_id_callback()
153 const NcclUniqueIdCallback* callback) { in CreateNcclClique()220 const NcclUniqueIdCallback* callback) in NcclCliqueRendezvous()250 const NcclUniqueIdCallback* callback_;332 const NcclUniqueIdCallback* callback) { in AcquireNcclClique()
113 const NcclUniqueIdCallback* nccl_unique_id_callback; // may be null
153 const NcclUniqueIdCallback* callback); // may be null
174 const gpu::NcclUniqueIdCallback* nccl_unique_id_callback = nullptr;