batch_norm2d.h

Go to the documentation of this file.

#ifndef __NN_BATCH_NORM2D_H
#define __NN_BATCH_NORM2D_H
 
#include <assert.h>
#include <math.h>
 
#include "tensor.h"
 
 
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
    );
 
 
#endif // __NN_BATCH_NORM2D_H