1 #ifndef BATCHNORM_LAYER_H 2 #define BATCHNORM_LAYER_H 3 4 #include "image.h" 5 #include "layer.h" 6 #include "network.h" 7 8 #ifdef __cplusplus 9 extern "C" { 10 #endif 11 layer make_batchnorm_layer(int batch, int w, int h, int c, int train); 12 void forward_batchnorm_layer(layer l, network_state state); 13 void backward_batchnorm_layer(layer l, network_state state); 14 void update_batchnorm_layer(layer l, int batch, float learning_rate, float momentum, float decay); 15 16 void resize_batchnorm_layer(layer *l, int w, int h); 17 18 #ifdef GPU 19 void forward_batchnorm_layer_gpu(layer l, network_state state); 20 void backward_batchnorm_layer_gpu(layer l, network_state state); 21 void update_batchnorm_layer_gpu(layer l, int batch, float learning_rate_init, float momentum, float decay, float loss_scale); 22 void pull_batchnorm_layer(layer l); 23 void push_batchnorm_layer(layer l); 24 #endif 25 26 #ifdef __cplusplus 27 } 28 #endif 29 #endif 30