Searched refs:set_intra_op_thread_pool (Results 1 – 16 of 16) 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()
158 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()
271 run_opts.set_intra_op_thread_pool(
904 options.set_intra_op_thread_pool(&device); in BM_ParallelFusion()
130 run_options_.set_intra_op_thread_pool(pool); in set_thread_pool()
84 run_options.set_intra_op_thread_pool(&ctx->eigen_cpu_device()); in Run()
105 run_options.set_intra_op_thread_pool(&device); in EigenMatrixMultiply()
274 run_options.set_intra_op_thread_pool(&ctx->eigen_cpu_device()); in Compute()510 run_options.set_intra_op_thread_pool(&ctx->eigen_cpu_device()); in Compute()
329 run_options.set_intra_op_thread_pool(&thread_pool); in ReplayComputation()
449 run_options.set_intra_op_thread_pool( in GetServiceRunOptionsForDevice()
401 options.set_intra_op_thread_pool( in ExecuteParallelAndRegisterResult()498 options.set_intra_op_thread_pool( in ExecuteAndRegisterResult()
314 run_options.set_intra_op_thread_pool(&context->eigen_cpu_device()); in RunExecutable()
1752 run_options.set_intra_op_thread_pool( in EnqueueExecution()