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
9bf31c6b
Commit
9bf31c6b
authored
Nov 22, 2018
by
AntoinePrv
Browse files
Replace saved_variables to saved_tensors.
parent
db574780
Changes
4
Show whitespace changes
Inline
Side-by-side
Showing
4 changed files
with
4 additions
and
4 deletions
+4
-4
torch_scatter/div.py
torch_scatter/div.py
+1
-1
torch_scatter/max.py
torch_scatter/max.py
+1
-1
torch_scatter/min.py
torch_scatter/min.py
+1
-1
torch_scatter/mul.py
torch_scatter/mul.py
+1
-1
No files found.
torch_scatter/div.py
View file @
9bf31c6b
...
@@ -18,7 +18,7 @@ class ScatterDiv(Function):
...
@@ -18,7 +18,7 @@ class ScatterDiv(Function):
@
staticmethod
@
staticmethod
def
backward
(
ctx
,
grad_out
):
def
backward
(
ctx
,
grad_out
):
out
,
src
,
index
=
ctx
.
saved_
variable
s
out
,
src
,
index
=
ctx
.
saved_
tensor
s
grad_src
=
None
grad_src
=
None
if
ctx
.
needs_input_grad
[
1
]:
if
ctx
.
needs_input_grad
[
1
]:
...
...
torch_scatter/max.py
View file @
9bf31c6b
...
@@ -19,7 +19,7 @@ class ScatterMax(Function):
...
@@ -19,7 +19,7 @@ class ScatterMax(Function):
@
staticmethod
@
staticmethod
def
backward
(
ctx
,
grad_out
,
grad_arg
):
def
backward
(
ctx
,
grad_out
,
grad_arg
):
index
,
arg
=
ctx
.
saved_
variable
s
index
,
arg
=
ctx
.
saved_
tensor
s
grad_src
=
None
grad_src
=
None
if
ctx
.
needs_input_grad
[
1
]:
if
ctx
.
needs_input_grad
[
1
]:
...
...
torch_scatter/min.py
View file @
9bf31c6b
...
@@ -19,7 +19,7 @@ class ScatterMin(Function):
...
@@ -19,7 +19,7 @@ class ScatterMin(Function):
@
staticmethod
@
staticmethod
def
backward
(
ctx
,
grad_out
,
grad_arg
):
def
backward
(
ctx
,
grad_out
,
grad_arg
):
index
,
arg
=
ctx
.
saved_
variable
s
index
,
arg
=
ctx
.
saved_
tensor
s
grad_src
=
None
grad_src
=
None
if
ctx
.
needs_input_grad
[
1
]:
if
ctx
.
needs_input_grad
[
1
]:
...
...
torch_scatter/mul.py
View file @
9bf31c6b
...
@@ -18,7 +18,7 @@ class ScatterMul(Function):
...
@@ -18,7 +18,7 @@ class ScatterMul(Function):
@
staticmethod
@
staticmethod
def
backward
(
ctx
,
grad_out
):
def
backward
(
ctx
,
grad_out
):
out
,
src
,
index
=
ctx
.
saved_
variable
s
out
,
src
,
index
=
ctx
.
saved_
tensor
s
grad_src
=
None
grad_src
=
None
if
ctx
.
needs_input_grad
[
1
]:
if
ctx
.
needs_input_grad
[
1
]:
...
...
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