Searched defs:EigenThreadPoolWrapper (Results 1 – 6 of 6) sorted by relevance
107 struct LocalClientTestBase::EigenThreadPoolWrapper { struct in xla::LocalClientTestBase108 explicit EigenThreadPoolWrapper() in EigenThreadPoolWrapper() function115 std::unique_ptr<tensorflow::thread::ThreadPool> pool;116 std::unique_ptr<tensorflow::EigenThreadPoolWrapper> wrapper;117 std::unique_ptr<Eigen::ThreadPoolDevice> device;
26 explicit EigenThreadPoolWrapper(thread::ThreadPool* pool) : pool_(pool) {} in EigenThreadPoolWrapper() function
30 class EigenThreadPoolWrapper : public Eigen::ThreadPoolInterface { class32 explicit EigenThreadPoolWrapper(thread::ThreadPool* pool) : pool_(pool) {} in EigenThreadPoolWrapper() function in tensorflow::grappler::__anondc970a030111::EigenThreadPoolWrapper
62 class EigenThreadPoolWrapper : public Eigen::ThreadPoolInterface { class64 explicit EigenThreadPoolWrapper(thread::ThreadPool* pool) : pool_(pool) {} in EigenThreadPoolWrapper() function in tensorflow::grappler::__anon118fb5a70111::EigenThreadPoolWrapper
54 class EigenThreadPoolWrapper : public Eigen::ThreadPoolInterface { class57 explicit EigenThreadPoolWrapper(Eigen::ThreadPool* pool) : pool_(pool) {} in EigenThreadPoolWrapper() function in tflite::eigen_support::__anon64c280760111::EigenThreadPoolWrapper
71 class EigenThreadPoolWrapper : public Eigen::ThreadPoolInterface { class73 explicit EigenThreadPoolWrapper(tensorflow::thread::ThreadPool* pool) in EigenThreadPoolWrapper() function in xla::__anonf38de3580111::EigenThreadPoolWrapper