提交 8a41c263 authored 作者: Frederic Bastien's avatar Frederic Bastien

Fix a bug introduced in gh-4865

上级 dd9adf80
...@@ -5024,12 +5024,12 @@ def local_useless_elemwise_comparison(node): ...@@ -5024,12 +5024,12 @@ def local_useless_elemwise_comparison(node):
return return
def zeros_like(model, dtype): def zeros_like(model, dtype):
ret = T.zeros_like(node.inputs[0], dtype=node.outputs[0].dtype) ret = T.zeros_like(model, dtype=node.outputs[0].dtype)
ret = pre_greedy_local_optimizer([local_useless_fill], ret) ret = pre_greedy_local_optimizer([local_useless_fill], ret)
return ret return ret
def ones_like(model, dtype): def ones_like(model, dtype):
ret = T.ones_like(node.inputs[0], dtype=node.outputs[0].dtype) ret = T.ones_like(model, dtype=node.outputs[0].dtype)
ret = pre_greedy_local_optimizer([local_useless_fill], ret) ret = pre_greedy_local_optimizer([local_useless_fill], ret)
return ret return ret
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论