diff options
Diffstat (limited to 'src/SeparableConvolution_kernel.h')
| -rw-r--r-- | src/SeparableConvolution_kernel.h | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/SeparableConvolution_kernel.h b/src/SeparableConvolution_kernel.h index b400579..11a0238 100644 --- a/src/SeparableConvolution_kernel.h +++ b/src/SeparableConvolution_kernel.h @@ -4,9 +4,9 @@ void SeparableConvolution_kernel_forward( THCState* state, - THCudaTensor* input1, - THCudaTensor* input2, - THCudaTensor* input3, + THCudaTensor* input, + THCudaTensor* vertical, + THCudaTensor* horizontal, THCudaTensor* output ); |