HomeSort by relevance Sort by last modified time
    Searched refs:EigenThreadPoolWrapper (Results 1 - 15 of 15) 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 {}
local_device.cc 46 new EigenThreadPoolWrapper(eigen_worker_threads_.workers));
  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_runner.h 91 struct EigenThreadPoolWrapper;
93 std::unique_ptr<EigenThreadPoolWrapper> thread_pool_wrapper_;
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));
backend.h 159 struct EigenThreadPoolWrapper;
191 std::unique_ptr<EigenThreadPoolWrapper> intra_op_thread_pool_wrapper_;
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()) {
local_client_test_base.h 77 struct EigenThreadPoolWrapper;
136 std::unique_ptr<EigenThreadPoolWrapper> thread_pool_wrapper_;
fusion_test.cc     [all...]
  /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/compiler/xla/service/cpu/
cpu_runtime_test.cc 103 tensorflow::EigenThreadPoolWrapper tp(&pool);
  /external/tensorflow/tensorflow/core/kernels/
quantization_utils_test.cc 215 EigenThreadPoolWrapper wrapper(&threadpool);
297 EigenThreadPoolWrapper wrapper(&threadpool);
609 EigenThreadPoolWrapper wrapper(&threadpool);
640 EigenThreadPoolWrapper wrapper(&threadpool);
651 EigenThreadPoolWrapper wrapper(&threadpool);
719 EigenThreadPoolWrapper wrapper(&threadpool);
    [all...]
quantized_batch_norm_op_test.cc 221 EigenThreadPoolWrapper wrapper(&threadpool);
nn_ops_test.cc     [all...]
  /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 698 milliseconds