ONE - On-device Neural Engine
All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Modules Pages
GELU.h File Reference
#include "cker/Shape.h"
#include "cker/Types.h"
#include "cker/eigen/Utils.h"
#include <Eigen/Core>
#include <cmath>

Go to the source code of this file.

Namespaces

namespace  nnfw
 
namespace  nnfw::cker
 
namespace  nnfw::cker::gelu_internal
 

Functions

void nnfw::cker::GELU (const GELUParams &params, const Shape &input_shape, const float *input_data, const Shape &output_shape, float *output_data)
 

Variables

constexpr float nnfw::cker::gelu_internal::kSqrt2dPi = M_2_SQRTPI * M_SQRT1_2