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