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.cc111 struct LocalClientTestBase::EigenThreadPoolWrapper { struct in xla::LocalClientTestBase
112 explicit EigenThreadPoolWrapper() in EigenThreadPoolWrapper() function
118 std::unique_ptr<tensorflow::thread::ThreadPool> pool;
119 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::__anon8b445db20111::EigenThreadPoolWrapper