Skip to content
Snippets Groups Projects
Commit 330d4b1a authored by Yaman Umuroglu's avatar Yaman Umuroglu
Browse files

[Transformation] comment + check empty name in GiveReadableTensorNames

parent 45726278
No related branches found
No related tags found
No related merge requests found
...@@ -111,8 +111,9 @@ class GiveRandomTensorNames(Transformation): ...@@ -111,8 +111,9 @@ class GiveRandomTensorNames(Transformation):
class GiveReadableTensorNames(Transformation): class GiveReadableTensorNames(Transformation):
"""Give more human-readable names to all internal tensors. It's recommended """Give more human-readable names to all internal tensors. You should
to apply give_unique_node_names prior to this transform.""" apply GiveUniqueNodeNames prior to this transform to avoid empty node names,
as the readable names are based on the node names."""
def apply(self, model): def apply(self, model):
# to ensure we can use rename_tensor safely (without renaming existing # to ensure we can use rename_tensor safely (without renaming existing
...@@ -120,6 +121,7 @@ class GiveReadableTensorNames(Transformation): ...@@ -120,6 +121,7 @@ class GiveReadableTensorNames(Transformation):
model = model.transform(GiveRandomTensorNames()) model = model.transform(GiveRandomTensorNames())
graph = model.graph graph = model.graph
for n in graph.node: for n in graph.node:
assert n.name != "", "Found empty node name"
out_num = 0 out_num = 0
for o in n.output: for o in n.output:
model.rename_tensor(o, "%s_out%d" % (n.name, out_num)) model.rename_tensor(o, "%s_out%d" % (n.name, out_num))
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment