/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() function 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()
|