Searched defs:nccl_unique_id_callback (Results 1 – 2 of 2) sorted by relevance
43 NcclUniqueIdCallback nccl_unique_id_callback) { in set_nccl_unique_id_callback()48 const NcclUniqueIdCallback& GpuExecutableRunOptions::nccl_unique_id_callback() in nccl_unique_id_callback() function in xla::gpu::GpuExecutableRunOptions
92 const NcclUniqueIdCallback* nccl_unique_id_callback; // may be null member