From 153c2d4e8f15bfab81d6dca4261fee72739419b8 Mon Sep 17 00:00:00 2001
From: mmrahorovic <mmrahorovic@hotmail.com>
Date: Mon, 31 Jul 2023 12:25:43 +0100
Subject: [PATCH] [custom op]: update tensor datatype for consistency
---
src/finn/custom_op/fpgadataflow/thresholding_batch.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/src/finn/custom_op/fpgadataflow/thresholding_batch.py b/src/finn/custom_op/fpgadataflow/thresholding_batch.py
index 3bcc5c05c..72ee2f7af 100644
--- a/src/finn/custom_op/fpgadataflow/thresholding_batch.py
+++ b/src/finn/custom_op/fpgadataflow/thresholding_batch.py
@@ -211,6 +211,8 @@ class Thresholding_Batch(HLSCustomOp):
threshold_tensor
).all(), "Thresholds can't be expressed with type %s" % str(tdt)
self.set_nodeattr("weightDataType", tdt.name)
+ # Update QONNX DataType of tensor for consistency
+ model.set_tensor_datatype(self.onnx_node.input[1], tdt)
return DataType[self.get_nodeattr("weightDataType")]
def get_instream_width(self, ind=0):
--
GitLab