This commit is contained in:
Joseph Redmon
2016-05-11 10:45:50 -07:00
parent c7b10ceadb
commit 9942d48412
7 changed files with 136 additions and 85 deletions

View File

@@ -12,6 +12,8 @@ void backward_batchnorm_layer(layer l, network_state state);
#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