Searched refs:eigen_worker_threads_ (Results 1 – 4 of 4) sorted by relevance
48 eigen_worker_threads_.num_threads = kNumThreads; in SingleThreadedCpuDevice()49 eigen_worker_threads_.workers = GraphRunnerThreadPool(); in SingleThreadedCpuDevice()51 new EigenThreadPoolWrapper(eigen_worker_threads_.workers)); in SingleThreadedCpuDevice()53 eigen_threadpool_wrapper_.get(), eigen_worker_threads_.num_threads)); in SingleThreadedCpuDevice()54 set_tensorflow_cpu_worker_threads(&eigen_worker_threads_); in SingleThreadedCpuDevice()81 DeviceBase::CpuWorkerThreads eigen_worker_threads_; member in tensorflow::__anona485c93d0111::SingleThreadedCpuDevice
95 eigen_worker_threads_.num_threads = intra_op_parallelism_threads; in EigenThreadPoolInfo()96 eigen_worker_threads_.workers = new thread::ThreadPool( in EigenThreadPoolInfo()100 new EigenThreadPoolWrapper(eigen_worker_threads_.workers)); in EigenThreadPoolInfo()105 eigen_threadpool_wrapper_.get(), eigen_worker_threads_.num_threads, in EigenThreadPoolInfo()112 delete eigen_worker_threads_.workers; in ~EigenThreadPoolInfo()115 DeviceBase::CpuWorkerThreads eigen_worker_threads_; member164 set_tensorflow_cpu_worker_threads(&tp_info->eigen_worker_threads_); in LocalDevice()
54 eigen_worker_threads_.num_threads = port::NumSchedulableCPUs(); in DeviceSimple()55 eigen_worker_threads_.workers = new thread::ThreadPool( in DeviceSimple()56 Env::Default(), "evaluation_utils", eigen_worker_threads_.num_threads); in DeviceSimple()58 new EigenThreadPoolWrapper(eigen_worker_threads_.workers)); in DeviceSimple()60 eigen_threadpool_wrapper_.get(), eigen_worker_threads_.num_threads)); in DeviceSimple()61 set_tensorflow_cpu_worker_threads(&eigen_worker_threads_); in DeviceSimple()68 delete eigen_worker_threads_.workers; in ~DeviceSimple()
49 DeviceBase::CpuWorkerThreads eigen_worker_threads_;