Home
last modified time | relevance | path

Searched refs:eigen_device_ (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dsingle_threaded_cpu_device.cc52 eigen_device_.reset(new Eigen::ThreadPoolDevice( in SingleThreadedCpuDevice()
55 set_eigen_cpu_device(eigen_device_.get()); in SingleThreadedCpuDevice()
60 eigen_device_.reset(); in ~SingleThreadedCpuDevice()
83 std::unique_ptr<Eigen::ThreadPoolDevice> eigen_device_; member in tensorflow::__anona485c93d0111::SingleThreadedCpuDevice
Dlocal_device.cc104 eigen_device_.reset(new Eigen::ThreadPoolDevice( in EigenThreadPoolInfo()
111 eigen_device_.reset(); in ~EigenThreadPoolInfo()
117 std::unique_ptr<Eigen::ThreadPoolDevice> eigen_device_; member
165 set_eigen_cpu_device(tp_info->eigen_device_.get()); in LocalDevice()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Devaluation_utils.cc59 eigen_device_.reset(new Eigen::ThreadPoolDevice( in DeviceSimple()
62 set_eigen_cpu_device(eigen_device_.get()); in DeviceSimple()
67 eigen_device_.reset(); in ~DeviceSimple()
Devaluation_utils.h51 std::unique_ptr<Eigen::ThreadPoolDevice> eigen_device_; variable