From 47528e37cf29e0f9da6885213e5aee044bed84ef Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Wed, 15 Apr 2015 08:04:38 +0000
Subject: [PATCH] crop layer scaling and trans on cpu
---
src/col2im_kernels.cu | 6 +++---
1 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/src/col2im_kernels.cu b/src/col2im_kernels.cu
index 76a86e6..67c0b03 100644
--- a/src/col2im_kernels.cu
+++ b/src/col2im_kernels.cu
@@ -37,9 +37,9 @@
}
}
-void col2im_ongpu(float *im,
+void col2im_ongpu(float *data_col,
int channels, int height, int width,
- int ksize, int stride, int pad, float *data_col){
+ int ksize, int stride, int pad, float *data_im){
// We are going to launch channels * height_col * width_col kernels, each
// kernel responsible for copying a single-channel grid.
pad = pad ? ksize/2 : 0;
@@ -50,7 +50,7 @@
BLOCK>>>(
num_kernels, data_col, height, width, ksize, pad,
stride, height_col,
- width_col, im);
+ width_col, data_im);
}
/*
--
Gitblit v1.10.0