From d9f1b0b16edeb59281355a855e18a8be343fc33c Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Fri, 08 Aug 2014 19:04:15 +0000
Subject: [PATCH] probably how maxpool layers should be
---
src/im2col.cl | 6 +++---
1 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/src/im2col.cl b/src/im2col.cl
index 0226d28..765a92d 100644
--- a/src/im2col.cl
+++ b/src/im2col.cl
@@ -1,7 +1,7 @@
-__kernel void im2col(__global float *data_im,
- const int batch, const int channels, const int height, const int width,
- const int ksize, const int stride, __global float *data_col)
+__kernel void im2col(__global float *data_im, const int im_offset,
+ const int channels, const int height, const int width,
+ const int ksize, const int stride, __global float *data_col, const int col_offset)
{
int b = get_global_id(0);
int c = get_global_id(1);
--
Gitblit v1.10.0