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
103ef0f2
Commit
103ef0f2
authored
Dec 27, 2017
by
rusty1s
Browse files
loop over E and K
parent
7203c6ca
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
21 additions
and
0 deletions
+21
-0
torch_spline_conv/src/generic/cpu.c
torch_spline_conv/src/generic/cpu.c
+21
-0
No files found.
torch_spline_conv/src/generic/cpu.c
View file @
103ef0f2
...
@@ -3,12 +3,33 @@
...
@@ -3,12 +3,33 @@
#else
#else
void
spline_
(
linear
)(
THFloatTensor
*
amount
,
THLongTensor
*
index
,
THFloatTensor
*
input
,
THLongTensor
*
kernel
,
THByteTensor
*
open
)
{
void
spline_
(
linear
)(
THFloatTensor
*
amount
,
THLongTensor
*
index
,
THFloatTensor
*
input
,
THLongTensor
*
kernel
,
THByteTensor
*
open
)
{
// s = (m+1)^d
// amount: E x s
// index: E x s
// input: E x d
// kernel: d
// open: d
//
int64_t
i
,
d
;
int64_t
E
=
THLongTensor_size
(
index
,
0
);
int64_t
K
=
THLongTensor_size
(
index
,
1
);
int64_t
D
=
THLongTensor_size
(
kernel
,
0
);
for
(
i
=
0
;
i
<
E
*
K
;
i
++
)
{
for
(
d
=
0
;
d
<
D
;
d
++
)
{
}
}
}
}
void
spline_
(
quadratic
)(
THFloatTensor
*
amount
,
THLongTensor
*
index
,
THFloatTensor
*
input
,
THLongTensor
*
kernel
,
THByteTensor
*
open
)
{
void
spline_
(
quadratic
)(
THFloatTensor
*
amount
,
THLongTensor
*
index
,
THFloatTensor
*
input
,
THLongTensor
*
kernel
,
THByteTensor
*
open
)
{
int64_t
i
;
for
(
i
=
0
;
i
<
THLongTensor_size
(
input
,
dim
);
i
++
)
{
}
}
}
void
spline_
(
cubic
)(
THFloatTensor
*
amount
,
THLongTensor
*
index
,
THFloatTensor
*
input
,
THLongTensor
*
kernel
,
THByteTensor
*
open
)
{
void
spline_
(
cubic
)(
THFloatTensor
*
amount
,
THLongTensor
*
index
,
THFloatTensor
*
input
,
THLongTensor
*
kernel
,
THByteTensor
*
open
)
{
int64_t
i
;
for
(
i
=
0
;
i
<
THLongTensor_size
(
input
,
dim
);
i
++
)
{
}
}
}
#endif
#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