Searched refs:GpuExecutableRunOptions (Results 1 – 12 of 12) sorted by relevance
31 GpuExecutableRunOptions& GpuExecutableRunOptions::set_gpu_global_device_ids( in set_gpu_global_device_ids()38 GpuExecutableRunOptions::gpu_global_device_ids() const { in gpu_global_device_ids()42 GpuExecutableRunOptions& GpuExecutableRunOptions::set_nccl_unique_id_callback( in set_nccl_unique_id_callback()48 const NcclUniqueIdCallback& GpuExecutableRunOptions::nccl_unique_id_callback() in nccl_unique_id_callback()58 const GpuExecutableRunOptions* gpu_options = in NcclExecuteParams()
61 class GpuExecutableRunOptions {67 GpuExecutableRunOptions& set_gpu_global_device_ids(74 GpuExecutableRunOptions& set_nccl_unique_id_callback(
42 class GpuExecutableRunOptions; variable155 const gpu::GpuExecutableRunOptions* gpu_executable_run_options);156 const gpu::GpuExecutableRunOptions* gpu_executable_run_options() const;170 const gpu::GpuExecutableRunOptions* gpu_executable_run_options_ = nullptr;
106 const gpu::GpuExecutableRunOptions* gpu_executable_run_options) { in set_gpu_executable_run_options()111 const gpu::GpuExecutableRunOptions*
154 std::unique_ptr<gpu::GpuExecutableRunOptions> gpu_run_options);286 gpu::GpuExecutableRunOptions* gpu_run_options() const { in gpu_run_options()367 std::unique_ptr<gpu::GpuExecutableRunOptions> gpu_run_options_;
341 gpu::GpuExecutableRunOptions* gpu_executable_run_options) { in BuildDistributedDevices()441 auto gpu_run_options = std::make_unique<gpu::GpuExecutableRunOptions>(); in GetGpuClient()
211 std::unique_ptr<gpu::GpuExecutableRunOptions> gpu_run_options) in PjRtStreamExecutorClient()
198 xla::gpu::GpuExecutableRunOptions& gpu_options);
151 xla::gpu::GpuExecutableRunOptions& gpu_options) { in ResolveDeviceAssignment()
82 xla::gpu::GpuExecutableRunOptions gpu_options; in Run()
327 xla::gpu::GpuExecutableRunOptions gpu_options; in RunExecutable()
354 xla::gpu::GpuExecutableRunOptions gpu_options; in ComputeAsync()