blame | history | patch | commit | commitdiff | ignore whitespace
IlyaOvodov
2018-05-30 01bec657080d123a5b44a10ec94ebe132e20b1f3 
 src/normalization_layer.h


New file


@@ -0,0 +1,19 @@


#ifndef NORMALIZATION_LAYER_H


#define NORMALIZATION_LAYER_H




#include "image.h"


#include "layer.h"


#include "network.h"




layer make_normalization_layer(int batch, int w, int h, int c, int size, float alpha, float beta, float kappa);


void resize_normalization_layer(layer *layer, int h, int w);


void forward_normalization_layer(const layer layer, network_state state);


void backward_normalization_layer(const layer layer, network_state state);


void visualize_normalization_layer(layer layer, char *window);




#ifdef GPU


void forward_normalization_layer_gpu(const layer layer, network_state state);


void backward_normalization_layer_gpu(const layer layer, network_state state);


#endif




#endif