Home
last modified time | relevance | path

Searched defs:intra_op_threadpool (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/runtime/
H A Dtf_threadpool_concurrent_work_queue.h38 tensorflow::thread::ThreadPoolInterface* intra_op_threadpool, in TfThreadPoolWorkQueue()
44 int64_t id, tensorflow::thread::ThreadPoolInterface* intra_op_threadpool, in TfThreadPoolWorkQueue()
H A Dwork_queue_interface.h49 thread::ThreadPoolInterface** intra_op_threadpool) const { in InitializeRequest()
H A Dtf_threadpool_concurrent_work_queue.cc93 TfThreadPool intra_op_threadpool; in CreateDefaultTfThreadPoolWorkQueue() member
H A Dtf_threadpool_concurrent_work_queue_test.cc55 tensorflow::thread::ThreadPoolInterface* intra_op_threadpool = nullptr; in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/cc/client/
H A Dclient_session_test.cc152 auto intra_op_threadpool = in TEST() local
220 auto intra_op_threadpool = in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/core/platform/
H A Dthreadpool_options.h29 thread::ThreadPoolInterface* intra_op_threadpool = nullptr; member
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
H A Dkernel_fallback_compat_request_state.h134 tensorflow::thread::ThreadPoolInterface* intra_op_threadpool() const { in intra_op_threadpool() function
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/run_handler_thread_pool/
H A Drun_handler_concurrent_work_queue_test.cc62 tensorflow::thread::ThreadPoolInterface* intra_op_threadpool = nullptr; in SetUp() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dbef_thunk.cc340 tensorflow::thread::ThreadPoolInterface* intra_op_threadpool = nullptr; in ExecuteOnStream() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/graph_executor/
H A Dgraph_executor.cc99 tensorflow::thread::ThreadPoolInterface* intra_op_threadpool = nullptr; in SetUpRequestContext() local
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
H A Druntime_fallback_batch_tf_opkernels.cc497 auto* intra_op_threadpool = src_fallback_request_state->intra_op_threadpool(); in SetUpKernelFallbackCompatRequestContextForBatch() local