Go to the source code of this file.
Functions | |
void | backward_batchnorm_layer (const layer l, network_state state) |
void | backward_scale_cpu (float *x_norm, float *delta, int batch, int n, int size, float *scale_updates) |
void | forward_batchnorm_layer (layer l, network_state state) |
layer | make_batchnorm_layer (int batch, int w, int h, int c) |
void | mean_delta_cpu (float *delta, float *variance, int batch, int filters, int spatial, float *mean_delta) |
void | normalize_delta_cpu (float *x, float *mean, float *variance, float *mean_delta, float *variance_delta, int batch, int filters, int spatial, float *delta) |
void | resize_batchnorm_layer (layer *layer, int w, int h) |
void | variance_delta_cpu (float *x, float *delta, float *mean, float *variance, int batch, int filters, int spatial, float *variance_delta) |
void backward_batchnorm_layer | ( | const layer | l, |
network_state | state | ||
) |
Definition at line 147 of file batchnorm_layer.c.
void backward_scale_cpu | ( | float * | x_norm, |
float * | delta, | ||
int | batch, | ||
int | n, | ||
int | size, | ||
float * | scale_updates | ||
) |
Definition at line 59 of file batchnorm_layer.c.
void forward_batchnorm_layer | ( | layer | l, |
network_state | state | ||
) |
Definition at line 122 of file batchnorm_layer.c.
layer make_batchnorm_layer | ( | int | batch, |
int | w, | ||
int | h, | ||
int | c | ||
) |
Definition at line 5 of file batchnorm_layer.c.
void mean_delta_cpu | ( | float * | delta, |
float * | variance, | ||
int | batch, | ||
int | filters, | ||
int | spatial, | ||
float * | mean_delta | ||
) |
Definition at line 74 of file batchnorm_layer.c.
void normalize_delta_cpu | ( | float * | x, |
float * | mean, | ||
float * | variance, | ||
float * | mean_delta, | ||
float * | variance_delta, | ||
int | batch, | ||
int | filters, | ||
int | spatial, | ||
float * | delta | ||
) |
Definition at line 104 of file batchnorm_layer.c.
void resize_batchnorm_layer | ( | layer * | layer, |
int | w, | ||
int | h | ||
) |
Definition at line 117 of file batchnorm_layer.c.
void variance_delta_cpu | ( | float * | x, |
float * | delta, | ||
float * | mean, | ||
float * | variance, | ||
int | batch, | ||
int | filters, | ||
int | spatial, | ||
float * | variance_delta | ||
) |
Definition at line 89 of file batchnorm_layer.c.