src/SeparableConvolution_cuda.c


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

#include <THC.h>
#include <THCGeneral.h>

#include "SeparableConvolution_kernel.h"

extern THCState* state;

int SeparableConvolution_cuda_forward(
	THCudaTensor* input1,
	THCudaTensor* input2,
	THCudaTensor* input3,
	THCudaTensor* output
) {
	SeparableConvolution_kernel_forward(
		state,
		input1,
		input2,
		input3,
		output
	);

	return 1;
}