diff --git a/src/finn/transformation/fpgadataflow/insert_hook.py b/src/finn/transformation/fpgadataflow/insert_hook.py index 22050c008b2991671b8483404e1a6e8772de691e..c1fce40c574eb58b67e728b78d31454f0c709b78 100644 --- a/src/finn/transformation/fpgadataflow/insert_hook.py +++ b/src/finn/transformation/fpgadataflow/insert_hook.py @@ -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: