Home
last modified time | relevance | path

Searched refs:intra_op_thread_pool_ (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dbackend.cc168 intra_op_thread_pool_.reset(new IntraOpThreadPool(num_threads)); in Backend()
180 if (intra_op_thread_pool_ == nullptr) { in eigen_intra_op_thread_pool_device()
183 return intra_op_thread_pool_->device.get(); in eigen_intra_op_thread_pool_device()
187 if (intra_op_thread_pool_ == nullptr) { in eigen_intra_op_thread_pool()
190 return intra_op_thread_pool_->pool.get(); in eigen_intra_op_thread_pool()
Dbackend.h186 std::unique_ptr<IntraOpThreadPool> intra_op_thread_pool_; variable
/external/tensorflow/tensorflow/compiler/xla/
Dexecutable_run_options.cc60 intra_op_thread_pool_ = intra_op_thread_pool; in set_intra_op_thread_pool()
66 return intra_op_thread_pool_; in intra_op_thread_pool()
Dexecutable_run_options.h88 const Eigen::ThreadPoolDevice* intra_op_thread_pool_ = nullptr; variable