Home
last modified time | relevance | path

Searched refs:RunHandlerThreadWorkQueue (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/run_handler_thread_pool/
H A Drun_handler_concurrent_work_queue.cc28 RunHandlerThreadWorkQueue::RunHandlerThreadWorkQueue(const Options& options) in RunHandlerThreadWorkQueue() function in tfrt::tf::RunHandlerThreadWorkQueue
60 RunHandlerThreadWorkQueue::InitializeRequest( in InitializeRequest()
79 void RunHandlerThreadWorkQueue::AddTask(TaskFunction work) { in AddTask()
83 Optional<TaskFunction> RunHandlerThreadWorkQueue::AddBlockingTask( in AddBlockingTask()
93 void RunHandlerThreadWorkQueue::Quiesce() { in Quiesce()
99 void RunHandlerThreadWorkQueue::Await( in Await()
104 bool RunHandlerThreadWorkQueue::IsInWorkerThread() const { in IsInWorkerThread()
H A Drun_handler_concurrent_work_queue_test.cc45 RunHandlerThreadWorkQueue::Options options; in SetUp()
49 pool_ = std::make_unique<RunHandlerThreadWorkQueue>(options); in SetUp()
72 std::unique_ptr<RunHandlerThreadWorkQueue> pool_;
178 RunHandlerThreadWorkQueue::Options options; in TEST_F()
183 auto queue = std::make_unique<RunHandlerThreadWorkQueue>(options); in TEST_F()
H A Drun_handler_concurrent_work_queue.h33 class RunHandlerThreadWorkQueue
79 explicit RunHandlerThreadWorkQueue(const Options& options);
80 ~RunHandlerThreadWorkQueue() override {} in ~RunHandlerThreadWorkQueue()
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/saved_model/tests/
H A Dsaved_model_test.cc487 tfrt::tf::RunHandlerThreadWorkQueue::Options queue_options; in TEST()
493 std::make_unique<tfrt::tf::RunHandlerThreadWorkQueue>(queue_options)); in TEST()
550 tfrt::tf::RunHandlerThreadWorkQueue::Options queue_options; in TEST()
555 tfrt::tf::RunHandlerThreadWorkQueue run_handler_queue(queue_options); in TEST()