AlexeyAB
2018-05-17 24f563ce7165c2c44486fe8d0a2caea8d7c5a887
src/batchnorm_layer.h
@@ -12,6 +12,8 @@
#ifdef GPU
void forward_batchnorm_layer_gpu(layer l, network_state state);
void backward_batchnorm_layer_gpu(layer l, network_state state);
void pull_batchnorm_layer(layer l);
void push_batchnorm_layer(layer l);
#endif
#endif