Searched refs:set_gpu_executable_run_options (Results 1 – 5 of 5) sorted by relevance
105 ExecutableRunOptions& ExecutableRunOptions::set_gpu_executable_run_options( in set_gpu_executable_run_options() function in xla::ExecutableRunOptions
154 ExecutableRunOptions& set_gpu_executable_run_options(
252 run_options.set_gpu_executable_run_options(&gpu_options); in ResolveDeviceAssignment()
350 run_options.set_gpu_executable_run_options(&gpu_options); in RunExecutable()
1984 run_options.set_gpu_executable_run_options(client_->gpu_run_options()); in EnqueueExecution()