ONE - On-device Neural Engine
Loading...
Searching...
No Matches
RmsNorm.h File Reference
#include "cker/Shape.h"
#include "cker/Types.h"
#include "cker/Utils.h"
#include <cmath>
#include <stdexcept>

Go to the source code of this file.

Namespaces

namespace  nnfw
 
namespace  nnfw::cker
 

Functions

void nnfw::cker::RmsNorm (const RmsNormParams &params, const Shape &input_shape, const float *input_data, const Shape &gamma_shape, const float *gamma_data, const Shape &output_shape, float *output_data)