From 176d65b76583803cf10194c4c70bdc51897f2ae3 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Mon, 11 Aug 2014 19:52:07 +0000
Subject: [PATCH] Nist NIN testing multi-crop
---
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