rms_norm.h

Go to the documentation of this file.

#ifndef __NN_RMS_NORM_H
#define __NN_RMS_NORM_H
 
#include <assert.h>
#include <math.h>
 
#include "tensor.h"
#include "ops/rms_norm.h"
 
 
void NN_rms_norm(Tensor *y, const Tensor *x, const Tensor *w, float eps);
 
 
#endif // __NN_RMS_NORM_H