| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | eigen_thread_pool.h | 24 class EigenThreadPoolWrapper : public Eigen::ThreadPoolInterface {
26 explicit EigenThreadPoolWrapper(thread::ThreadPool* pool) : pool_(pool) {} in EigenThreadPoolWrapper() function
27 ~EigenThreadPoolWrapper() override {} in ~EigenThreadPoolWrapper()
|
| D | single_threaded_cpu_device.cc | 51 new EigenThreadPoolWrapper(eigen_worker_threads_.workers)); in SingleThreadedCpuDevice()
|
| D | local_device.cc | 100 new EigenThreadPoolWrapper(eigen_worker_threads_.workers)); in EigenThreadPoolInfo()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | eigen_support.cc | 54 class EigenThreadPoolWrapper : public Eigen::ThreadPoolInterface { class
57 explicit EigenThreadPoolWrapper(Eigen::ThreadPool* pool) : pool_(pool) {} in EigenThreadPoolWrapper() function in tflite::eigen_support::__anon64c280760111::EigenThreadPoolWrapper
58 ~EigenThreadPoolWrapper() override {} in ~EigenThreadPoolWrapper()
80 thread_pool_wrapper_.reset(new EigenThreadPoolWrapper( in GetThreadPoolDevice()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | evaluation_utils.cc | 30 class EigenThreadPoolWrapper : public Eigen::ThreadPoolInterface { class
32 explicit EigenThreadPoolWrapper(thread::ThreadPool* pool) : pool_(pool) {} in EigenThreadPoolWrapper() function in tensorflow::grappler::__anondc970a030111::EigenThreadPoolWrapper
33 ~EigenThreadPoolWrapper() override {} in ~EigenThreadPoolWrapper()
58 new EigenThreadPoolWrapper(eigen_worker_threads_.workers)); in DeviceSimple()
|
| D | constant_folding.cc | 62 class EigenThreadPoolWrapper : public Eigen::ThreadPoolInterface { class
64 explicit EigenThreadPoolWrapper(thread::ThreadPool* pool) : pool_(pool) {} in EigenThreadPoolWrapper() function in tensorflow::grappler::__anon118fb5a70111::EigenThreadPoolWrapper
65 ~EigenThreadPoolWrapper() override {} in ~EigenThreadPoolWrapper()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | local_client_test_base.cc | 107 struct LocalClientTestBase::EigenThreadPoolWrapper { struct in xla::LocalClientTestBase
108 explicit EigenThreadPoolWrapper() in EigenThreadPoolWrapper() argument
111 wrapper(new tensorflow::EigenThreadPoolWrapper(pool.get())), in EigenThreadPoolWrapper()
116 std::unique_ptr<tensorflow::EigenThreadPoolWrapper> wrapper; argument
123 thread_pool_wrapper_(new EigenThreadPoolWrapper()) { in LocalClientTestBase()
|
| D | local_client_test_base.h | 76 struct EigenThreadPoolWrapper;
131 std::unique_ptr<EigenThreadPoolWrapper> thread_pool_wrapper_;
|
| D | fusion_test.cc | 899 tensorflow::EigenThreadPoolWrapper tp(&pool); in BM_ParallelFusion()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | backend.cc | 71 class EigenThreadPoolWrapper : public Eigen::ThreadPoolInterface { class
73 explicit EigenThreadPoolWrapper(tensorflow::thread::ThreadPool* pool) in EigenThreadPoolWrapper() function in xla::__anonf38de3580111::EigenThreadPoolWrapper
75 ~EigenThreadPoolWrapper() override {} in ~EigenThreadPoolWrapper()
95 wrapper(new EigenThreadPoolWrapper(pool.get())), in IntraOpThreadPool()
100 std::unique_ptr<EigenThreadPoolWrapper> wrapper;
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base_test.cc | 32 EigenThreadPoolWrapper wrapper(&pool); in TEST()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantization_utils_test.cc | 215 EigenThreadPoolWrapper wrapper(&threadpool); in TestRequantizeManyInNewRangeEigenVsNonEigen()
297 EigenThreadPoolWrapper wrapper(&threadpool); in TimeRequantizeManyInNewRange()
609 EigenThreadPoolWrapper wrapper(&threadpool); in TestRequantizeManyInNewRange32To8BitUsingEigen()
640 EigenThreadPoolWrapper wrapper(&threadpool); in TestFloatToQuantizedInPlaceUsingEigen()
651 EigenThreadPoolWrapper wrapper(&threadpool); in TestOverflowWithEigen()
719 EigenThreadPoolWrapper wrapper(&threadpool); in TestQuantizedToFloatInPlaceUsingEigen()
|
| D | nn_ops_test.cc | 738 EigenThreadPoolWrapper wrapper(&threadpool); in BM_LRNFloat()
820 EigenThreadPoolWrapper wrapper(&threadpool); in BM_AvgPool()
912 EigenThreadPoolWrapper wrapper(&threadpool); in BM_AvgPoolBk()
1016 EigenThreadPoolWrapper wrapper(&threadpool); in BM_MaxPool()
1196 EigenThreadPoolWrapper wrapper(&threadpool); in BM_ReluFloat()
|
| D | quantized_batch_norm_op_test.cc | 221 EigenThreadPoolWrapper wrapper(&threadpool); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_runtime_test.cc | 104 tensorflow::EigenThreadPoolWrapper tp(&pool); in EigenMatrixMultiply()
|