Searched refs:TfThreadPoolWorkQueue (Results 1 – 5 of 5) sorted by relevance
37 TfThreadPoolWorkQueue::InitializeRequest( in InitializeRequest()43 return {std::make_unique<TfThreadPoolWorkQueue>(request_context_builder->id(), in InitializeRequest()48 void TfThreadPoolWorkQueue::AddTask(tfrt::TaskFunction work) { in AddTask()57 llvm::Optional<tfrt::TaskFunction> TfThreadPoolWorkQueue::AddBlockingTask( in AddBlockingTask()63 void TfThreadPoolWorkQueue::Quiesce() { in Quiesce()70 void TfThreadPoolWorkQueue::Await( in Await()84 bool TfThreadPoolWorkQueue::IsInWorkerThread() const { in IsInWorkerThread()89 std::unique_ptr<TfThreadPoolWorkQueue> CreateDefaultTfThreadPoolWorkQueue( in CreateDefaultTfThreadPoolWorkQueue()101 class Wrapper : public TfThreadPoolWorkQueue { in CreateDefaultTfThreadPoolWorkQueue()104 : TfThreadPoolWorkQueue(&thread_pools->inter_op_threadpool, in CreateDefaultTfThreadPoolWorkQueue()
35 class TfThreadPoolWorkQueue : public WorkQueueInterface {37 TfThreadPoolWorkQueue( in TfThreadPoolWorkQueue() function40 : TfThreadPoolWorkQueue(/*id=*/0, intra_op_threadpool, in TfThreadPoolWorkQueue()43 TfThreadPoolWorkQueue( in TfThreadPoolWorkQueue() function80 std::unique_ptr<TfThreadPoolWorkQueue> CreateDefaultTfThreadPoolWorkQueue(
41 std::unique_ptr<TfThreadPoolWorkQueue> tf_threadpool_cwq_;
95 tensorflow::tfrt_stub::TfThreadPoolWorkQueue* GetTfThreadPoolWorkQueue() { in GetTfThreadPoolWorkQueue()115 std::unique_ptr<tensorflow::tfrt_stub::TfThreadPoolWorkQueue>
58 std::make_unique<tensorflow::tfrt_stub::TfThreadPoolWorkQueue>( in TfrtContext()