Searched refs:eigen_worker_threads_ (Results 1 – 6 of 6) sorted by relevance
35 eigen_worker_threads_.num_threads = kDeviceSimpleThreads; in DeviceSimple()36 eigen_worker_threads_.workers = new thread::ThreadPool( in DeviceSimple()37 Env::Default(), "evaluation_utils", eigen_worker_threads_.num_threads); in DeviceSimple()39 eigen_worker_threads_.workers->AsEigenThreadPool(), in DeviceSimple()40 eigen_worker_threads_.num_threads)); in DeviceSimple()41 set_tensorflow_cpu_worker_threads(&eigen_worker_threads_); in DeviceSimple()47 delete eigen_worker_threads_.workers; in ~DeviceSimple()
49 DeviceBase::CpuWorkerThreads eigen_worker_threads_;
48 eigen_worker_threads_.num_threads = kNumThreads; in SingleThreadedCpuDevice()49 eigen_worker_threads_.workers = GraphRunnerThreadPool(); in SingleThreadedCpuDevice()51 eigen_worker_threads_.workers->AsEigenThreadPool(), in SingleThreadedCpuDevice()52 eigen_worker_threads_.num_threads)); in SingleThreadedCpuDevice()53 set_tensorflow_cpu_worker_threads(&eigen_worker_threads_); in SingleThreadedCpuDevice()90 DeviceBase::CpuWorkerThreads eigen_worker_threads_; member in tensorflow::__anon6eac52db0111::SingleThreadedCpuDevice
89 eigen_worker_threads_.num_threads = intra_op_parallelism_threads; in EigenThreadPoolInfo()90 eigen_worker_threads_.workers = new thread::ThreadPool( in EigenThreadPoolInfo()96 eigen_worker_threads_.workers->AsEigenThreadPool(); in EigenThreadPoolInfo()101 threadpool, eigen_worker_threads_.num_threads, eigen_allocator_.get())); in EigenThreadPoolInfo()106 delete eigen_worker_threads_.workers; in ~EigenThreadPoolInfo()109 DeviceBase::CpuWorkerThreads eigen_worker_threads_; member157 set_tensorflow_cpu_worker_threads(&tp_info->eigen_worker_threads_); in LocalDevice()
61 eigen_worker_threads_.workers = underlying_threadpool_.get(); in RenamedDevice()62 eigen_worker_threads_.num_threads = underlying_threadpool->NumThreads(); in RenamedDevice()63 set_tensorflow_cpu_worker_threads(&eigen_worker_threads_); in RenamedDevice()
167 DeviceBase::CpuWorkerThreads eigen_worker_threads_; variable