diff options
Diffstat (limited to 'src/SeparableConvolution_cuda.c')
| -rw-r--r-- | src/SeparableConvolution_cuda.c | 12 |
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 ); |