summaryrefslogtreecommitdiff
path: root/src/SeparableConvolution_cuda.h
diff options
context:
space:
mode:
Diffstat (limited to 'src/SeparableConvolution_cuda.h')
-rw-r--r--src/SeparableConvolution_cuda.h6
1 files changed, 6 insertions, 0 deletions
diff --git a/src/SeparableConvolution_cuda.h b/src/SeparableConvolution_cuda.h
new file mode 100644
index 0000000..7320fee
--- /dev/null
+++ b/src/SeparableConvolution_cuda.h
@@ -0,0 +1,6 @@
+int SeparableConvolution_cuda_forward(
+ THCudaTensor* input1,
+ THCudaTensor* input2,
+ THCudaTensor* input3,
+ THCudaTensor* output
+); \ No newline at end of file