Home
last modified time | relevance | path

Searched refs:set_nccl_unique_id_callback (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable_run_options.cc41 GpuExecutableRunOptions& GpuExecutableRunOptions::set_nccl_unique_id_callback( in set_nccl_unique_id_callback() function in xla::gpu::GpuExecutableRunOptions
Dgpu_executable_run_options.h72 GpuExecutableRunOptions& set_nccl_unique_id_callback(
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dgpu_device.cc298 gpu_executable_run_options->set_nccl_unique_id_callback( in BuildDistributedDevices()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_execute_op.cc344 gpu_options.set_nccl_unique_id_callback(uid_callback); in RunExecutable()