提交 4cdb561f authored 作者: David Warde-Farley's avatar David Warde-Farley

Better naming for IfElse created by infer_shape.

上级 e0ad27db
...@@ -129,12 +129,15 @@ class IfElse(PureOp): ...@@ -129,12 +129,15 @@ class IfElse(PureOp):
assert len(new_ts_inputs) == len(new_fs_inputs) assert len(new_ts_inputs) == len(new_fs_inputs)
if len(new_ts_inputs + new_fs_inputs) > 0: if len(new_ts_inputs + new_fs_inputs) > 0:
name_tokens = ['shape']
if self.name is not None:
name_tokens.append(self.name)
new_ifelse = IfElse( new_ifelse = IfElse(
n_outs=len(new_ts_inputs), n_outs=len(new_ts_inputs),
as_view=False, as_view=False,
gpu=False, gpu=False,
name='shape_' + str(self.name)) name='_'.join(name_tokens))
new_outs = new_ifelse.make_node(node.inputs[0], new_outs = new_ifelse.make_node(node.inputs[0],
*(new_ts_inputs + new_fs_inputs)).outputs *(new_ts_inputs + new_fs_inputs)).outputs
else: else:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论