From b0e9a1d3bd2749240e2f465955a7f996d1de7de9 Mon Sep 17 00:00:00 2001
From: auphelia <jakobapk@web.de>
Date: Thu, 9 Jul 2020 11:38:35 +0100
Subject: [PATCH] [Transform] Fix comments in convert to hls layers trafo
---
.../transformation/fpgadataflow/convert_to_hls_layers.py | 5 ++---
1 file changed, 2 insertions(+), 3 deletions(-)
diff --git a/src/finn/transformation/fpgadataflow/convert_to_hls_layers.py b/src/finn/transformation/fpgadataflow/convert_to_hls_layers.py
index 877b4a0dc..859352ad4 100644
--- a/src/finn/transformation/fpgadataflow/convert_to_hls_layers.py
+++ b/src/finn/transformation/fpgadataflow/convert_to_hls_layers.py
@@ -689,8 +689,7 @@ class InferVVAU(Transformation):
# see if we have any following thresholds
consumer = model.find_consumer(mm_output)
if consumer is not None and consumer.op_type == "MultiThreshold":
- # TODO ensure integer thresholds?
- # create MVTU (i.e. including activation)
+ # create VVAU (i.e. including activation)
mt_output = consumer.output[0]
mt_out_shape = model.get_tensor_shape(mt_output)
mt_thres = consumer.input[1]
@@ -742,7 +741,7 @@ class InferVVAU(Transformation):
odt = model.get_tensor_datatype(mm_output)
model.set_tensor_shape(mm_input, mm_in_shape)
model.set_tensor_shape(mm_output, mm_out_shape)
- # create and insert new StreamingFCLayer node
+ # create and insert new VVAU node
new_node = helper.make_node(
"Vector_Vector_Activate_Batch",
[mm_input, mm_weight],
--
GitLab