AlexeyAB
2018-05-17 9ba0beb413d8b443376c36c1701a0dbce1e8d566
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