提交 4a6132af authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Fix Alloc's R op.

上级 167cf345
...@@ -2494,10 +2494,9 @@ class Alloc(gof.Op): ...@@ -2494,10 +2494,9 @@ class Alloc(gof.Op):
return ret return ret
def R_op(self, inputs, eval_points): def R_op(self, inputs, eval_points):
# There is only one entry in eval_points and inputs if eval_points[0] is None:
if None in eval_points:
return [None] return [None]
return self.make_node(*eval_points).outputs return self.make_node(eval_points[0], *inputs[1:]).outputs
alloc = Alloc() alloc = Alloc()
pprint.assign(alloc, printing.FunctionPrinter('alloc')) pprint.assign(alloc, printing.FunctionPrinter('alloc'))
...@@ -5335,5 +5334,3 @@ class Outer(Op): ...@@ -5335,5 +5334,3 @@ class Outer(Op):
def __str__(self): def __str__(self):
return "outer" return "outer"
outer = Outer() outer = Outer()
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论