Functions | |
void | backward_local_avgpool_layer_gpu (Darknet::Layer &l, Darknet::NetworkState state) |
__global__ void | backward_local_avgpool_layer_kernel (int n, int in_h, int in_w, int in_c, int stride_x, int stride_y, int size, int pad, float *delta, float *prev_delta) |
__global__ void | backward_maxpool_depth_layer_kernel (int n, int w, int h, int c, int batch, float *delta, float *prev_delta, int *indexes) |
void | backward_maxpool_layer_gpu (Darknet::Layer &l, Darknet::NetworkState state) |
__global__ void | backward_maxpool_layer_kernel (int n, int in_h, int in_w, int in_c, int stride_x, int stride_y, int size, int pad, float *delta, float *prev_delta, int *indexes) |
__global__ void | backward_zero_nonmax_kernel (int n, int *indexes, float *prev_delta) |
void | forward_local_avgpool_layer_gpu (Darknet::Layer &l, Darknet::NetworkState state) |
__global__ void | forward_local_avgpool_layer_kernel (int n, int in_h, int in_w, int in_c, int stride_x, int stride_y, int size, int pad, float *input, float *output) |
__global__ void | forward_maxpool_depth_layer_kernel (int n, int w, int h, int c, int out_c, int batch, float *input, float *output, int *indexes) |
void | forward_maxpool_layer_gpu (Darknet::Layer &l, Darknet::NetworkState state) |
__global__ void | forward_maxpool_layer_kernel (int n, int in_h, int in_w, int in_c, int stride_x, int stride_y, int size, int pad, float *input, float *output, int *indexes) |
__global__ void | forward_zero_nonmax_kernel (int n, float *input, float *output) |
void backward_local_avgpool_layer_gpu | ( | Darknet::Layer & | l, |
Darknet::NetworkState | state | ||
) |
__global__ void backward_local_avgpool_layer_kernel | ( | int | n, |
int | in_h, | ||
int | in_w, | ||
int | in_c, | ||
int | stride_x, | ||
int | stride_y, | ||
int | size, | ||
int | pad, | ||
float * | delta, | ||
float * | prev_delta | ||
) |
__global__ void backward_maxpool_depth_layer_kernel | ( | int | n, |
int | w, | ||
int | h, | ||
int | c, | ||
int | batch, | ||
float * | delta, | ||
float * | prev_delta, | ||
int * | indexes | ||
) |
void backward_maxpool_layer_gpu | ( | Darknet::Layer & | l, |
Darknet::NetworkState | state | ||
) |
__global__ void backward_maxpool_layer_kernel | ( | int | n, |
int | in_h, | ||
int | in_w, | ||
int | in_c, | ||
int | stride_x, | ||
int | stride_y, | ||
int | size, | ||
int | pad, | ||
float * | delta, | ||
float * | prev_delta, | ||
int * | indexes | ||
) |
__global__ void backward_zero_nonmax_kernel | ( | int | n, |
int * | indexes, | ||
float * | prev_delta | ||
) |
void forward_local_avgpool_layer_gpu | ( | Darknet::Layer & | l, |
Darknet::NetworkState | state | ||
) |
__global__ void forward_local_avgpool_layer_kernel | ( | int | n, |
int | in_h, | ||
int | in_w, | ||
int | in_c, | ||
int | stride_x, | ||
int | stride_y, | ||
int | size, | ||
int | pad, | ||
float * | input, | ||
float * | output | ||
) |
__global__ void forward_maxpool_depth_layer_kernel | ( | int | n, |
int | w, | ||
int | h, | ||
int | c, | ||
int | out_c, | ||
int | batch, | ||
float * | input, | ||
float * | output, | ||
int * | indexes | ||
) |
void forward_maxpool_layer_gpu | ( | Darknet::Layer & | l, |
Darknet::NetworkState | state | ||
) |
__global__ void forward_maxpool_layer_kernel | ( | int | n, |
int | in_h, | ||
int | in_w, | ||
int | in_c, | ||
int | stride_x, | ||
int | stride_y, | ||
int | size, | ||
int | pad, | ||
float * | input, | ||
float * | output, | ||
int * | indexes | ||
) |
__global__ void forward_zero_nonmax_kernel | ( | int | n, |
float * | input, | ||
float * | output | ||
) |