1 #ifndef CONVOLUTIONAL_LAYER_H 2 #define CONVOLUTIONAL_LAYER_H 20 void add_bias_gpu(
float *output,
float *biases,
int batch,
int n,
int size);
21 void backward_bias_gpu(
float *bias_updates,
float *delta,
int batch,
int n,
int size);
22 void adam_update_gpu(
float *w,
float *d,
float *m,
float *v,
float B1,
float B2,
float eps,
float decay,
float rate,
int n,
int batch,
int t);
24 void cudnn_convolutional_setup(
layer *l);
28 convolutional_layer make_convolutional_layer(
int batch,
int h,
int w,
int c,
int n,
int groups,
int size,
int stride,
int padding,
ACTIVATION activation,
int batch_normalize,
int binary,
int xnor,
int adam);
35 void binarize_weights2(
float *weights,
int n,
int size,
char *binary,
float *scales);
39 void add_bias(
float *output,
float *biases,
int batch,
int n,
int size);
40 void backward_bias(
float *bias_updates,
float *delta,
int batch,
int n,
int size);
int convolutional_out_height(convolutional_layer layer)
void forward_convolutional_layer_gpu(convolutional_layer l, network net)
void add_bias_gpu(float *output, float *biases, int batch, int n, int size)
layer convolutional_layer
void backward_bias_gpu(float *bias_updates, float *delta, int batch, int n, int size)
convolutional_layer make_convolutional_layer(int batch, int h, int w, int c, int n, int groups, int size, int stride, int padding, ACTIVATION activation, int batch_normalize, int binary, int xnor, int adam)
void pull_convolutional_layer(layer l)
void add_bias(float *output, float *biases, int batch, int n, int size)
void forward_convolutional_layer(const convolutional_layer layer, network net)
void resize_convolutional_layer(convolutional_layer *layer, int w, int h)
void update_convolutional_layer(convolutional_layer layer, update_args a)
void swap_binary(convolutional_layer *l)
void backward_bias(float *bias_updates, float *delta, int batch, int n, int size)
image get_convolutional_delta(convolutional_layer layer)
void backward_convolutional_layer_gpu(convolutional_layer l, network net)
image get_convolutional_image(convolutional_layer layer)
image get_convolutional_weight(convolutional_layer layer, int i)
void binarize_weights(float *weights, int n, int size, float *binary)
image * visualize_convolutional_layer(convolutional_layer layer, char *window, image *prev_weights)
void push_convolutional_layer(layer l)
int convolutional_out_width(convolutional_layer layer)
void update_convolutional_layer_gpu(layer l, update_args a)
void backward_convolutional_layer(convolutional_layer layer, network net)
void binarize_weights2(float *weights, int n, int size, char *binary, float *scales)
void adam_update_gpu(float *w, float *d, float *m, float *v, float B1, float B2, float eps, float decay, float rate, int n, int batch, int t)