Functions | |
void | backward_avgpool_layer_gpu (Darknet::Layer &l, Darknet::NetworkState state) |
__global__ void | backward_avgpool_layer_kernel (int n, int w, int h, int c, float *in_delta, float *out_delta) |
void | forward_avgpool_layer_gpu (Darknet::Layer &l, Darknet::NetworkState state) |
__global__ void | forward_avgpool_layer_kernel (int n, int w, int h, int c, float *input, float *output) |
void backward_avgpool_layer_gpu | ( | Darknet::Layer & | l, |
Darknet::NetworkState | state | ||
) |
__global__ void backward_avgpool_layer_kernel | ( | int | n, |
int | w, | ||
int | h, | ||
int | c, | ||
float * | in_delta, | ||
float * | out_delta | ||
) |
void forward_avgpool_layer_gpu | ( | Darknet::Layer & | l, |
Darknet::NetworkState | state | ||
) |
__global__ void forward_avgpool_layer_kernel | ( | int | n, |
int | w, | ||
int | h, | ||
int | c, | ||
float * | input, | ||
float * | output | ||
) |