AlexeyAB
2018-02-22 dda993f3dd3c753dfd580d485b39c1001830fee4
src/convolutional_layer.h
@@ -20,7 +20,7 @@
void add_bias_gpu(float *output, float *biases, int batch, int n, int size);
void backward_bias_gpu(float *bias_updates, float *delta, int batch, int n, int size);
#ifdef CUDNN
void cudnn_convolutional_setup(layer *l);
void cudnn_convolutional_setup(layer *l, int cudnn_preference);
#endif
#endif