Edmond Yoo
2018-09-08 ff68352bb11e413fbc67effd04067b7345b88829
src/convolutional_layer.h
@@ -21,6 +21,7 @@
void backward_bias_gpu(float *bias_updates, float *delta, int batch, int n, int size);
#ifdef CUDNN
void cudnn_convolutional_setup(layer *l, int cudnn_preference);
void cuda_convert_f32_to_f16(float* input_f32, size_t size, float *output_f16);
#endif
#endif
@@ -34,6 +35,8 @@
void swap_binary(convolutional_layer *l);
void binarize_weights2(float *weights, int n, int size, char *binary, float *scales);
void binary_align_weights(convolutional_layer *l);
void backward_convolutional_layer(convolutional_layer layer, network_state state);
void add_bias(float *output, float *biases, int batch, int n, int size);