ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Utils.h File Reference
#include <Eigen/Core>
#include <type_traits>
#include "cker/Shape.h"

Go to the source code of this file.

Namespaces

namespace  nnfw
 
namespace  nnfw::cker
 

Typedefs

template<typename Scalar >
using nnfw::cker::VectorMap = typename std::conditional< std::is_const< Scalar >::value, Eigen::Map< const Eigen::Matrix< typename std::remove_const< Scalar >::type, Eigen::Dynamic, 1 > >, Eigen::Map< Eigen::Matrix< Scalar, Eigen::Dynamic, 1 > > >::type
 
template<typename Scalar >
using nnfw::cker::MatrixMap = typename std::conditional< std::is_const< Scalar >::value, Eigen::Map< const Eigen::Matrix< typename std::remove_const< Scalar >::type, Eigen::Dynamic, Eigen::Dynamic > >, Eigen::Map< Eigen::Matrix< Scalar, Eigen::Dynamic, Eigen::Dynamic > > >::type
 

Functions

template<typename Scalar >
VectorMap< Scalar > nnfw::cker::MapAsVector (Scalar *data, const Shape &shape)
 
template<typename Scalar >
MatrixMap< Scalar > nnfw::cker::MapAsMatrixWithLastDimAsRows (Scalar *data, const Shape &shape)