Searched defs:IntraOpThreadPool (Results 1 – 3 of 3) sorted by relevance
70 struct Backend::IntraOpThreadPool { struct in xla::Backend71 explicit IntraOpThreadPool(const int num_threads) in IntraOpThreadPool() function77 std::unique_ptr<tensorflow::thread::ThreadPool> pool;78 std::unique_ptr<Eigen::ThreadPoolDevice> device;
74 class IntraOpThreadPool : public ThreadPoolInterface { class76 explicit IntraOpThreadPool(int64_t num_threads) in IntraOpThreadPool() function in tensorflow::__anon4e1314fe0111::IntraOpThreadPool
120 TEST(RunHandlerUtilTest, IntraOpThreadPool) { in TEST() argument