Edmond Yoo
2018-09-06 c8b7dc4f215b89f821b3f242ccb2ada7f535efeb
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);