diff options
| author | sniklaus <simon.niklaus@outlook.com> | 2017-09-18 22:20:04 -0700 |
|---|---|---|
| committer | sniklaus <simon.niklaus@outlook.com> | 2017-09-18 22:20:04 -0700 |
| commit | cfd6a91a628c603eeeecf517340ac0474a126496 (patch) | |
| tree | 7c891bc8660904771bb612f56aca8f22760d0cdb /src/SeparableConvolution_cuda.h | |
| parent | e123297d61dc9915b70060def498560ca5d3d073 (diff) | |
no message
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 |