nnfw::cker::eigen_support Namespace Reference

Data Structures
struct	EigenContext
class	EigenThreadPoolWrapper
struct	MatMulConvFunctor

Typedefs
typedef Eigen::TensorMap< Eigen::Tensor< float, 2, Eigen::RowMajor, Eigen::DenseIndex >, Eigen::Aligned >	EigenMatrix
typedef Eigen::TensorMap< Eigen::Tensor< const float, 2, Eigen::RowMajor, Eigen::DenseIndex >, Eigen::Aligned >	ConstEigenMatrix
typedef Eigen::TensorMap< Eigen::Tensor< float, 4, Eigen::RowMajor, Eigen::DenseIndex >, Eigen::Aligned >	EigenTensor
typedef Eigen::TensorMap< Eigen::Tensor< const float, 4, Eigen::RowMajor, Eigen::DenseIndex >, Eigen::Aligned >	ConstEigenTensor

Functions
const Eigen::ThreadPoolDevice *	GetThreadPoolDevice ()
template<typename T >
int64_t	kPacketSize ()
int	getThreadCount ()

Typedef Documentation

ConstEigenMatrix

typedef Eigen::TensorMap<Eigen::Tensor<const float, 2, Eigen::RowMajor, Eigen::DenseIndex>, Eigen::Aligned> nnfw::cker::eigen_support::ConstEigenMatrix

Definition at line 45 of file EigenSupport.h.

ConstEigenTensor

typedef Eigen::TensorMap<Eigen::Tensor<const float, 4, Eigen::RowMajor, Eigen::DenseIndex>, Eigen::Aligned> nnfw::cker::eigen_support::ConstEigenTensor

Definition at line 52 of file EigenSupport.h.

EigenMatrix

typedef Eigen::TensorMap<Eigen::Tensor<float, 2, Eigen::RowMajor, Eigen::DenseIndex>, Eigen::Aligned> nnfw::cker::eigen_support::EigenMatrix

Definition at line 42 of file EigenSupport.h.

EigenTensor

typedef Eigen::TensorMap<Eigen::Tensor<float, 4, Eigen::RowMajor, Eigen::DenseIndex>, Eigen::Aligned> nnfw::cker::eigen_support::EigenTensor

Definition at line 49 of file EigenSupport.h.

Function Documentation

getThreadCount()

int nnfw::cker::eigen_support::getThreadCount ( )

inline

Definition at line 122 of file EigenSupport.h.

{
  const Eigen::ThreadPoolDevice &d = *eigen_support::GetThreadPoolDevice();
  return d.numThreads();
}

References GetThreadPoolDevice().

Referenced by onert::backend::train::ops::DepthwiseConvolutionLayer::configureBackward().

GetThreadPoolDevice()

const Eigen::ThreadPoolDevice * nnfw::cker::eigen_support::GetThreadPoolDevice ( )

inline

Definition at line 110 of file EigenSupport.h.

{
  auto &ctx = EigenContext::GetEigenContext();
  return ctx.device.get();
}

References nnfw::cker::eigen_support::EigenContext::GetEigenContext().

Referenced by nnfw::cker::train::Adam(), nnfw::cker::bias_op::biasHelper(), nnfw::cker::functor::biasReductionHelper(), nnfw::cker::BroadcastTo(), nnfw::cker::train::CategoricalCrossEntropyWithLogits(), nnfw::cker::multithreaded::Conv(), getThreadCount(), nnfw::cker::train::GradientDescent(), nnfw::cker::FusedBatchNorm::operator()(), nnfw::cker::train::LaunchConv2DBackpropInputOp< T >::operator()(), nnfw::cker::train::LaunchConv2DBackpropFilterOp< T >::operator()(), nnfw::cker::depthwise_conv_op::LaunchDepthwiseConvOp< CPUDevice, T >::operator()(), nnfw::cker::depthwise_conv_op::LaunchDepthwiseConvBackpropInputOp< CPUDevice, T >::operator()(), and nnfw::cker::depthwise_conv_op::LaunchDepthwiseConvBackpropFilterOp< CPUDevice, T >::operator()().

kPacketSize()

template<typename T >

int64_t nnfw::cker::eigen_support::kPacketSize

(

)

Definition at line 116 of file EigenSupport.h.

{
  typedef typename Eigen::internal::packet_traits<T>::type Packet;
  return sizeof(Packet) / sizeof(T);
}