Home
last modified time | relevance | path

Searched refs:RunHandlerWorkQueue (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/run_handler_thread_pool/
H A Drun_handler.h412 class RunHandlerWorkQueue : public tensorflow::tfrt_stub::WorkQueueInterface {
414 explicit RunHandlerWorkQueue(std::unique_ptr<RunHandler> run_handler) in RunHandlerWorkQueue() function
418 ~RunHandlerWorkQueue() override = default;
H A Drun_handler.cc1062 int RunHandlerWorkQueue::GetParallelismLevel() const { in GetParallelismLevel()
1066 void RunHandlerWorkQueue::AddTask(TaskFunction work) { in AddTask()
1071 Optional<TaskFunction> RunHandlerWorkQueue::AddBlockingTask( in AddBlockingTask()
1078 void RunHandlerWorkQueue::Await(ArrayRef<RCReference<AsyncValue>> values) { in Await()
1082 bool RunHandlerWorkQueue::IsInWorkerThread() const { in IsInWorkerThread()
H A Drun_handler_concurrent_work_queue.cc76 return {std::make_unique<RunHandlerWorkQueue>(std::move(handler))}; in InitializeRequest()