Home
last modified time | relevance | path

Searched refs:AddWorkToQueue (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/run_handler_thread_pool/
H A Drun_handler_test.cc168 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()
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()
268 run_handler_thread_pool.AddWorkToQueue( in TEST_P()
297 run_handler_thread_pool.AddWorkToQueue( in TEST_P()
[all …]
H A Drun_handler.cc412 void RunHandlerThreadPool::AddWorkToQueue(ThreadWorkSource* tws, in AddWorkToQueue() function in tfrt::tf::internal::RunHandlerThreadPool
997 pool_impl_->run_handler_thread_pool()->AddWorkToQueue(tws(), true, in ScheduleInterOpClosure()
1003 pool_impl_->run_handler_thread_pool()->AddWorkToQueue(tws(), false, in ScheduleIntraOpClosure()
H A Drun_handler.h336 void AddWorkToQueue(ThreadWorkSource* tws, bool is_blocking, TaskFunction fn);
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Drun_handler_test.cc130 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()
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()
217 run_handler_thread_pool.AddWorkToQueue(thread_work_sources[3], in TEST()
246 run_handler_thread_pool.AddWorkToQueue(thread_work_sources[2], in TEST()
[all …]
H A Drun_handler.cc391 void RunHandlerThreadPool::AddWorkToQueue(ThreadWorkSource* tws, in AddWorkToQueue() function in tensorflow::internal::RunHandlerThreadPool
1068 pool_impl_->run_handler_thread_pool()->AddWorkToQueue(tws(), true, in ScheduleInterOpClosure()
1074 pool_impl_->run_handler_thread_pool()->AddWorkToQueue(tws(), false, in ScheduleIntraOpClosure()
H A Drun_handler.h235 void AddWorkToQueue(ThreadWorkSource* tws, bool is_blocking,