AlexeyAB
2018-04-18 6056b835eb76b8a078aab18db3e7aba87314f4ce
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