Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/framework/
Ddevice_base.cc56 const Eigen::ThreadPoolDevice* DeviceBase::eigen_cpu_device() { in eigen_cpu_device() function in tensorflow::DeviceBase
Dop_kernel.h1163 const Eigen::ThreadPoolDevice& eigen_cpu_device() const { in eigen_cpu_device() function
/external/tensorflow/tensorflow/core/common_runtime/
Drenamed_device.h75 const Eigen::ThreadPoolDevice* eigen_cpu_device() override { in eigen_cpu_device() function
/external/tensorflow/tensorflow/core/kernels/
Dnn_ops_test.cc743 Eigen::ThreadPoolDevice eigen_cpu_device(threadpool.AsEigenThreadPool(), in BM_LRNFloat() local
826 Eigen::ThreadPoolDevice eigen_cpu_device(threadpool.AsEigenThreadPool(), in BM_AvgPool() local
919 Eigen::ThreadPoolDevice eigen_cpu_device(threadpool.AsEigenThreadPool(), in BM_AvgPoolBk() local
1024 Eigen::ThreadPoolDevice eigen_cpu_device(threadpool.AsEigenThreadPool(), in BM_MaxPool() local
1205 Eigen::ThreadPoolDevice eigen_cpu_device(threadpool.AsEigenThreadPool(), in BM_ReluFloat() local
1278 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