Home
last modified time | relevance | path

Searched refs:run_handler_thread_pool (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Drun_handler_test.cc121 internal::RunHandlerThreadPool run_handler_thread_pool( in TEST() local
130 run_handler_thread_pool.AddWorkToQueue(&tws, /*is_blocking=*/true, fn); in TEST()
132 run_handler_thread_pool.AddWorkToQueue(&tws, /*is_blocking=*/true, fn2); in TEST()
139 run_handler_thread_pool.AddWorkToQueue(&tws, /*is_blocking=*/false, fn); in TEST()
141 run_handler_thread_pool.AddWorkToQueue(&tws, /*is_blocking=*/false, fn2); in TEST()
154 internal::RunHandlerThreadPool run_handler_thread_pool( in TEST() local
168 run_handler_thread_pool.AddWorkToQueue(thread_work_sources[2], in TEST()
171 run_handler_thread_pool.AddWorkToQueue(thread_work_sources[2], in TEST()
174 run_handler_thread_pool.AddWorkToQueue(thread_work_sources[3], in TEST()
177 run_handler_thread_pool.AddWorkToQueue(thread_work_sources[3], in TEST()
[all …]
Drun_handler.cc817 internal::RunHandlerThreadPool* run_handler_thread_pool() { in run_handler_thread_pool() function in tensorflow::RunHandlerPool::Impl
983 int num_threads = run_handler_thread_pool()->NumThreads(); in RecomputePoolStats()
984 int num_blocking_threads = run_handler_thread_pool()->NumBlockingThreads(); in RecomputePoolStats()
992 run_handler_thread_pool()->SetThreadWorkSources( in RecomputePoolStats()
1001 run_handler_thread_pool()->SetThreadWorkSources( in RecomputePoolStats()
1035 return run_handler_impl_->pool_impl_->run_handler_thread_pool()->NumThreads(); in NumThreads()
1039 return run_handler_impl_->pool_impl_->run_handler_thread_pool() in CurrentThreadId()
1056 pool_impl_->run_handler_thread_pool()->AddWorkToQueue(tws(), true, in ScheduleInterOpClosure()
1062 pool_impl_->run_handler_thread_pool()->AddWorkToQueue(tws(), false, in ScheduleIntraOpClosure()