Searched refs:set_eigen_cpu_device (Results 1 – 4 of 4) sorted by relevance
82 set_eigen_cpu_device(tp_info->eigen_device_.get()); in LocalDevice()
155 void set_eigen_cpu_device(Eigen::ThreadPoolDevice* d) { in set_eigen_cpu_device() function
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()
74 set_eigen_cpu_device(eigen_device_.get()); in DeviceSimple()