Home
last modified time | relevance | path

Searched refs:task_from_blocking_queue (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.cc233 [&](bool* task_from_blocking_queue, internal::Task* t) { in TEST_P() argument
240 task_from_blocking_queue, &tws); in TEST_P()
242 bool task_from_blocking_queue; in TEST_P() local
244 find_blocking_task_from_all_handlers(&task_from_blocking_queue, &t); in TEST_P()
245 EXPECT_EQ(task_from_blocking_queue, true); in TEST_P()
249 find_blocking_task_from_all_handlers(&task_from_blocking_queue, &t); in TEST_P()
250 EXPECT_EQ(task_from_blocking_queue, true); in TEST_P()
254 find_blocking_task_from_all_handlers(&task_from_blocking_queue, &t); in TEST_P()
255 EXPECT_EQ(task_from_blocking_queue, true); in TEST_P()
259 find_blocking_task_from_all_handlers(&task_from_blocking_queue, &t); in TEST_P()
[all …]
H A Drun_handler.cc477 bool* task_from_blocking_queue, ThreadWorkSource** tws) { in FindTask() argument
480 *task_from_blocking_queue = false; in FindTask()
495 *task_from_blocking_queue = true; in FindTask()
521 bool task_from_blocking_queue = true; in WorkerLoop() local
556 &task_from_blocking_queue, &tws); in WorkerLoop()
563 &task_from_blocking_queue, &tws); in WorkerLoop()
571 &task_from_blocking_queue, &tws); in WorkerLoop()
574 VLOG(2) << "Running " << (task_from_blocking_queue ? "inter" : "intra") in WorkerLoop()
576 tws->IncrementInflightTaskCount(task_from_blocking_queue); in WorkerLoop()
578 tws->DecrementInflightTaskCount(task_from_blocking_queue); in WorkerLoop()
H A Drun_handler.h363 bool* task_from_blocking_queue, ThreadWorkSource** tws);
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Drun_handler_test.cc182 [&](bool* task_from_blocking_queue, internal::Task* t) { in TEST() argument
189 task_from_blocking_queue, &tws); in TEST()
191 bool task_from_blocking_queue; in TEST() local
193 find_blocking_task_from_all_handlers(&task_from_blocking_queue, &t); in TEST()
194 EXPECT_EQ(task_from_blocking_queue, true); in TEST()
198 find_blocking_task_from_all_handlers(&task_from_blocking_queue, &t); in TEST()
199 EXPECT_EQ(task_from_blocking_queue, true); in TEST()
203 find_blocking_task_from_all_handlers(&task_from_blocking_queue, &t); in TEST()
204 EXPECT_EQ(task_from_blocking_queue, true); in TEST()
208 find_blocking_task_from_all_handlers(&task_from_blocking_queue, &t); in TEST()
[all …]
H A Drun_handler.cc494 bool* task_from_blocking_queue, ThreadWorkSource** tws) { in FindTask() argument
497 *task_from_blocking_queue = false; in FindTask()
512 *task_from_blocking_queue = true; in FindTask()
538 bool task_from_blocking_queue = true; in WorkerLoop() local
572 &task_from_blocking_queue, &tws); in WorkerLoop()
579 &task_from_blocking_queue, &tws); in WorkerLoop()
587 &task_from_blocking_queue, &tws); in WorkerLoop()
610 task_from_blocking_queue = false; in WorkerLoop()
619 task_from_blocking_queue = false; in WorkerLoop()
628 return strings::StrCat(task_from_blocking_queue ? "inter" : "intra", in WorkerLoop()
[all …]
H A Drun_handler.h265 bool* task_from_blocking_queue, ThreadWorkSource** tws);