Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
torch-spline-conv
Commits
e75ddcb5
Commit
e75ddcb5
authored
Mar 09, 2018
by
rusty1s
Browse files
added forward/backward boilerplate
parent
0696c8c1
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
12 additions
and
0 deletions
+12
-0
torch_spline_conv/src/cpu.h
torch_spline_conv/src/cpu.h
+6
-0
torch_spline_conv/src/generic/cpu.c
torch_spline_conv/src/generic/cpu.c
+6
-0
No files found.
torch_spline_conv/src/cpu.h
View file @
e75ddcb5
...
...
@@ -6,3 +6,9 @@ void spline_basis_quadratic_Double(THDoubleTensor *basis, THLongTensor *weight_i
void
spline_basis_cubic_Float
(
THFloatTensor
*
basis
,
THLongTensor
*
weight_index
,
THFloatTensor
*
pseudo
,
THLongTensor
*
kernel_size
,
THByteTensor
*
is_open_spline
,
int
K
);
void
spline_basis_cubic_Double
(
THDoubleTensor
*
basis
,
THLongTensor
*
weight_index
,
THDoubleTensor
*
pseudo
,
THLongTensor
*
kernel_size
,
THByteTensor
*
is_open_spline
,
int
K
);
void
spline_edgewise_forward_Float
(
THFloatTensor
*
output
,
THFloatTensor
*
input
,
THFloatTensor
*
weight
,
THFloatTensor
*
basis
,
THLongTensor
*
weight_index
);
void
spline_edgewise_backward_Double
(
THDoubleTensor
*
output
,
THDoubleTensor
*
input
,
THDoubleTensor
*
weight
,
THDoubleTensor
*
basis
,
THLongTensor
*
weight_index
);
void
spline_edgewise_backward_Float
(
THFloatTensor
*
grad_input
,
THFloatTensor
*
grad_weight
,
THFloatTensor
*
grad_output
,
THFloatTensor
*
input
,
THFloatTensor
*
weight
,
THFloatTensor
*
basis
,
THLongTensor
*
weight_index
);
void
spline_edgewise_backward_Double
(
THDoubleTensor
*
grad_input
,
THDoubleTensor
*
grad_weight
,
THDoubleTensor
*
grad_output
,
THDoubleTensor
*
input
,
THDoubleTensor
*
weight
,
THDoubleTensor
*
basis
,
THLongTensor
*
weight_index
);
torch_spline_conv/src/generic/cpu.c
View file @
e75ddcb5
...
...
@@ -50,4 +50,10 @@ void spline_(basis_cubic)(THTensor *basis, THLongTensor *weight_index, THTensor
)
}
void
spline_
(
edgewise_forward
)(
THTensor
*
output
,
THTensor
*
input
,
THTensor
*
weight
,
THTensor
*
basis
,
THLongTensor
*
weight_index
)
{
}
void
spline_
(
edgewise_backward
)(
THTensor
*
grad_input
,
THTensor
*
grad_weight
,
THTensor
*
grad_output
,
THTensor
*
input
,
THTensor
*
weight
,
THTensor
*
basis
,
THLongTensor
*
weight_index
)
{
}
#endif
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment