Home
last modified time | relevance | path

Searched refs:eigen_cpu_devices_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Ddevice_base.cc26 DeviceBase::~DeviceBase() { gtl::STLDeleteElements(&eigen_cpu_devices_); } in ~DeviceBase()
43 eigen_cpu_devices_.push_back(new Eigen::ThreadPoolDevice( in set_eigen_cpu_device()
57 std::min<int>(GetPerThreadMaxParallelism(), eigen_cpu_devices_.size())); in eigen_cpu_device()
58 return eigen_cpu_devices_[parallelism - 1]; in eigen_cpu_device()
Ddevice_base.h204 bool has_eigen_cpu_device() const { return !eigen_cpu_devices_.empty(); } in has_eigen_cpu_device()
270 std::vector<Eigen::ThreadPoolDevice*> eigen_cpu_devices_; variable