00001 #ifndef BATCHNORM_LAYER_H 00002 #define BATCHNORM_LAYER_H 00003 00004 #include "image.h" 00005 #include "layer.h" 00006 #include "network.h" 00007 00008 layer make_batchnorm_layer(int batch, int w, int h, int c); 00009 void forward_batchnorm_layer(layer l, network_state state); 00010 void backward_batchnorm_layer(layer l, network_state state); 00011 00012 #ifdef GPU 00013 void forward_batchnorm_layer_gpu(layer l, network_state state); 00014 void backward_batchnorm_layer_gpu(layer l, network_state state); 00015 void pull_batchnorm_layer(layer l); 00016 void push_batchnorm_layer(layer l); 00017 #endif 00018 00019 #endif