batch_norm2d.h File Reference

#include <assert.h>
#include <math.h>
#include "tensor.h"

Include dependency graph for batch_norm2d.h:

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

Go to the source code of this file.

Functions
void	NN_batch_norm2d (Tensor *out, const Tensor *in, const Tensor *weight, const Tensor *bias, float eps, const Tensor *running_mean, const Tensor *running_va)