Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/run_handler_thread_pool/
H A Drun_handler_test.cc150 internal::RunHandlerThreadPool run_handler_thread_pool( in TEST_P() local
168 run_handler_thread_pool.AddWorkToQueue(&tws, /*is_blocking=*/true, in TEST_P()
171 run_handler_thread_pool.AddWorkToQueue(&tws, /*is_blocking=*/true, in TEST_P()
179 run_handler_thread_pool.AddWorkToQueue(&tws, /*is_blocking=*/false, in TEST_P()
182 run_handler_thread_pool.AddWorkToQueue(&tws, /*is_blocking=*/false, in TEST_P()
196 internal::RunHandlerThreadPool run_handler_thread_pool( in TEST_P() local
219 run_handler_thread_pool.AddWorkToQueue( in TEST_P()
222 run_handler_thread_pool.AddWorkToQueue( in TEST_P()
225 run_handler_thread_pool.AddWorkToQueue( in TEST_P()
228 run_handler_thread_pool.AddWorkToQueue( in TEST_P()
[all …]
H A Drun_handler.cc763 internal::RunHandlerThreadPool* run_handler_thread_pool() { in run_handler_thread_pool() function in tfrt::tf::RunHandlerPool::Impl
955 int num_threads = run_handler_thread_pool()->NumThreads(); in RecomputePoolStats()
956 int num_blocking_threads = run_handler_thread_pool()->NumBlockingThreads(); in RecomputePoolStats()
960 run_handler_thread_pool()->SetThreadWorkSources(i, version, in RecomputePoolStats()
997 pool_impl_->run_handler_thread_pool()->AddWorkToQueue(tws(), true, in ScheduleInterOpClosure()
1003 pool_impl_->run_handler_thread_pool()->AddWorkToQueue(tws(), false, in ScheduleIntraOpClosure()
1016 return run_handler_->pool_impl_->run_handler_thread_pool()->NumThreads(); in NumThreads()
1020 return run_handler_->pool_impl_->run_handler_thread_pool()->CurrentThreadId(); in CurrentThreadId()
1050 return impl_->pool_impl()->run_handler_thread_pool()->NumThreads(); in NumThreads()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A 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 …]
H A Drun_handler.cc825 internal::RunHandlerThreadPool* run_handler_thread_pool() { in run_handler_thread_pool() function in tensorflow::RunHandlerPool::Impl
995 int num_threads = run_handler_thread_pool()->NumThreads(); in RecomputePoolStats()
996 int num_blocking_threads = run_handler_thread_pool()->NumBlockingThreads(); in RecomputePoolStats()
1004 run_handler_thread_pool()->SetThreadWorkSources( in RecomputePoolStats()
1013 run_handler_thread_pool()->SetThreadWorkSources( in RecomputePoolStats()
1047 return run_handler_impl_->pool_impl_->run_handler_thread_pool()->NumThreads(); in NumThreads()
1051 return run_handler_impl_->pool_impl_->run_handler_thread_pool() in CurrentThreadId()
1068 pool_impl_->run_handler_thread_pool()->AddWorkToQueue(tws(), true, in ScheduleInterOpClosure()
1074 pool_impl_->run_handler_thread_pool()->AddWorkToQueue(tws(), false, in ScheduleIntraOpClosure()
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/saved_model/tests/
H A DBUILD609 "//tensorflow/core/tfrt/run_handler_thread_pool:run_handler_concurrent_work_queue",
/aosp_15_r20/external/tensorflow/
H A D.bazelrc671 …tests/core_runtime,tensorflow/core/tfrt/gpu,tensorflow/core/tfrt/run_handler_thread_pool,tensorflo…
674 …tests/core_runtime,tensorflow/core/tfrt/gpu,tensorflow/core/tfrt/run_handler_thread_pool,tensorflo…
/aosp_15_r20/out/.module_paths/
Dfiles.db[all...]