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.cc80 thread_pool_wrapper_.reset(new EigenThreadPoolWrapper( in GetThreadPoolDevice()
82 device_.reset(new Eigen::ThreadPoolDevice(thread_pool_wrapper_.get(), in GetThreadPoolDevice()
96 thread_pool_wrapper_.reset(); in SetNumThreads()
104 std::unique_ptr<Eigen::ThreadPoolInterface> thread_pool_wrapper_; member in tflite::eigen_support::__anon0b33ddad0111::LazyEigenThreadPoolHolder
/external/tensorflow/tensorflow/compiler/xla/tests/
Dlocal_client_test_base.cc123 thread_pool_wrapper_(new EigenThreadPoolWrapper()) { in LocalClientTestBase()
152 run_options.set_intra_op_thread_pool(thread_pool_wrapper_->device.get()); in DefaultExecutableRunOptions()
Dlocal_client_test_base.h131 std::unique_ptr<EigenThreadPoolWrapper> thread_pool_wrapper_; variable