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.cc50 eigen_device_.reset(new Eigen::ThreadPoolDevice( in SingleThreadedCpuDevice()
54 set_eigen_cpu_device(eigen_device_.get()); in SingleThreadedCpuDevice()
57 ~SingleThreadedCpuDevice() override { eigen_device_.reset(); } in ~SingleThreadedCpuDevice()
91 std::unique_ptr<Eigen::ThreadPoolDevice> eigen_device_; member in tensorflow::__anonfe4177f40111::SingleThreadedCpuDevice
Dlocal_device.cc100 eigen_device_.reset(new Eigen::ThreadPoolDevice( in EigenThreadPoolInfo()
105 eigen_device_.reset(); in ~EigenThreadPoolInfo()
110 std::unique_ptr<Eigen::ThreadPoolDevice> eigen_device_; member
158 set_eigen_cpu_device(tp_info->eigen_device_.get()); in LocalDevice()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Devaluation_utils.cc38 eigen_device_.reset(new Eigen::ThreadPoolDevice( in DeviceSimple()
42 set_eigen_cpu_device(eigen_device_.get()); in DeviceSimple()
46 eigen_device_.reset(); in ~DeviceSimple()
Devaluation_utils.h50 std::unique_ptr<Eigen::ThreadPoolDevice> eigen_device_; variable