Searched refs:set_eigen_cpu_device (Results 1 – 7 of 7) sorted by relevance
36 void DeviceBase::set_eigen_cpu_device(Eigen::ThreadPoolDevice* d) { in set_eigen_cpu_device() function in tensorflow::DeviceBase
35 dbase.set_eigen_cpu_device(&eigen_device); in TEST()
174 void set_eigen_cpu_device(Eigen::ThreadPoolDevice* d);
55 set_eigen_cpu_device(eigen_device_.get()); in SingleThreadedCpuDevice()
165 set_eigen_cpu_device(tp_info->eigen_device_.get()); in LocalDevice()
62 set_eigen_cpu_device(eigen_device_.get()); in DeviceSimple()
740 device->set_eigen_cpu_device(&eigen_cpu_device); in BM_LRNFloat()822 device->set_eigen_cpu_device(&eigen_cpu_device); in BM_AvgPool()914 device->set_eigen_cpu_device(&eigen_cpu_device); in BM_AvgPoolBk()1018 device->set_eigen_cpu_device(&eigen_cpu_device); in BM_MaxPool()1198 device->set_eigen_cpu_device(&eigen_cpu_device); in BM_ReluFloat()