提交 8e0958a7 authored 作者: Ricardo Vieira's avatar Ricardo Vieira 提交者: Ricardo Vieira

Git rid of new `"output"` uses

Useless since 9ba6d99f
上级 79862c63
...@@ -509,8 +509,6 @@ def svd_uv_merge(fgraph, node): ...@@ -509,8 +509,6 @@ def svd_uv_merge(fgraph, node):
# Else, has to replace the s of this node with s of an SVD Op that compute_uv=False. # Else, has to replace the s of this node with s of an SVD Op that compute_uv=False.
# First, iterate to see if there is an SVD Op that can be reused. # First, iterate to see if there is an SVD Op that can be reused.
for cl, _ in fgraph.clients[x]: for cl, _ in fgraph.clients[x]:
if cl == "output":
continue
if isinstance(cl.op, Blockwise) and isinstance(cl.op.core_op, SVD): if isinstance(cl.op, Blockwise) and isinstance(cl.op.core_op, SVD):
if not cl.op.core_op.compute_uv: if not cl.op.core_op.compute_uv:
return { return {
...@@ -529,8 +527,6 @@ def svd_uv_merge(fgraph, node): ...@@ -529,8 +527,6 @@ def svd_uv_merge(fgraph, node):
# We want rewrite if there is another one with compute_uv=True. # We want rewrite if there is another one with compute_uv=True.
# For this case, just reuse the `s` from the one with compute_uv=True. # For this case, just reuse the `s` from the one with compute_uv=True.
for cl, _ in fgraph.clients[x]: for cl, _ in fgraph.clients[x]:
if cl == "output":
continue
if isinstance(cl.op, Blockwise) and isinstance(cl.op.core_op, SVD): if isinstance(cl.op, Blockwise) and isinstance(cl.op.core_op, SVD):
if cl.op.core_op.compute_uv and ( if cl.op.core_op.compute_uv and (
len(fgraph.clients[cl.outputs[0]]) > 0 len(fgraph.clients[cl.outputs[0]]) > 0
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论