Searched refs:set_eigen_cpu_device (Results 1 – 9 of 9) sorted by relevance
34 dbase.set_eigen_cpu_device(&eigen_device); in TEST()
81 void DeviceBase::set_eigen_cpu_device(Eigen::ThreadPoolDevice* d) { in set_eigen_cpu_device() function in tensorflow::DeviceBase
181 void set_eigen_cpu_device(Eigen::ThreadPoolDevice* d);
68 set_eigen_cpu_device(&eigen_threadpool_device); in RenamedDevice()
54 set_eigen_cpu_device(eigen_device_.get()); in SingleThreadedCpuDevice()
158 set_eigen_cpu_device(tp_info->eigen_device_.get()); in LocalDevice()
42 set_eigen_cpu_device(eigen_device_.get()); in DeviceSimple()
55 set_eigen_cpu_device(eigen_device_.get()); in SimpleDevice()
781 device->set_eigen_cpu_device(&eigen_cpu_device); in BM_LRNFloat()862 device->set_eigen_cpu_device(&eigen_cpu_device); in BM_AvgPool()952 device->set_eigen_cpu_device(&eigen_cpu_device); in BM_AvgPoolBk()1057 device->set_eigen_cpu_device(&eigen_cpu_device); in BM_MaxPool()1243 device->set_eigen_cpu_device(&eigen_cpu_device); in BM_ReluFloat()1314 device->set_eigen_cpu_device(&eigen_cpu_device); in BM_SoftplusFloat()