Searched refs:set_gpu_global_device_ids (Results 1 – 4 of 4) sorted by relevance
30 GpuExecutableRunOptions& GpuExecutableRunOptions::set_gpu_global_device_ids( in set_gpu_global_device_ids() function in xla::gpu::GpuExecutableRunOptions
65 GpuExecutableRunOptions& set_gpu_global_device_ids(
294 gpu_executable_run_options->set_gpu_global_device_ids( in BuildDistributedDevices()
332 gpu_options.set_gpu_global_device_ids(gpu_global_ids); in RunExecutable()