Searched refs:set_nccl_unique_id_callback (Results 1 – 4 of 4) sorted by relevance
41 GpuExecutableRunOptions& GpuExecutableRunOptions::set_nccl_unique_id_callback( in set_nccl_unique_id_callback() function in xla::gpu::GpuExecutableRunOptions
72 GpuExecutableRunOptions& set_nccl_unique_id_callback(
298 gpu_executable_run_options->set_nccl_unique_id_callback( in BuildDistributedDevices()
344 gpu_options.set_nccl_unique_id_callback(uid_callback); in RunExecutable()