Functions | |
void | backward_local_avgpool_layer (Darknet::Layer &l, Darknet::NetworkState state) |
void | backward_maxpool_layer (Darknet::Layer &l, Darknet::NetworkState state) |
void | create_maxpool_cudnn_tensors (Darknet::Layer *l) |
void | cudnn_local_avgpool_setup (Darknet::Layer *l) |
void | cudnn_maxpool_setup (Darknet::Layer *l) |
void | forward_local_avgpool_layer (Darknet::Layer &l, Darknet::NetworkState state) |
void | forward_maxpool_layer (Darknet::Layer &l, Darknet::NetworkState state) |
Darknet::Image | get_maxpool_delta (Darknet::Layer &l) |
Darknet::Image | get_maxpool_image (Darknet::Layer &l) |
Darknet::Layer | make_maxpool_layer (int batch, int h, int w, int c, int size, int stride_x, int stride_y, int padding, int maxpool_depth, int out_channels, int antialiasing, int avgpool, int train) |
void | resize_maxpool_layer (Darknet::Layer *l, int w, int h) |
void backward_local_avgpool_layer | ( | Darknet::Layer & | l, |
Darknet::NetworkState | state | ||
) |
void backward_maxpool_layer | ( | Darknet::Layer & | l, |
Darknet::NetworkState | state | ||
) |
void create_maxpool_cudnn_tensors | ( | Darknet::Layer * | l | ) |
void cudnn_local_avgpool_setup | ( | Darknet::Layer * | l | ) |
void cudnn_maxpool_setup | ( | Darknet::Layer * | l | ) |
void forward_local_avgpool_layer | ( | Darknet::Layer & | l, |
Darknet::NetworkState | state | ||
) |
void forward_maxpool_layer | ( | Darknet::Layer & | l, |
Darknet::NetworkState | state | ||
) |
Darknet::Image get_maxpool_delta | ( | Darknet::Layer & | l | ) |
Darknet::Image get_maxpool_image | ( | Darknet::Layer & | l | ) |
Darknet::Layer make_maxpool_layer | ( | int | batch, |
int | h, | ||
int | w, | ||
int | c, | ||
int | size, | ||
int | stride_x, | ||
int | stride_y, | ||
int | padding, | ||
int | maxpool_depth, | ||
int | out_channels, | ||
int | antialiasing, | ||
int | avgpool, | ||
int | train | ||
) |
void resize_maxpool_layer | ( | Darknet::Layer * | l, |
int | w, | ||
int | h | ||
) |