THCCConv.c 1.12 KB
Newer Older
rusty1s's avatar
rusty1s committed
1
2
3
4
5
6
#ifndef THC_GENERIC_FILE
#define THC_GENERIC_FILE "generic/THCCConv.c"
#else

void THCCTensor_(convForward)(THCTensor *self, THCTensor *src, THCTensor *weight, THCTensor *basis,
                              THCudaLongTensor *weightIndex) {
rusty1s's avatar
rusty1s committed
7
  THCTensor_(convForward)(state, self, src, weight, basis, weightIndex);
rusty1s's avatar
rusty1s committed
8
9
10
11
}

void THCCTensor_(convBackwardSrc)(THCTensor *self, THCTensor *gradOutput, THCTensor *weight,
                                  THCTensor *basis, THCudaLongTensor *weightIndex) {
rusty1s's avatar
rusty1s committed
12
  THCTensor_(convBackwardSrc)(state, self, gradOutput, weight, basis, weightIndex);
rusty1s's avatar
rusty1s committed
13
14
15
16
}

void THCCTensor_(convBackwardBasis)(THCTensor *self, THCTensor *gradOutput, THCTensor *src,
                                    THCTensor *weight, THCudaLongTensor *weightIndex) {
rusty1s's avatar
rusty1s committed
17
  THCTensor_(convBackwardBasis)(state, self, gradOutput, src, weight, weightIndex);
rusty1s's avatar
rusty1s committed
18
19
20
21
}

void THCCTensor_(convBackwardWeight)(THCTensor *self, THCTensor *gradOutput, THCTensor *src,
                                     THCTensor *basis, THCudaLongTensor *weightIndex) {
rusty1s's avatar
rusty1s committed
22
  THCTensor_(convBackwardWeight)(state, self, gradOutput, src, basis, weightIndex);
rusty1s's avatar
rusty1s committed
23
24
25
26
}

#endif  // THC_GENERIC_FILE