Home
last modified time | relevance | path

Searched defs:EigenThreadPoolWrapper (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dlocal_client_test_base.cc113 struct LocalClientTestBase::EigenThreadPoolWrapper { struct in xla::LocalClientTestBase
114 explicit EigenThreadPoolWrapper() in EigenThreadPoolWrapper() argument
120 std::unique_ptr<tensorflow::thread::ThreadPool> pool;
121 std::unique_ptr<Eigen::ThreadPoolDevice> device;
/external/tensorflow/tensorflow/lite/kernels/
Deigen_support.cc65 class EigenThreadPoolWrapper : public Eigen::ThreadPoolInterface { class
68 explicit EigenThreadPoolWrapper(int num_threads) { in EigenThreadPoolWrapper() function in tflite::eigen_support::__anonf8036f140111::EigenThreadPoolWrapper