Home
last modified time | relevance | path

Searched defs:eigen_cpu_device (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Ddevice_base.cc93 const Eigen::ThreadPoolDevice* DeviceBase::eigen_cpu_device() { in eigen_cpu_device() function in tensorflow::DeviceBase
Dop_kernel.h1109 const Eigen::ThreadPoolDevice& eigen_cpu_device() const { in eigen_cpu_device() function
/external/tensorflow/tensorflow/core/kernels/
Dnn_ops_test.cc779 Eigen::ThreadPoolDevice eigen_cpu_device(threadpool.AsEigenThreadPool(), in BM_LRNFloat() local
860 Eigen::ThreadPoolDevice eigen_cpu_device(threadpool.AsEigenThreadPool(), in BM_AvgPool() local
950 Eigen::ThreadPoolDevice eigen_cpu_device(threadpool.AsEigenThreadPool(), in BM_AvgPoolBk() local
1055 Eigen::ThreadPoolDevice eigen_cpu_device(threadpool.AsEigenThreadPool(), in BM_MaxPool() local
1241 Eigen::ThreadPoolDevice eigen_cpu_device(threadpool.AsEigenThreadPool(), in BM_ReluFloat() local
1312 Eigen::ThreadPoolDevice eigen_cpu_device(threadpool.AsEigenThreadPool(), in BM_SoftplusFloat() local
Dquantized_batch_norm_op_test.cc220 Eigen::ThreadPoolDevice eigen_cpu_device(threadpool.AsEigenThreadPool(), 1); in TEST_F() local
/external/tensorflow/tensorflow/core/common_runtime/
Drenamed_device.h71 const Eigen::ThreadPoolDevice* eigen_cpu_device() override { in eigen_cpu_device() function
/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
Dkernel_fallback_execute_compat.cc790 const Eigen::ThreadPoolDevice* eigen_cpu_device() override { in eigen_cpu_device() function in tensorflow::tfd::__anon0fecb2ce0711::DeviceWithCustomAllocator