Skip to content

Commit 7ad078a

Browse files
authored
Fix _rename_placeholder (#458)
* fix rename_placeholder * node.name=node.target
1 parent 7065b30 commit 7ad078a

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

graph_net/torch/fx_graph_parse_util.py

100644100755
Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -116,8 +116,8 @@ def _rename_placeholder(name, pattern2replacement):
116116
if not (name[:2] == "L_" or name[:2] == "l_"):
117117
return name
118118
name = name[2:]
119-
if name[0] == "l":
120-
name = "L" + name[1:]
119+
if name[:2] == "l_":
120+
name = "L_" + name[2:]
121121
for pattern, replacement in pattern2replacement.items():
122122
name = name.replace(pattern, replacement)
123123
return name
@@ -161,7 +161,7 @@ def get_input_names_from_placeholder():
161161
if node.op != "placeholder":
162162
continue
163163
node.target = _rename_placeholder(node.target, pattern2replacement)
164-
node.name = _rename_placeholder(node.name, pattern2replacement)
164+
node.name = node.target
165165

166166
def get_diff_input_names():
167167
placeholder_names = set(get_input_names_from_placeholder())

0 commit comments

Comments
 (0)