From fd5f7f692700836ef68bbba156020e6b25cdf297 Mon Sep 17 00:00:00 2001
From: auphelia <jakobapk@web.de>
Date: Wed, 11 Dec 2019 10:41:39 +0000
Subject: [PATCH] [multithreshold] filled verify_node function
---
src/finn/custom_op/multithreshold.py | 46 ++++++++++++++++++++++++++++
1 file changed, 46 insertions(+)
diff --git a/src/finn/custom_op/multithreshold.py b/src/finn/custom_op/multithreshold.py
index 3cff87b3f..52cf2504b 100644
--- a/src/finn/custom_op/multithreshold.py
+++ b/src/finn/custom_op/multithreshold.py
@@ -85,3 +85,49 @@ class MultiThreshold(CustomOp):
output = multithreshold(v, thresholds, out_scale, out_bias)
# setting context according to output
context[node.output[0]] = output
+
+ def verify_node(self):
+ info_messages = []
+
+ # verify number of attributes
+ num_of_attr = 3
+ if len(self.onnx_node.attribute) == num_of_attr:
+ info_messages.append("The number of attributes is correct")
+ else:
+ info_messages.append(
+ """The number of attributes is incorrect,
+ {} should have {} attributes""".format(
+ self.onnx_node.op_type, num_of_attr
+ )
+ )
+
+ # verify that "domain" is set to "finn"
+ domain_value = self.onnx_node.domain
+ if domain_value == "finn":
+ info_messages.append("Attribute domain is set correctly")
+ else:
+ info_messages.append('Attribute domain should be set to "finn"')
+
+ # verify that all necessary attributes exist
+ try:
+ self.get_nodeattr("out_scale")
+ self.get_nodeattr("out_bias")
+ self.get_nodeattr("out_dtype")
+ info_messages.append("All necessary attributes exist")
+ except Exception:
+ info_messages.append(
+ """The necessary attributes do not exist.
+ MultiThreshold needs the following attributes:
+ out_scale, out_bias, out_dtype"""
+ )
+
+ # verify the number of inputs
+ if len(self.onnx_node.input) == 2:
+ info_messages.append("The number of inputs is correct")
+ else:
+ info_messages.append(
+ """MultiThreshold needs 2 inputs
+ (data input and threshold values)"""
+ )
+
+ return info_messages
--
GitLab