提交 4e1e90ea authored 作者: Olivier Mastropietro's avatar Olivier Mastropietro

Fixed owner.inputs return value

上级 d6b3b7a6
......@@ -125,7 +125,7 @@ def local_alloc_dimshuffle(node):
continue
else:
return False
return input_.inputs
return input_.owner.inputs
return False
......@@ -147,5 +147,5 @@ def local_reshape_dimshuffle(node):
continue
elif i != dim + offset:
return False
return input_.inputs
return input_.owner.inputs
return False
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论