Searched refs:gpu_run_options (Results 1 – 3 of 3) sorted by relevance
327 auto gpu_run_options = absl::make_unique<gpu::GpuExecutableRunOptions>(); in GetGpuClient() local331 &devices, gpu_run_options.get())); in GetGpuClient()341 /*gpu_run_options=*/std::move(gpu_run_options))); in GetGpuClient()
130 std::unique_ptr<gpu::GpuExecutableRunOptions> gpu_run_options);228 gpu::GpuExecutableRunOptions* gpu_run_options() const { in gpu_run_options() function
209 std::unique_ptr<gpu::GpuExecutableRunOptions> gpu_run_options) in PjRtStreamExecutorClient() argument219 gpu_run_options_(std::move(gpu_run_options)), in PjRtStreamExecutorClient()1757 run_options.set_gpu_executable_run_options(client_->gpu_run_options()); in EnqueueExecution()