From 779ebda87751d0a50250f4be68670af13f3a29c2 Mon Sep 17 00:00:00 2001
From: auphelia <jakobapk@web.de>
Date: Wed, 27 May 2020 16:42:18 +0100
Subject: [PATCH] [Core - Modelwrapper] Change name of variable in sparsity
utility fct
---
src/finn/core/modelwrapper.py | 14 +++++++-------
1 file changed, 7 insertions(+), 7 deletions(-)
diff --git a/src/finn/core/modelwrapper.py b/src/finn/core/modelwrapper.py
index 8a330a88c..2896b09e0 100644
--- a/src/finn/core/modelwrapper.py
+++ b/src/finn/core/modelwrapper.py
@@ -530,16 +530,16 @@ class ModelWrapper:
qnt_annotations = graph.quantization_annotation
ret = util.get_by_name(qnt_annotations, tensor_name, "tensor_name")
if ret is not None:
- ret_tl = util.get_by_name(
+ ret_ts = util.get_by_name(
ret.quant_parameter_tensor_names, "tensor_sparsity", "key"
)
- if ret_tl is not None:
- ret_tl.value = str(sparsity_dict)
+ if ret_ts is not None:
+ ret_ts.value = str(sparsity_dict)
else:
- tl = onnx.StringStringEntryProto()
- tl.key = "tensor_sparsity"
- tl.value = str(sparsity_dict)
- ret.quant_parameter_tensor_names.append(tl)
+ ts = onnx.StringStringEntryProto()
+ ts.key = "tensor_sparsity"
+ ts.value = str(sparsity_dict)
+ ret.quant_parameter_tensor_names.append(ts)
else:
qa = onnx.TensorAnnotation()
dt = onnx.StringStringEntryProto()
--
GitLab