layer_norm.h File Reference

#include <assert.h>
#include <math.h>
#include "tensor.h"
#include "ops/sum.h"
#include "ops/add.h"
#include "ops/add1.h"
#include "ops/mul.h"
#include "ops/mul1.h"
#include "ops/sqr.h"

Include dependency graph for layer_norm.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Functions
void	NN_layer_norm (Tensor *out, const Tensor *in, size_t normalized_dims, const Tensor *weight, const Tensor *bias, const float eps)