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-scatter
Commits
9e9b2e74
Commit
9e9b2e74
authored
Feb 14, 2020
by
rusty1s
Browse files
removed load_library calls
parent
47745860
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
0 additions
and
30 deletions
+0
-30
torch_scatter/scatter.py
torch_scatter/scatter.py
+0
-10
torch_scatter/segment_coo.py
torch_scatter/segment_coo.py
+0
-10
torch_scatter/segment_csr.py
torch_scatter/segment_csr.py
+0
-10
No files found.
torch_scatter/scatter.py
View file @
9e9b2e74
import
os
import
importlib
import
os.path
as
osp
from
typing
import
Optional
,
Tuple
import
torch
from
.utils
import
broadcast
try
:
torch
.
ops
.
load_library
(
importlib
.
machinery
.
PathFinder
().
find_spec
(
'_scatter'
,
[
osp
.
dirname
(
__file__
)]).
origin
)
except
AttributeError
as
e
:
if
os
.
getenv
(
'BUILD_DOCS'
,
'0'
)
!=
'1'
:
raise
AttributeError
(
e
)
@
torch
.
jit
.
script
def
scatter_sum
(
src
:
torch
.
Tensor
,
index
:
torch
.
Tensor
,
dim
:
int
=
-
1
,
...
...
torch_scatter/segment_coo.py
View file @
9e9b2e74
import
os
import
importlib
import
os.path
as
osp
from
typing
import
Optional
,
Tuple
import
torch
try
:
torch
.
ops
.
load_library
(
importlib
.
machinery
.
PathFinder
().
find_spec
(
'_segment_coo'
,
[
osp
.
dirname
(
__file__
)]).
origin
)
except
AttributeError
as
e
:
if
os
.
getenv
(
'BUILD_DOCS'
,
'0'
)
!=
'1'
:
raise
AttributeError
(
e
)
@
torch
.
jit
.
script
def
segment_sum_coo
(
src
:
torch
.
Tensor
,
index
:
torch
.
Tensor
,
...
...
torch_scatter/segment_csr.py
View file @
9e9b2e74
import
os
import
importlib
import
os.path
as
osp
from
typing
import
Optional
,
Tuple
import
torch
try
:
torch
.
ops
.
load_library
(
importlib
.
machinery
.
PathFinder
().
find_spec
(
'_segment_csr'
,
[
osp
.
dirname
(
__file__
)]).
origin
)
except
AttributeError
as
e
:
if
os
.
getenv
(
'BUILD_DOCS'
,
'0'
)
!=
'1'
:
raise
AttributeError
(
e
)
@
torch
.
jit
.
script
def
segment_sum_csr
(
src
:
torch
.
Tensor
,
indptr
:
torch
.
Tensor
,
...
...
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