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;
30 class EigenThreadPoolWrapper : public Eigen::ThreadPoolInterface { class32 explicit EigenThreadPoolWrapper(thread::ThreadPool* pool) : pool_(pool) {} in EigenThreadPoolWrapper() function in tensorflow::grappler::__anon6915ea5a0111::EigenThreadPoolWrapper
62 class EigenThreadPoolWrapper : public Eigen::ThreadPoolInterface { class64 explicit EigenThreadPoolWrapper(thread::ThreadPool* pool) : pool_(pool) {} in EigenThreadPoolWrapper() function in tensorflow::grappler::__anon9e0e95fe0111::EigenThreadPoolWrapper
54 class EigenThreadPoolWrapper : public Eigen::ThreadPoolInterface { class57 explicit EigenThreadPoolWrapper(Eigen::ThreadPool* pool) : pool_(pool) {} in EigenThreadPoolWrapper() function in tflite::eigen_support::__anon0b33ddad0111::EigenThreadPoolWrapper
26 explicit EigenThreadPoolWrapper(thread::ThreadPool* pool) : pool_(pool) {} in EigenThreadPoolWrapper() function
71 class EigenThreadPoolWrapper : public Eigen::ThreadPoolInterface { class73 explicit EigenThreadPoolWrapper(tensorflow::thread::ThreadPool* pool) in EigenThreadPoolWrapper() function in xla::__anonfbcb6a4f0111::EigenThreadPoolWrapper