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

[Transform] add created ValueInfos to protobuf in InsertHook

parent dac25880
No related branches found
No related tags found
No related merge requests found
......@@ -84,6 +84,9 @@ class InsertHook(Transformation):
)
# insert checksum node
graph.node.insert(node_ind + 1, chk_node)
# insert newly-created tensors
graph.value_info.append(chk_otensor)
graph.value_info.append(chk_result)
# set chk output tensor as new input tensor of second node
if len(consumers) == 1:
......
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