Searched refs:EigenThreadPoolWrapper (Results 1 – 3 of 3) sorted by relevance
65 class EigenThreadPoolWrapper : public Eigen::ThreadPoolInterface { class68 explicit EigenThreadPoolWrapper(int num_threads) { in EigenThreadPoolWrapper() function in tflite::eigen_support::__anonf8036f140111::EigenThreadPoolWrapper74 ~EigenThreadPoolWrapper() override {} in ~EigenThreadPoolWrapper()104 new EigenThreadPoolWrapper(target_num_threads_)); in GetThreadPoolDevice()
113 struct LocalClientTestBase::EigenThreadPoolWrapper { struct in xla::LocalClientTestBase114 explicit EigenThreadPoolWrapper() in EigenThreadPoolWrapper() argument127 thread_pool_wrapper_(new EigenThreadPoolWrapper()) { in LocalClientTestBase()
81 struct EigenThreadPoolWrapper;142 std::unique_ptr<EigenThreadPoolWrapper> thread_pool_wrapper_;