nnfw::cker::eigen_support::EigenContext Struct Reference

#include <EigenSupport.h>

Public Member Functions
	EigenContext ()

Static Public Member Functions
static EigenContext &	GetEigenContext ()

Data Fields
std::unique_ptr< Eigen::ThreadPoolInterface >	thread_pool_wrapper
std::unique_ptr< Eigen::ThreadPoolDevice >	device

Static Public Attributes
static constexpr int	default_num_threadpool_threads = 4

Detailed Description

Definition at line 85 of file EigenSupport.h.

Constructor & Destructor Documentation

EigenContext()

nnfw::cker::eigen_support::EigenContext::EigenContext ( )

inline

Definition at line 91 of file EigenSupport.h.

  {
    int num_threads = std::thread::hardware_concurrency();
    if (num_threads == 0)
    {
      num_threads = default_num_threadpool_threads;
    }
    device.reset(); // destroy before we invalidate the thread pool
    thread_pool_wrapper.reset(new EigenThreadPoolWrapper(new Eigen::ThreadPool(num_threads)));
    device.reset(new Eigen::ThreadPoolDevice(thread_pool_wrapper.get(), num_threads));
  }

References default_num_threadpool_threads, device, and thread_pool_wrapper.

Member Function Documentation

GetEigenContext()

static EigenContext & nnfw::cker::eigen_support::EigenContext::GetEigenContext ( )

inlinestatic

Definition at line 103 of file EigenSupport.h.

  {
    static EigenContext instance;
    return instance;
  }

Referenced by nnfw::cker::eigen_support::GetThreadPoolDevice().

Field Documentation

default_num_threadpool_threads

constexpr int nnfw::cker::eigen_support::EigenContext::default_num_threadpool_threads = 4

staticconstexpr

Definition at line 87 of file EigenSupport.h.

Referenced by EigenContext().

device

std::unique_ptr<Eigen::ThreadPoolDevice> nnfw::cker::eigen_support::EigenContext::device

Definition at line 89 of file EigenSupport.h.

Referenced by EigenContext().

thread_pool_wrapper

std::unique_ptr<Eigen::ThreadPoolInterface> nnfw::cker::eigen_support::EigenContext::thread_pool_wrapper

Definition at line 88 of file EigenSupport.h.

Referenced by EigenContext().

---

The documentation for this struct was generated from the following file:

• compute/cker/include/cker/eigen/EigenSupport.h