Searched refs:GpuExecutableRunOptions (Results 1 – 9 of 9) sorted by relevance
30 GpuExecutableRunOptions& GpuExecutableRunOptions::set_gpu_global_device_ids( in set_gpu_global_device_ids()37 GpuExecutableRunOptions::gpu_global_device_ids() const { in gpu_global_device_ids()41 GpuExecutableRunOptions& GpuExecutableRunOptions::set_nccl_unique_id_callback( in set_nccl_unique_id_callback()47 const NcclUniqueIdCallback& GpuExecutableRunOptions::nccl_unique_id_callback() in nccl_unique_id_callback()
59 class GpuExecutableRunOptions {65 GpuExecutableRunOptions& set_gpu_global_device_ids(72 GpuExecutableRunOptions& set_nccl_unique_id_callback(
184 const GpuExecutableRunOptions* gpu_options = in ExecuteThunks()
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*
130 std::unique_ptr<gpu::GpuExecutableRunOptions> gpu_run_options);228 gpu::GpuExecutableRunOptions* gpu_run_options() const { in gpu_run_options()273 std::unique_ptr<gpu::GpuExecutableRunOptions> gpu_run_options_;
248 gpu::GpuExecutableRunOptions* gpu_executable_run_options) { in BuildDistributedDevices()327 auto gpu_run_options = absl::make_unique<gpu::GpuExecutableRunOptions>(); in GetGpuClient()
209 std::unique_ptr<gpu::GpuExecutableRunOptions> gpu_run_options) in PjRtStreamExecutorClient()
325 xla::gpu::GpuExecutableRunOptions gpu_options; in RunExecutable()