From b6a7f60bb7108fe83be7e501bb16dea98e5d3423 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <yamanu@xilinx.com>
Date: Thu, 18 Jun 2020 15:49:25 +0100
Subject: [PATCH] [Transform] set output tensor dt in
InferChannelwiseLinearLayer
---
src/finn/transformation/fpgadataflow/convert_to_hls_layers.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/src/finn/transformation/fpgadataflow/convert_to_hls_layers.py b/src/finn/transformation/fpgadataflow/convert_to_hls_layers.py
index 0cd3471ed..c34a7ef23 100644
--- a/src/finn/transformation/fpgadataflow/convert_to_hls_layers.py
+++ b/src/finn/transformation/fpgadataflow/convert_to_hls_layers.py
@@ -577,6 +577,7 @@ class InferChannelwiseLinearLayer(Transformation):
else:
odt = DataType.get_smallest_possible(idt.max() * idt.max())
model.set_initializer(ll_const, ll_cinit.reshape(ch))
+ model.set_tensor_datatype(ll_output, odt)
# create and insert node
new_node = helper.make_node(
"ChannelwiseOp_Batch",
--
GitLab