From 153cd9ccad171f5f3f51bc154154aa3a85c7c7fd Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <yamanu@xilinx.com>
Date: Mon, 4 Nov 2019 14:58:44 +0000
Subject: [PATCH] [Wrapper] add initializer renaming to rename_tensor
---
src/finn/core/modelwrapper.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/src/finn/core/modelwrapper.py b/src/finn/core/modelwrapper.py
index 1f8cef788..cb70545f3 100644
--- a/src/finn/core/modelwrapper.py
+++ b/src/finn/core/modelwrapper.py
@@ -183,6 +183,9 @@ class ModelWrapper:
# sweep over value_info
if util.get_by_name(graph.value_info, old_name) is not None:
util.get_by_name(graph.value_info, old_name).name = new_name
+ # sweep over initializers
+ if util.get_by_name(graph.initializer, old_name) is not None:
+ util.get_by_name(graph.initializer, old_name).name = new_name
# sweep over quantization annotations
if (
util.get_by_name(graph.quantization_annotation, old_name, "tensor_name")
--
GitLab