Joseph Redmon
2014-10-26 edbccdfcaf46f11e631afe98796f3e6e170da5d0
src/col2im.cl
@@ -23,11 +23,11 @@
    int w_start = (w<ksize)?0:(w-ksize)/stride + 1;
    int w_end = w/stride + 1;
    if(width_col < w_end) w_end = width_col;
    w_end = (width_col < w_end) ? width_col : w_end;
    int h_start = (h<ksize)?0:(h-ksize)/stride+1;
    int h_end = h/stride + 1;
    if(height_col < h_end) h_end = height_col;
    h_end = (height_col < h_end) ? height_col : h_end;
    int rows = channels * ksize * ksize;
    int cols = height_col*width_col;