Searched refs:RunHandlerThreadPool (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | run_handler.cc | 313 RunHandlerThreadPool::RunHandlerThreadPool( in RunHandlerThreadPool() function in tensorflow::internal::RunHandlerThreadPool
344 RunHandlerThreadPool::~RunHandlerThreadPool() { in ~RunHandlerThreadPool()
357 void RunHandlerThreadPool::Start() { in Start()
384 void RunHandlerThreadPool::StartOneThreadForTesting() { in StartOneThreadForTesting()
391 void RunHandlerThreadPool::AddWorkToQueue(ThreadWorkSource* tws, in AddWorkToQueue()
408 void RunHandlerThreadPool::SetThreadWorkSources( in SetThreadWorkSources()
451 RunHandlerThreadPool::PerThread* RunHandlerThreadPool::GetPerThread() { in GetPerThread()
452 thread_local RunHandlerThreadPool::PerThread per_thread_; in GetPerThread()
453 RunHandlerThreadPool::PerThread* pt = &per_thread_; in GetPerThread()
457 int RunHandlerThreadPool::CurrentThreadId() const { in CurrentThreadId()
[all …]
|
| H A D | run_handler_test.cc | 116 TEST(RunHandlerThreadPool, EnqueueTask) { in TEST() argument
121 internal::RunHandlerThreadPool run_handler_thread_pool( in TEST()
149 TEST(RunHandlerThreadPool, FindTask) { in TEST() argument
154 internal::RunHandlerThreadPool run_handler_thread_pool( in TEST()
429 TEST(RunHandlerThreadPool, RoundRobinExecution) { in TEST() argument
440 internal::RunHandlerThreadPool* run_handler_thread_pool = in TEST()
441 new internal::RunHandlerThreadPool( in TEST()
498 TEST(RunHandlerThreadPool, MultipleSubThreadPool) { in TEST() argument
511 internal::RunHandlerThreadPool* run_handler_thread_pool = in TEST()
512 new internal::RunHandlerThreadPool( in TEST()
|
| H A D | run_handler.h | 215 class RunHandlerThreadPool {
219 RunHandlerThreadPool* pool; // Parent pool, or null for normal threads.
223 RunHandlerThreadPool(int num_blocking_threads, int num_non_blocking_threads,
229 ~RunHandlerThreadPool();
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/run_handler_thread_pool/ |
| H A D | run_handler.cc | 335 RunHandlerThreadPool::RunHandlerThreadPool( in RunHandlerThreadPool() function in tfrt::tf::internal::RunHandlerThreadPool
373 RunHandlerThreadPool::~RunHandlerThreadPool() { in ~RunHandlerThreadPool()
386 void RunHandlerThreadPool::Start() { in Start()
405 void RunHandlerThreadPool::StartOneThreadForTesting() { in StartOneThreadForTesting()
412 void RunHandlerThreadPool::AddWorkToQueue(ThreadWorkSource* tws, in AddWorkToQueue()
423 void RunHandlerThreadPool::SetThreadWorkSources( in SetThreadWorkSources()
444 RunHandlerThreadPool::PerThread* RunHandlerThreadPool::GetPerThread() { in GetPerThread()
445 thread_local RunHandlerThreadPool::PerThread per_thread_; in GetPerThread()
446 RunHandlerThreadPool::PerThread* pt = &per_thread_; in GetPerThread()
450 int RunHandlerThreadPool::CurrentThreadId() const { in CurrentThreadId()
[all …]
|
| H A D | run_handler_test.cc | 150 internal::RunHandlerThreadPool run_handler_thread_pool( in TEST_P()
151 internal::RunHandlerThreadPool::Options( in TEST_P()
196 internal::RunHandlerThreadPool run_handler_thread_pool( in TEST_P()
197 internal::RunHandlerThreadPool::Options( in TEST_P()
485 internal::RunHandlerThreadPool* run_handler_thread_pool = in TEST_P()
486 new internal::RunHandlerThreadPool( in TEST_P()
487 internal::RunHandlerThreadPool::Options( in TEST_P()
556 internal::RunHandlerThreadPool* run_handler_thread_pool = in TEST_P()
557 new internal::RunHandlerThreadPool( in TEST_P()
558 internal::RunHandlerThreadPool::Options( in TEST_P()
|
| H A D | run_handler.h | 284 class RunHandlerThreadPool {
320 RunHandlerThreadPool* pool; // Parent pool, or null for normal threads.
324 RunHandlerThreadPool(Options options, tensorflow::Env* env,
330 ~RunHandlerThreadPool();
|