Functions | |
void | backward_batchnorm_layer (Darknet::Layer &l, Darknet::NetworkState state) |
void | backward_batchnorm_layer_gpu (Darknet::Layer &l, Darknet::NetworkState state) |
void | backward_scale_cpu (float *x_norm, float *delta, int batch, int n, int size, float *scale_updates) |
void | forward_batchnorm_layer (Darknet::Layer &l, Darknet::NetworkState state) |
void | forward_batchnorm_layer_gpu (Darknet::Layer &l, Darknet::NetworkState state) |
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 | pull_batchnorm_layer (Darknet::Layer &l) |
void | push_batchnorm_layer (Darknet::Layer &l) |
void | resize_batchnorm_layer (Darknet::Layer *l, int w, int h) |
void | update_batchnorm_layer (Darknet::Layer &l, int batch, float learning_rate, float momentum, float decay) |
void | update_batchnorm_layer_gpu (Darknet::Layer &l, int batch, float learning_rate_init, float momentum, float decay, float loss_scale) |
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 | ( | Darknet::Layer & | l, |
Darknet::NetworkState | state | ||
) |
void backward_batchnorm_layer_gpu | ( | Darknet::Layer & | l, |
Darknet::NetworkState | state | ||
) |
void backward_scale_cpu | ( | float * | x_norm, |
float * | delta, | ||
int | batch, | ||
int | n, | ||
int | size, | ||
float * | scale_updates | ||
) |
void forward_batchnorm_layer | ( | Darknet::Layer & | l, |
Darknet::NetworkState | state | ||
) |
void forward_batchnorm_layer_gpu | ( | Darknet::Layer & | l, |
Darknet::NetworkState | state | ||
) |
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 pull_batchnorm_layer | ( | Darknet::Layer & | l | ) |
void push_batchnorm_layer | ( | Darknet::Layer & | l | ) |
void resize_batchnorm_layer | ( | Darknet::Layer * | l, |
int | w, | ||
int | h | ||
) |
void update_batchnorm_layer | ( | Darknet::Layer & | l, |
int | batch, | ||
float | learning_rate, | ||
float | momentum, | ||
float | decay | ||
) |
void update_batchnorm_layer_gpu | ( | Darknet::Layer & | l, |
int | batch, | ||
float | learning_rate_init, | ||
float | momentum, | ||
float | decay, | ||
float | loss_scale | ||
) |
void variance_delta_cpu | ( | float * | x, |
float * | delta, | ||
float * | mean, | ||
float * | variance, | ||
int | batch, | ||
int | filters, | ||
int | spatial, | ||
float * | variance_delta | ||
) |