Searched refs:gpu_run_options_ (Results 1 – 2 of 2) sorted by relevance
229 return gpu_run_options_.get(); in gpu_run_options()273 std::unique_ptr<gpu::GpuExecutableRunOptions> gpu_run_options_; variable
219 gpu_run_options_(std::move(gpu_run_options)), in PjRtStreamExecutorClient()