int index(int row, int col)
|
{
|
|
}
|
|
__kernel void col2im(__global float *data_col, int batch,
|
int channels, int height, int width,
|
int ksize, int stride, int pad, __global float *data_im)
|
{
|
int id = get_global_id(0);
|
int index = id;
|
int w = id%width;
|
id /= width;
|
int h = id%height;
|
id /= height;
|
int c = id%channels;
|
id /= channels;
|
int b = id%batch;
|
|
int height_col = (height - ksize) / stride + 1;
|
int width_col = (width - ksize) / stride + 1;
|
int rows = channels * ksize * ksize;
|
if (pad){
|
height_col = 1 + (height-1) / stride;
|
width_col = 1 + (width-1) / stride;
|
pad = ksize/2;
|
}
|
int cols = height_col*width_col;
|
int batch_offset = b*cols*rows;
|
int channel_offset = c*cols*ksize*ksize;
|
data_col[index] = 0;
|
int i,j;
|
for(i = 0; i < ksize; ++i){
|
row_offset = i*height_col*width_col;
|
for(j = 0; j < ksize; ++j){
|
col_offset =
|
}
|
}
|
|
data_col[col_index] = im2col_get_pixel(data_im, height, width, channels, b, im_row, im_col, c_im, pad);
|
}
|
