Home
last modified time | relevance | path

Searched refs:user_intra_op_threadpool (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
H A Dkernel_fallback_compat_request_state.cc68 tensorflow::thread::ThreadPoolInterface* user_intra_op_threadpool, in KernelFallbackCompatRequestState() argument
82 intra_op_threadpool_(user_intra_op_threadpool), in KernelFallbackCompatRequestState()
90 if (user_intra_op_threadpool != nullptr) { in KernelFallbackCompatRequestState()
94 /*isolate_session_state=*/false, user_intra_op_threadpool); in KernelFallbackCompatRequestState()
106 tensorflow::thread::ThreadPoolInterface* user_intra_op_threadpool, in KernelFallbackCompatRequestState() argument
130 runner_table, resource_array, user_intra_op_threadpool,
H A Dkernel_fallback_execute_compat.h52 tensorflow::thread::ThreadPoolInterface* user_intra_op_threadpool = nullptr,
62 tensorflow::thread::ThreadPoolInterface* user_intra_op_threadpool = nullptr,
H A Dkernel_fallback_compat_request_state.h82 tensorflow::thread::ThreadPoolInterface* user_intra_op_threadpool,
92 tensorflow::thread::ThreadPoolInterface* user_intra_op_threadpool,
H A Dkernel_fallback_execute_compat.cc113 tensorflow::thread::ThreadPoolInterface* user_intra_op_threadpool, in SetUpKernelFallbackCompatRequestContext() argument
130 runner_table, resource_array, user_intra_op_threadpool, model_metadata, in SetUpKernelFallbackCompatRequestContext()
139 tensorflow::thread::ThreadPoolInterface* user_intra_op_threadpool, in SetUpKernelFallbackCompatRequestContext() argument
160 runner_table, resource_array, user_intra_op_threadpool, in SetUpKernelFallbackCompatRequestContext()
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/util/
H A Dfallback_test_util.cc27 tensorflow::thread::ThreadPoolInterface* user_intra_op_threadpool) { in CreateFallbackTestExecutionContext() argument
56 eager_context->pflr(), user_intra_op_threadpool); in CreateFallbackTestExecutionContext()
H A Dfallback_test_util.h27 tensorflow::thread::ThreadPoolInterface* user_intra_op_threadpool =
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dsingle_threaded_executor.cc301 if (args.user_intra_op_threadpool != nullptr) { in Run()
304 /*isolate_session_state=*/false, args.user_intra_op_threadpool); in Run()
H A Dexecutor.h104 thread::ThreadPoolInterface* user_intra_op_threadpool = nullptr; member
H A Dexecutor.cc444 if (args.user_intra_op_threadpool != nullptr) { in ExecutorState()
447 device->name(), device, false, false, args.user_intra_op_threadpool); in ExecutorState()
H A Dsingle_threaded_executor_test.cc193 args.user_intra_op_threadpool = &dummy_thread_pool; in TEST_F()
H A Dfunction.cc1027 exec_args->user_intra_op_threadpool = run_opts.user_intra_op_threadpool; in ExecutorArgsFromOptions()
H A Ddirect_session.cc668 args.user_intra_op_threadpool = threadpool_options.intra_op_threadpool; in RunInternal()
728 args->user_intra_op_threadpool = in RunInternal()
H A Dfunction_test.cc2074 opts.user_intra_op_threadpool = &UserIntraOpThreadPoolOp::dummy_thread_pool(); in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dfunction.h878 thread::ThreadPoolInterface* user_intra_op_threadpool = nullptr; member