Baremetal-NN
Baremetal-NN API documentation
Loading...
Searching...
No Matches
rms_norm.h
Go to the documentation of this file.
1#ifndef __NN_RMS_NORM_H
2#define __NN_RMS_NORM_H
3
4#include <stddef.h>
5#include <math.h>
6
7#include "sqr.h"
8#include "sum.h"
9#include "mul1.h"
10#include "mul.h"
11
12
13void NN_rms_norm_f32(size_t n,
14 float* y, size_t incy,
15 const float* x, size_t incx,
16 const float* w, size_t incw,
17 float eps
18 );
19
20
21#endif // __NN_RMS_NORM_H
void NN_rms_norm_f32(size_t n, float *y, size_t incy, const float *x, size_t incx, const float *w, size_t incw, float eps)