Searched refs:set_intra_op_thread_pool (Results 1 – 17 of 17) sorted by relevance
74 ExecutableRunOptions& ExecutableRunOptions::set_intra_op_thread_pool( in set_intra_op_thread_pool() function in xla::ExecutableRunOptions
118 ExecutableRunOptions& set_intra_op_thread_pool(
41 execute_options.set_intra_op_thread_pool( in CompileAndExecute()
155 run_options.set_intra_op_thread_pool(thread_pool_wrapper_->device.get()); in DefaultExecutableRunOptions()
173 exec_run_options.set_intra_op_thread_pool( in ExecuteAndFetchProfile()
277 run_opts.set_intra_op_thread_pool(
899 options.set_intra_op_thread_pool(&device); in BM_ParallelFusion()
131 run_options_.set_intra_op_thread_pool(pool); in set_thread_pool()
92 run_options.set_intra_op_thread_pool(&ctx->eigen_cpu_device()); in Run()
107 run_options.set_intra_op_thread_pool(&device); in EigenMatrixMultiply()
431 options.set_intra_op_thread_pool( in ExecuteParallelAndRegisterResult()529 options.set_intra_op_thread_pool( in ExecuteAndRegisterResult()
479 run_options.set_intra_op_thread_pool( in GetServiceRunOptionsForDevice()
315 run_options.set_intra_op_thread_pool(&thread_pool); in ReplayComputation()
316 run_options.set_intra_op_thread_pool(&context->eigen_cpu_device()); in RunExecutable()
183 run_options.set_intra_op_thread_pool(&ctx->eigen_cpu_device()); in RunExecutable()
1460 run_options.set_intra_op_thread_pool(client_->eigen_intraop_device()); in ExecuteHelper()
1979 run_options.set_intra_op_thread_pool( in EnqueueExecution()