AlexeyAB
2017-10-26 b714004546b97e9a43fae3e385dbefb56cecafb6
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