From 1b0aa1f5e3209c3f757aa558c6695fc0ca7c5b49 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Tue, 21 Jan 2020 22:54:04 +0000
Subject: [PATCH] [ModelWrapper] add get/set for metadata_prop
---
src/finn/core/modelwrapper.py | 19 +++++++++++++++++++
1 file changed, 19 insertions(+)
diff --git a/src/finn/core/modelwrapper.py b/src/finn/core/modelwrapper.py
index cd2a2ec59..483fc0bcf 100644
--- a/src/finn/core/modelwrapper.py
+++ b/src/finn/core/modelwrapper.py
@@ -285,6 +285,25 @@ class ModelWrapper:
fanout += 1
return fanout
+ def get_metadata_prop(self, key):
+ """Returns the value associated with metadata_prop with given key,
+ or None otherwise."""
+ metadata_prop = util.get_by_name(self.model.metadata_props, key, "key")
+ if metadata_prop is None:
+ return None
+ else:
+ return metadata_prop.value
+
+ def set_metadata_prop(self, key, value):
+ metadata_prop = util.get_by_name(self.model.metadata_props, key, "key")
+ if metadata_prop is None:
+ metadata_prop = onnx.StringStringEntryProto()
+ metadata_prop.key = key
+ metadata_prop.value = value
+ self.model.metadata_props.append(metadata_prop)
+ else:
+ metadata_prop.value = value
+
def set_attribute(self, node, attribute_name, value):
"""Sets a custom node attribute of given name with given value"""
"""Data types of attributes in onnx are encoded:
--
GitLab