Searched defs:nccl_unique_id_callback (Results 1 – 3 of 3) sorted by relevance
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() function in xla::gpu::GpuExecutableRunOptions
113 const NcclUniqueIdCallback* nccl_unique_id_callback; // may be null member
174 const gpu::NcclUniqueIdCallback* nccl_unique_id_callback = nullptr; member