summaryrefslogtreecommitdiff
path: root/src/SeparableConvolution_cuda.c
diff options
context:
space:
mode:
authorsniklaus <simon.niklaus@outlook.com>2017-09-18 22:20:04 -0700
committersniklaus <simon.niklaus@outlook.com>2017-09-18 22:20:04 -0700
commitcfd6a91a628c603eeeecf517340ac0474a126496 (patch)
tree7c891bc8660904771bb612f56aca8f22760d0cdb /src/SeparableConvolution_cuda.c
parente123297d61dc9915b70060def498560ca5d3d073 (diff)
no message
Diffstat (limited to 'src/SeparableConvolution_cuda.c')
-rw-r--r--src/SeparableConvolution_cuda.c12
1 files changed, 6 insertions, 6 deletions
diff --git a/src/SeparableConvolution_cuda.c b/src/SeparableConvolution_cuda.c
index a48622e..973e4c1 100644
--- a/src/SeparableConvolution_cuda.c
+++ b/src/SeparableConvolution_cuda.c
@@ -6,16 +6,16 @@
extern THCState* state;
int SeparableConvolution_cuda_forward(
- THCudaTensor* input1,
- THCudaTensor* input2,
- THCudaTensor* input3,
+ THCudaTensor* input,
+ THCudaTensor* vertical,
+ THCudaTensor* horizontal,
THCudaTensor* output
) {
SeparableConvolution_kernel_forward(
state,
- input1,
- input2,
- input3,
+ input,
+ vertical,
+ horizontal,
output
);