Method Col2Im

src/model/layer/opencl_convolution.cc:199–221

void OpenclConvolution::Col2Im(Block* src, const int col_data_off,
                               const int height, const int width,
                               const int kernel_h, const int kernel_w,
                               const int pad_h, const int pad_w,
                               const int stride_h, const int stride_w,
                               const int conv_h, const int conv_w,
                               const int data_im_off, const int channels,
                               Block* dst, Context* ctx) {

  auto ocl_ctx = viennacl::ocl::get_context(ctx->vcl_ctx_id);
  auto kernel = ocl_ctx.get_kernel("opencl_im2col", "col2im");

  auto src_buf = WrapHandle(static_cast<cl_mem>(src->mutable_data()), ocl_ctx);
  auto dst_buf = WrapHandle(static_cast<cl_mem>(dst->mutable_data()), ocl_ctx);

  int num_kernels = channels * height * width;

  viennacl::ocl::enqueue(kernel(num_kernels, src_buf, col_data_off, channels,
                                height, width, kernel_h, kernel_w,
                                pad_h, pad_w, stride_h, stride_w,
                                1, 1, conv_h, conv_w,
                                dst_buf, data_im_off));
}


} // namespace singa