OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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