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

Fixed owner.inputs return value

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