Commit 9bf31c6b authored by AntoinePrv's avatar AntoinePrv
Browse files

Replace saved_variables to saved_tensors.

parent db574780
......@@ -18,7 +18,7 @@ class ScatterDiv(Function):
@staticmethod
def backward(ctx, grad_out):
out, src, index = ctx.saved_variables
out, src, index = ctx.saved_tensors
grad_src = None
if ctx.needs_input_grad[1]:
......
......@@ -19,7 +19,7 @@ class ScatterMax(Function):
@staticmethod
def backward(ctx, grad_out, grad_arg):
index, arg = ctx.saved_variables
index, arg = ctx.saved_tensors
grad_src = None
if ctx.needs_input_grad[1]:
......
......@@ -19,7 +19,7 @@ class ScatterMin(Function):
@staticmethod
def backward(ctx, grad_out, grad_arg):
index, arg = ctx.saved_variables
index, arg = ctx.saved_tensors
grad_src = None
if ctx.needs_input_grad[1]:
......
......@@ -18,7 +18,7 @@ class ScatterMul(Function):
@staticmethod
def backward(ctx, grad_out):
out, src, index = ctx.saved_variables
out, src, index = ctx.saved_tensors
grad_src = None
if ctx.needs_input_grad[1]:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment