From c2ac2947d4075d86747b60351c5e8306569866f3 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <yamanu@xilinx.com>
Date: Fri, 1 Nov 2019 15:02:37 +0000
Subject: [PATCH] [Transform] mark threshold node domain as finn
---
src/finn/transformation/streamline.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/src/finn/transformation/streamline.py b/src/finn/transformation/streamline.py
index 462005caa..313353cea 100644
--- a/src/finn/transformation/streamline.py
+++ b/src/finn/transformation/streamline.py
@@ -15,6 +15,7 @@ def convert_sign_to_thres(model):
thres_param_name = model.make_new_valueinfo_name()
thres_param = np.asarray([[0]], dtype=np.float32)
n.input.append(thres_param_name)
+ n.domain = "finn"
model.set_initializer(thres_param_name, thres_param)
# mark output tensor as bipolar
model.set_tensor_datatype(n.output[0], DataType.BIPOLAR)
--
GitLab