提交 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):
return
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)
return ret
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)
return ret
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论