cpu.h 3.5 KB
Newer Older
1
void    spline_linear_basis_forward_Float ( THFloatTensor *basis, THLongTensor *weight_index,  THFloatTensor *pseudo, THLongTensor *kernel_size, THByteTensor *is_open_spline, int K);
rusty1s's avatar
rusty1s committed
2
void    spline_linear_basis_forward_Double(THDoubleTensor *basis, THLongTensor *weight_index, THDoubleTensor *pseudo, THLongTensor *kernel_size, THByteTensor *is_open_spline, int K);
3
void spline_quadratic_basis_forward_Float ( THFloatTensor *basis, THLongTensor *weight_index,  THFloatTensor *pseudo, THLongTensor *kernel_size, THByteTensor *is_open_spline, int K);
4
void spline_quadratic_basis_forward_Double(THDoubleTensor *basis, THLongTensor *weight_index, THDoubleTensor *pseudo, THLongTensor *kernel_size, THByteTensor *is_open_spline, int K);
5
void     spline_cubic_basis_forward_Float ( THFloatTensor *basis, THLongTensor *weight_index,  THFloatTensor *pseudo, THLongTensor *kernel_size, THByteTensor *is_open_spline, int K);
rusty1s's avatar
rusty1s committed
6
void     spline_cubic_basis_forward_Double(THDoubleTensor *basis, THLongTensor *weight_index, THDoubleTensor *pseudo, THLongTensor *kernel_size, THByteTensor *is_open_spline, int K);
rusty1s's avatar
rusty1s committed
7

8
void    spline_linear_basis_backward_Float ( THFloatTensor *grad_pseudo,  THFloatTensor *grad_basis,  THFloatTensor *pseudo, THLongTensor *kernel_size, THByteTensor *is_open_spline);
rusty1s's avatar
rusty1s committed
9
void    spline_linear_basis_backward_Double(THDoubleTensor *grad_pseudo, THDoubleTensor *grad_basis, THDoubleTensor *pseudo, THLongTensor *kernel_size, THByteTensor *is_open_spline);
10
void spline_quadratic_basis_backward_Float ( THFloatTensor *grad_pseudo,  THFloatTensor *grad_basis,  THFloatTensor *pseudo, THLongTensor *kernel_size, THByteTensor *is_open_spline);
rusty1s's avatar
rusty1s committed
11
void spline_quadratic_basis_backward_Double(THDoubleTensor *grad_pseudo, THDoubleTensor *grad_basis, THDoubleTensor *pseudo, THLongTensor *kernel_size, THByteTensor *is_open_spline);
12
void     spline_cubic_basis_backward_Float ( THFloatTensor *grad_pseudo,  THFloatTensor *grad_basis,  THFloatTensor *pseudo, THLongTensor *kernel_size, THByteTensor *is_open_spline);
rusty1s's avatar
rusty1s committed
13
void     spline_cubic_basis_backward_Double(THDoubleTensor *grad_pseudo, THDoubleTensor *grad_basis, THDoubleTensor *pseudo, THLongTensor *kernel_size, THByteTensor *is_open_spline);
rusty1s's avatar
rusty1s committed
14

15
void spline_weighting_forward_Float ( THFloatTensor *output, THFloatTensor *input,  THFloatTensor *weight,  THFloatTensor *basis,  THLongTensor *weight_index);
rusty1s's avatar
rusty1s committed
16
void spline_weighting_forward_Double(THDoubleTensor *output, THDoubleTensor *input, THDoubleTensor *weight, THDoubleTensor *basis, THLongTensor *weight_index);
rusty1s's avatar
rusty1s committed
17

18
void spline_weighting_backward_input_Float ( THFloatTensor *grad_input,  THFloatTensor *grad_output,  THFloatTensor *weight,  THFloatTensor *basis, THLongTensor *weight_index);
19
20
void spline_weighting_backward_input_Double(THDoubleTensor *grad_input, THDoubleTensor *grad_output, THDoubleTensor *weight, THDoubleTensor *basis, THLongTensor *weight_index);

21
void spline_weighting_backward_basis_Float ( THFloatTensor *grad_basis,  THFloatTensor *grad_output,  THFloatTensor *input,  THFloatTensor *weight, THLongTensor *weight_index);
rusty1s's avatar
rusty1s committed
22
23
void spline_weighting_backward_basis_Double(THDoubleTensor *grad_basis, THDoubleTensor *grad_output, THDoubleTensor *input, THDoubleTensor *weight, THLongTensor *weight_index);

24
void spline_weighting_backward_weight_Float ( THFloatTensor *grad_weight,  THFloatTensor *grad_output,  THFloatTensor *input,  THFloatTensor *basis, THLongTensor *weight_index);
25
26
void spline_weighting_backward_weight_Double(THDoubleTensor *grad_weight, THDoubleTensor *grad_output, THDoubleTensor *input, THDoubleTensor *basis, THLongTensor *weight_index);