Searched refs:set_intra_op_thread_pool (Results 1 – 15 of 15) sorted by relevance
58 ExecutableRunOptions& ExecutableRunOptions::set_intra_op_thread_pool( in set_intra_op_thread_pool() function in xla::ExecutableRunOptions
68 ExecutableRunOptions& set_intra_op_thread_pool(
41 execute_options.set_intra_op_thread_pool( in CompileAndExecute()
152 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()
305 run_opts.set_intra_op_thread_pool( in BuildAndRunComputation()
905 options.set_intra_op_thread_pool(&device); in BM_ParallelFusion()
125 run_options_.set_intra_op_thread_pool(pool); in set_thread_pool()
363 run_options.set_intra_op_thread_pool(&ctx->eigen_cpu_device()); in Compute()537 run_options.set_intra_op_thread_pool(&ctx->eigen_cpu_device()); in Compute()
78 run_options.set_intra_op_thread_pool(&ctx->eigen_cpu_device()); in Run()
107 run_options.set_intra_op_thread_pool(&device); in EigenMatrixMultiply()
210 run_options.set_intra_op_thread_pool(&context->eigen_cpu_device()); in DoWork()
248 options.set_intra_op_thread_pool( in Execute()296 options.set_intra_op_thread_pool( in ExecutePerReplica()
451 options.set_intra_op_thread_pool( in ExecuteParallelAndRegisterResult()546 options.set_intra_op_thread_pool( in ExecuteAndRegisterResult()
448 run_options.set_intra_op_thread_pool( in GetServiceRunOptionsForDevice()