Lines Matching refs:stride_y

54 …ool_layer_kernel(int n, int in_h, int in_w, int in_c, int stride_x, int stride_y, int size, int pa…  in forward_maxpool_layer_kernel()  argument
56 int h = (in_h + pad - size) / stride_y + 1; in forward_maxpool_layer_kernel()
80 int cur_h = h_offset + i*stride_y + l; in forward_maxpool_layer_kernel()
94 …ool_layer_kernel(int n, int in_h, int in_w, int in_c, int stride_x, int stride_y, int size, int pa… in backward_maxpool_layer_kernel() argument
96 int h = (in_h + pad - size) / stride_y + 1; in backward_maxpool_layer_kernel()
100 int area_y = (size - 1) / stride_y; in backward_maxpool_layer_kernel()
122 int out_h = (i-h_offset)/stride_y + l; in backward_maxpool_layer_kernel()
179 …cuda_stream() >> > (n, layer.h, layer.w, layer.c, layer.stride_x, layer.stride_y, layer.size, laye… in forward_maxpool_layer_gpu()
226 …t_cuda_stream() >>>(n, layer.h, layer.w, layer.c, layer.stride_x, layer.stride_y, layer.size, laye… in backward_maxpool_layer_gpu()
233 …ool_layer_kernel(int n, int in_h, int in_w, int in_c, int stride_x, int stride_y, int size, int pa… in forward_local_avgpool_layer_kernel() argument
235 int h = (in_h + pad - size) / stride_y + 1; in forward_local_avgpool_layer_kernel()
259 int cur_h = h_offset + i*stride_y + l; in forward_local_avgpool_layer_kernel()
274 …ool_layer_kernel(int n, int in_h, int in_w, int in_c, int stride_x, int stride_y, int size, int pa… in backward_local_avgpool_layer_kernel() argument
276 int h = (in_h + pad - size) / stride_y + 1; in backward_local_avgpool_layer_kernel()
280 int area_y = (size - 1) / stride_y; in backward_local_avgpool_layer_kernel()
303 int out_h = (i - h_offset) / stride_y + l; in backward_local_avgpool_layer_kernel()
350 …cuda_stream() >> > (n, layer.h, layer.w, layer.c, layer.stride_x, layer.stride_y, layer.size, laye… in forward_local_avgpool_layer_gpu()
359 …_cuda_stream() >> >(n, layer.h, layer.w, layer.c, layer.stride_x, layer.stride_y, layer.size, laye… in backward_local_avgpool_layer_gpu()