From 23a817b4e3540cdd1467e6723b88bcb7ee3bcffc Mon Sep 17 00:00:00 2001
From: auphelia <jakobapk@web.de>
Date: Thu, 2 Apr 2020 14:39:31 +0100
Subject: [PATCH] [Test] Add remove reshape node trafo to end2end cnv test
---
tests/end2end/test_end2end_cnv_w1a1.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/tests/end2end/test_end2end_cnv_w1a1.py b/tests/end2end/test_end2end_cnv_w1a1.py
index 2dce28d72..6ee7e4e77 100644
--- a/tests/end2end/test_end2end_cnv_w1a1.py
+++ b/tests/end2end/test_end2end_cnv_w1a1.py
@@ -43,6 +43,7 @@ import onnx # NOQA
from finn.core.modelwrapper import ModelWrapper
from finn.transformation.double_to_single_float import DoubleToSingleFloat
from finn.transformation.infer_shapes import InferShapes
+from finn.transformation.move_reshape import MoveReshape
from finn.transformation.fold_constants import FoldConstants
from finn.transformation.general import GiveReadableTensorNames, GiveUniqueNodeNames
from finn.transformation.streamline import Streamline
@@ -99,4 +100,5 @@ def test_end2end_cnv_w1a1_convert_to_hls_layers():
model = model.transform(to_hls.InferQuantizedStreamingFCLayer())
model = model.transform(to_hls.InferConvInpGen())
model = model.transform(to_hls.InferStreamingMaxPool())
+ model = model.transform(MoveReshape())
model.save(build_dir + "/end2end_cnv_w1a1_hls_layers.onnx")
--
GitLab