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() argument104 ~EigenThreadPoolInfo() { in ~EigenThreadPoolInfo()109 DeviceBase::CpuWorkerThreads eigen_worker_threads_;110 std::unique_ptr<Eigen::ThreadPoolDevice> eigen_device_;111 std::unique_ptr<EigenAllocator> eigen_allocator_;