Searched refs:IntraOpThreadPool (Results 1 – 4 of 4) sorted by relevance
70 struct Backend::IntraOpThreadPool { struct in xla::Backend71 explicit IntraOpThreadPool(const int num_threads) in IntraOpThreadPool() function140 intra_op_thread_pool_.reset(new IntraOpThreadPool(num_threads)); in Backend()
189 struct IntraOpThreadPool;190 std::unique_ptr<IntraOpThreadPool> intra_op_thread_pool_;
74 class IntraOpThreadPool : public ThreadPoolInterface { class76 explicit IntraOpThreadPool(int64_t num_threads) in IntraOpThreadPool() function in tensorflow::__anonc9234a7e0111::IntraOpThreadPool94 : intra_op_(std::make_unique<IntraOpThreadPool>(num_threads)) { in RuntimeFallbackExecutor()
120 TEST(RunHandlerUtilTest, IntraOpThreadPool) { in TEST() argument