AlexeyAB
2018-02-25 85eafd3d590c449b35627af3e0f2a007c5a27d15
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