Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
THBasis.h 2 KB
Newer Older
rusty1s's avatar
new aten build  
rusty1s committed
1
2
3
4
5
6 
void     THFloatTensor_linearBasisForward( THFloatTensor *basis, THLongTensor *weightIndex,  THFloatTensor *pseudo, THLongTensor *kernelSize, THByteTensor *isOpenSpline);
void    THDoubleTensor_linearBasisForward(THDoubleTensor *basis, THLongTensor *weightIndex, THDoubleTensor *pseudo, THLongTensor *kernelSize, THByteTensor *isOpenSpline);
void  THFloatTensor_quadraticBasisForward( THFloatTensor *basis, THLongTensor *weightIndex,  THFloatTensor *pseudo, THLongTensor *kernelSize, THByteTensor *isOpenSpline);
void THDoubleTensor_quadraticBasisForward(THDoubleTensor *basis, THLongTensor *weightIndex, THDoubleTensor *pseudo, THLongTensor *kernelSize, THByteTensor *isOpenSpline);
void      THFloatTensor_cubicBasisForward( THFloatTensor *basis, THLongTensor *weightIndex,  THFloatTensor *pseudo, THLongTensor *kernelSize, THByteTensor *isOpenSpline);
void     THDoubleTensor_cubicBasisForward(THDoubleTensor *basis, THLongTensor *weightIndex, THDoubleTensor *pseudo, THLongTensor *kernelSize, THByteTensor *isOpenSpline);
rusty1s's avatar
backward boilerplate  
rusty1s committed
7
8
9
10
11
12
13 

void     THFloatTensor_linearBasisBackward( THFloatTensor *self,  THFloatTensor *gradBasis,  THFloatTensor *pseudo, THLongTensor *kernelSize, THByteTensor *isOpenSpline);
void    THDoubleTensor_linearBasisBackward(THDoubleTensor *self, THDoubleTensor *gradBasis, THDoubleTensor *pseudo, THLongTensor *kernelSize, THByteTensor *isOpenSpline);
void  THFloatTensor_quadraticBasisBackward( THFloatTensor *self,  THFloatTensor *gradBasis,  THFloatTensor *pseudo, THLongTensor *kernelSize, THByteTensor *isOpenSpline);
void THDoubleTensor_quadraticBasisBackward(THDoubleTensor *self, THDoubleTensor *gradBasis, THDoubleTensor *pseudo, THLongTensor *kernelSize, THByteTensor *isOpenSpline);
void      THFloatTensor_cubicBasisBackward( THFloatTensor *self,  THFloatTensor *gradBasis,  THFloatTensor *pseudo, THLongTensor *kernelSize, THByteTensor *isOpenSpline);
void     THDoubleTensor_cubicBasisBackward(THDoubleTensor *self, THDoubleTensor *gradBasis, THDoubleTensor *pseudo, THLongTensor *kernelSize, THByteTensor *isOpenSpline);