Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Deigen_support.cc103 thread_pool_wrapper_.reset( in GetThreadPoolDevice()
105 device_.reset(new Eigen::ThreadPoolDevice(thread_pool_wrapper_.get(), in GetThreadPoolDevice()
118 thread_pool_wrapper_.reset(); in SetNumThreads()
126 std::unique_ptr<Eigen::ThreadPoolInterface> thread_pool_wrapper_; member in tflite::eigen_support::__anonf8036f140111::LazyEigenThreadPoolHolder
/external/tensorflow/tensorflow/compiler/xla/tests/
Dlocal_client_test_base.cc127 thread_pool_wrapper_(new EigenThreadPoolWrapper()) { in LocalClientTestBase()
158 run_options.set_intra_op_thread_pool(thread_pool_wrapper_->device.get()); in DefaultExecutableRunOptions()
Dlocal_client_test_base.h142 std::unique_ptr<EigenThreadPoolWrapper> thread_pool_wrapper_; variable