Searched refs:ThreadPoolInterfaceWrapper (Results 1 – 3 of 3) sorted by relevance
42 class ThreadPoolInterfaceWrapper45 explicit ThreadPoolInterfaceWrapper(Eigen::ThreadPoolInterface* thread_pool) in ThreadPoolInterfaceWrapper() function109 std::unique_ptr<ThreadPoolInterfaceWrapper> eager_ctx_thread_pool_;114 std::unique_ptr<ThreadPoolInterfaceWrapper> local_thread_pool_wrapper_;
49 eager_ctx_thread_pool_ = std::make_unique<ThreadPoolInterfaceWrapper>( in TfrtContext()54 local_thread_pool_wrapper_ = std::make_unique<ThreadPoolInterfaceWrapper>( in TfrtContext()
753 class ThreadPoolInterfaceWrapper : public thread::ThreadPoolInterface { class in tensorflow::RunHandler::Impl755 explicit ThreadPoolInterfaceWrapper(Impl* run_handler_impl) in ThreadPoolInterfaceWrapper() function in tensorflow::RunHandler::Impl::ThreadPoolInterfaceWrapper757 ~ThreadPoolInterfaceWrapper() override {} in ~ThreadPoolInterfaceWrapper()1046 int RunHandler::Impl::ThreadPoolInterfaceWrapper::NumThreads() const { in NumThreads()1050 int RunHandler::Impl::ThreadPoolInterfaceWrapper::CurrentThreadId() const { in CurrentThreadId()1055 void RunHandler::Impl::ThreadPoolInterfaceWrapper::Schedule( in Schedule()1062 thread_pool_interface_.reset(new ThreadPoolInterfaceWrapper(this)); in Impl()