HomeSort by relevance Sort by last modified time
    Searched defs:EigenThreadPoolWrapper (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/core/common_runtime/
eigen_thread_pool.h 24 class EigenThreadPoolWrapper : public Eigen::ThreadPoolInterface {
26 explicit EigenThreadPoolWrapper(thread::ThreadPool* pool) : pool_(pool) {}
27 ~EigenThreadPoolWrapper() override {}
  /external/tensorflow/tensorflow/compiler/xla/service/
backend.cc 65 struct Backend::EigenThreadPoolWrapper {
66 explicit EigenThreadPoolWrapper(const int num_threads)
69 wrapper(new tensorflow::EigenThreadPoolWrapper(pool.get())),
74 std::unique_ptr<tensorflow::EigenThreadPoolWrapper> wrapper;
152 new EigenThreadPoolWrapper(num_threads));
hlo_runner.cc 97 struct HloRunner::EigenThreadPoolWrapper {
98 std::unique_ptr<EigenThreadPoolWrapper> pool;
  /external/tensorflow/tensorflow/compiler/xla/tests/
local_client_test_base.cc 105 struct LocalClientTestBase::EigenThreadPoolWrapper {
106 explicit EigenThreadPoolWrapper()
109 wrapper(new tensorflow::EigenThreadPoolWrapper(pool.get())),
114 std::unique_ptr<tensorflow::EigenThreadPoolWrapper> wrapper;
122 thread_pool_wrapper_(new EigenThreadPoolWrapper()) {
  /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
multithreaded_conv.h 38 class EigenThreadPoolWrapper : public Eigen::ThreadPoolInterface {
40 explicit EigenThreadPoolWrapper(Eigen::ThreadPool* pool) : pool_(pool) {}
41 ~EigenThreadPoolWrapper() override {}
60 static EigenThreadPoolWrapper* thread_pool_wrapper =
61 new EigenThreadPoolWrapper(tp);
  /external/tensorflow/tensorflow/core/grappler/optimizers/
constant_folding.cc 49 class EigenThreadPoolWrapper : public Eigen::ThreadPoolInterface {
51 explicit EigenThreadPoolWrapper(thread::ThreadPool* pool) : pool_(pool) {}
52 ~EigenThreadPoolWrapper() override {}
70 new EigenThreadPoolWrapper(eigen_worker_threads_.workers));
    [all...]

Completed in 153 milliseconds