提交 a5f7fe07 authored 作者: Frederic Bastien's avatar Frederic Bastien

Fix d3viz following OpFromGraph changes.

上级 ca0a627d
...@@ -241,6 +241,7 @@ class PyDotFormatter(object): ...@@ -241,6 +241,7 @@ class PyDotFormatter(object):
gf = PyDotFormatter() gf = PyDotFormatter()
# Use different node prefix for sub-graphs # Use different node prefix for sub-graphs
gf.__node_prefix = __node_id gf.__node_prefix = __node_id
node.op.prepare_node(node, None, None, 'py')
gf(node.op.fn, subgraph) gf(node.op.fn, subgraph)
graph.add_subgraph(subgraph) graph.add_subgraph(subgraph)
pd_node.get_attributes()['subg'] = subgraph.get_name() pd_node.get_attributes()['subg'] = subgraph.get_name()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论