Searched refs:set_eigen_cpu_device (Results 1 – 8 of 8) sorted by relevance
34 dbase.set_eigen_cpu_device(&eigen_device); in TEST()
44 void DeviceBase::set_eigen_cpu_device(Eigen::ThreadPoolDevice* d) { in set_eigen_cpu_device() function in tensorflow::DeviceBase
176 void set_eigen_cpu_device(Eigen::ThreadPoolDevice* d);
54 set_eigen_cpu_device(eigen_device_.get()); in SingleThreadedCpuDevice()
68 set_eigen_cpu_device(&eigen_threadpool_device); in RenamedDevice()
158 set_eigen_cpu_device(tp_info->eigen_device_.get()); in LocalDevice()
42 set_eigen_cpu_device(eigen_device_.get()); in DeviceSimple()
745 device->set_eigen_cpu_device(&eigen_cpu_device); in BM_LRNFloat()828 device->set_eigen_cpu_device(&eigen_cpu_device); in BM_AvgPool()921 device->set_eigen_cpu_device(&eigen_cpu_device); in BM_AvgPoolBk()1026 device->set_eigen_cpu_device(&eigen_cpu_device); in BM_MaxPool()1207 device->set_eigen_cpu_device(&eigen_cpu_device); in BM_ReluFloat()1280 device->set_eigen_cpu_device(&eigen_cpu_device); in BM_SoftplusFloat()