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
13
void
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
mul1.h
mul.h
NN_rms_norm_f32
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)
sum.h
sqr.h
nn
ops
rms_norm.h
Generated by
1.9.5