Searched refs:set_eigen_cpu_device (Results 1 – 8 of 8) sorted by relevance
34 dbase.set_eigen_cpu_device(&eigen_device); in TEST()
76 void DeviceBase::set_eigen_cpu_device(Eigen::ThreadPoolDevice* d) { in set_eigen_cpu_device() function in tensorflow::DeviceBase
177 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()
766 device->set_eigen_cpu_device(&eigen_cpu_device); in BM_LRNFloat()847 device->set_eigen_cpu_device(&eigen_cpu_device); in BM_AvgPool()937 device->set_eigen_cpu_device(&eigen_cpu_device); in BM_AvgPoolBk()1042 device->set_eigen_cpu_device(&eigen_cpu_device); in BM_MaxPool()1221 device->set_eigen_cpu_device(&eigen_cpu_device); in BM_ReluFloat()1292 device->set_eigen_cpu_device(&eigen_cpu_device); in BM_SoftplusFloat()