diff options
Diffstat (limited to 'src/SeparableConvolution_cuda.h')
| -rw-r--r-- | src/SeparableConvolution_cuda.h | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/SeparableConvolution_cuda.h b/src/SeparableConvolution_cuda.h index 7320fee..a3d5ed0 100644 --- a/src/SeparableConvolution_cuda.h +++ b/src/SeparableConvolution_cuda.h @@ -1,6 +1,6 @@ int SeparableConvolution_cuda_forward( - THCudaTensor* input1, - THCudaTensor* input2, - THCudaTensor* input3, + THCudaTensor* input, + THCudaTensor* vertical, + THCudaTensor* horizontal, THCudaTensor* output );
\ No newline at end of file |