| /aosp_15_r20/external/tensorflow/tensorflow/cc/client/ |
| H A D | client_session_test.cc | 152 auto intra_op_threadpool = in TEST() local
154 ASSERT_EQ(intra_op_threadpool->GetNumScheduleCalled(), 0); in TEST()
158 threadPoolOptions.intra_op_threadpool = intra_op_threadpool.get(); in TEST()
220 auto intra_op_threadpool = in TEST() local
222 ASSERT_EQ(intra_op_threadpool->GetNumScheduleCalled(), 0); in TEST()
226 threadPoolOptions.intra_op_threadpool = intra_op_threadpool.get(); in TEST()
239 intra_op_threadpool->Schedule([&barrier, num_threads]() { in TEST()
253 ASSERT_GT(intra_op_threadpool->GetNumScheduleCalled(), 0); in TEST()
257 intra_op_threadpool.reset(); in TEST()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/runtime/ |
| H A D | tf_threadpool_concurrent_work_queue.cc | 39 ThreadPoolInterface** intra_op_threadpool) const { in InitializeRequest()
40 DCHECK(intra_op_threadpool); in InitializeRequest()
41 *intra_op_threadpool = intra_op_threadpool_; in InitializeRequest()
93 TfThreadPool intra_op_threadpool; in CreateDefaultTfThreadPoolWorkQueue() member
97 intra_op_threadpool("default_work_queue_intra", in CreateDefaultTfThreadPoolWorkQueue()
105 &thread_pools->intra_op_threadpool), in CreateDefaultTfThreadPoolWorkQueue()
|
| H A D | tf_threadpool_concurrent_work_queue.h | 38 tensorflow::thread::ThreadPoolInterface* intra_op_threadpool, in TfThreadPoolWorkQueue() argument
40 : TfThreadPoolWorkQueue(/*id=*/0, intra_op_threadpool, in TfThreadPoolWorkQueue()
44 int64_t id, tensorflow::thread::ThreadPoolInterface* intra_op_threadpool, in TfThreadPoolWorkQueue() argument
47 intra_op_threadpool_(intra_op_threadpool), in TfThreadPoolWorkQueue()
52 tensorflow::thread::ThreadPoolInterface** intra_op_threadpool)
|
| H A D | tf_threadpool_concurrent_work_queue_test.cc | 55 tensorflow::thread::ThreadPoolInterface* intra_op_threadpool = nullptr; in TEST_F() local
57 tf_threadpool_cwq_->InitializeRequest(&ctx_builder, &intra_op_threadpool); in TEST_F()
60 EXPECT_NE(intra_op_threadpool, nullptr); in TEST_F()
|
| H A D | work_queue_interface.h | 49 thread::ThreadPoolInterface** intra_op_threadpool) const { in InitializeRequest() argument
50 *intra_op_threadpool = nullptr; in InitializeRequest()
|
| H A D | work_queue_interface.cc | 83 thread::ThreadPoolInterface** intra_op_threadpool) const override { in InitializeRequest()
84 *intra_op_threadpool = intra_thread_pool_; in InitializeRequest()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/run_handler_thread_pool/ |
| H A D | run_handler_concurrent_work_queue.cc | 62 tensorflow::thread::ThreadPoolInterface** intra_op_threadpool) const { in InitializeRequest()
63 DCHECK(intra_op_threadpool); in InitializeRequest()
74 *intra_op_threadpool = handler->AsIntraThreadPoolInterface(); in InitializeRequest()
|
| H A D | run_handler_concurrent_work_queue_test.cc | 62 tensorflow::thread::ThreadPoolInterface* intra_op_threadpool = nullptr; in SetUp() local
64 pool_->InitializeRequest(&req_ctx_builder, &intra_op_threadpool); in SetUp()
|
| H A D | run_handler_concurrent_work_queue.h | 93 intra_op_threadpool) const override;
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| H A D | runtime_fallback_batch_tf_opkernels.cc | 497 auto* intra_op_threadpool = src_fallback_request_state->intra_op_threadpool(); in SetUpKernelFallbackCompatRequestContextForBatch() local
507 builder, device_manager, pflr, intra_op_threadpool, session_metadata, in SetUpKernelFallbackCompatRequestContextForBatch()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/graph_executor/ |
| H A D | graph_executor.cc | 99 tensorflow::thread::ThreadPoolInterface* intra_op_threadpool = nullptr; in SetUpRequestContext() local
105 &intra_op_threadpool)); in SetUpRequestContext()
120 &fallback_state.process_function_library_runtime(), intra_op_threadpool, in SetUpRequestContext()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/platform/ |
| H A D | threadpool_options.h | 29 thread::ThreadPoolInterface* intra_op_threadpool = nullptr; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | bef_thunk.cc | 340 tensorflow::thread::ThreadPoolInterface* intra_op_threadpool = nullptr; in ExecuteOnStream() local
342 &request_context_builder, &intra_op_threadpool)); in ExecuteOnStream()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| H A D | kernel_fallback_compat_request_state.h | 134 tensorflow::thread::ThreadPoolInterface* intra_op_threadpool() const { in intra_op_threadpool() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/ |
| H A D | tf_jitrt_kernels.cc | 183 if (LLVM_LIKELY(fallback->intra_op_threadpool())) in GetWorkerThreads()
184 return fallback->intra_op_threadpool(); in GetWorkerThreads()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | direct_session.cc | 668 args.user_intra_op_threadpool = threadpool_options.intra_op_threadpool; in RunInternal()
|