THCBasis.h 1.64 KB
Newer Older
rusty1s's avatar
rusty1s committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
#ifndef THC_GENERIC_FILE
#define THC_GENERIC_FILE "generic/THCBasis.h"
#else

void THCTensor_(linearBasisForward)(THCState *state, THCTensor *basis,
                                    THCudaLongTensor *weightIndex, THCTensor *pseudo,
                                    THCudaLongTensor *kernelSize, THCudaByteTensor *isOpenSpline);

void THCTensor_(quadraticBasisForward)(THCState *state, THCTensor *basis,
                                       THCudaLongTensor *weightIndex, THCTensor *pseudo,
                                       THCudaLongTensor *kernelSize,
                                       THCudaByteTensor *isOpenSpline);

void THCTensor_(cubicBasisForward)(THCState *state, THCTensor *basis,
                                   THCudaLongTensor *weightIndex, THCTensor *pseudo,
                                   THCudaLongTensor *kernelSize, THCudaByteTensor *isOpenSpline);

rusty1s's avatar
rusty1s committed
18
19
20
21
22
23
24
25
26
27
28
void THCTensor_(linearBasisBackward)(THCState *state, THCTensor *self, THCTensor *gradBasis,
                                     THCTensor *pseudo, THCudaLongTensor *kernelSize,
                                     THCudaByteTensor *isOpenSpline);

void THCTensor_(quadraticBasisBackward)(THCState *state, THCTensor *self, THCTensor *gradBasis,
                                        THCTensor *pseudo, THCudaLongTensor *kernelSize,
                                        THCudaByteTensor *isOpenSpline);

void THCTensor_(cubicBasisBackward)(THCState *state, THCTensor *self, THCTensor *gradBasis,
                                    THCTensor *pseudo, THCudaLongTensor *kernelSize,
                                    THCudaByteTensor *isOpenSpline);
rusty1s's avatar
rusty1s committed
29

rusty1s's avatar
rusty1s committed
30
#endif // THC_GENERIC_FILE