Searched defs:EigenThreadPoolInfo (Results 1 – 1 of 1) sorted by relevance
58 struct LocalDevice::EigenThreadPoolInfo { struct in tensorflow::LocalDevice61 class EigenAllocator : public Eigen::Allocator {73 explicit EigenThreadPoolInfo(const SessionOptions& options, int numa_node, in EigenThreadPoolInfo() function109 ~EigenThreadPoolInfo() { in ~EigenThreadPoolInfo()115 DeviceBase::CpuWorkerThreads eigen_worker_threads_;116 std::unique_ptr<Eigen::ThreadPoolInterface> eigen_threadpool_wrapper_;117 std::unique_ptr<Eigen::ThreadPoolDevice> eigen_device_;118 std::unique_ptr<EigenAllocator> eigen_allocator_;