From 078c8ef065721cc7fc2014433a915f53b92619bc Mon Sep 17 00:00:00 2001
From: auphelia <jakobapk@web.de>
Date: Mon, 15 Jun 2020 09:19:57 +0100
Subject: [PATCH] [CustomOp] Delte rounding in execute node of QuantAvgPool2d
---
src/finn/custom_op/quantavgpool2d.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/finn/custom_op/quantavgpool2d.py b/src/finn/custom_op/quantavgpool2d.py
index 2a8d3f16d..3bc328a9f 100644
--- a/src/finn/custom_op/quantavgpool2d.py
+++ b/src/finn/custom_op/quantavgpool2d.py
@@ -66,7 +66,7 @@ class QuantAvgPool2d(CustomOp):
outputs=[outp],
)
model_avgpool = helper.make_model(graph_avgpool)
- idict = {node.input[0]: np.round(context[node.input[0]])}
+ idict = {node.input[0]: context[node.input[0]]}
sess = rt.InferenceSession(model_avgpool.SerializeToString())
result_temp = sess.run(None, idict)
# remove scaling introduced by average
--
GitLab