commit f61c2705ad97273a9409a2aff427470ac131f596
author Matthew Bentham <matthew.bentham@arm.com> Tue Apr 23 16:43:27 2019 +0100
committer derek.lamberti <derek.lamberti@arm.com> Thu Apr 25 12:12:48 2019 +0000
tree 8835dae435f174a54afe5bd41e2763d5eb1236bc
parent 5404c01b4b0a2455aea0f5d1de0a45e2e859a466

MLCE-117 Add a unit test for implicit flatten of FC layer input

Change-Id: Ia4dd63927a54aa0cc24d5a378f30189c957f12e8
Signed-off-by: Matthew Bentham <matthew.bentham@arm.com>

test/1.0/FullyConnectedReshape.cpp

diff --git a/test/1.0/FullyConnectedReshape.cpp b/test/1.0/FullyConnectedReshape.cpp
new file mode 100644
index 0000000..250f883
--- /dev/null
+++ b/test/1.0/FullyConnectedReshape.cpp
@@ -0,0 +1,20 @@
+//
+// Copyright © 2017 Arm Ltd. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#include "../DriverTestHelpers.hpp"
+#include "../../1.0/FullyConnected.hpp"
+
+#include <boost/test/unit_test.hpp>
+
+BOOST_AUTO_TEST_SUITE(FullyConnectedReshapeTests)
+
+BOOST_AUTO_TEST_CASE(TestFlattenFullyConnectedInput)
+{
+    using armnn::TensorShape;
+    BOOST_TEST(FlattenFullyConnectedInput(TensorShape({97,1,1,2048}), TensorShape({512, 2048})) ==
+               TensorShape({97, 2048}));
+}
+
+BOOST_AUTO_TEST_SUITE_END()

test/Android.mk

diff --git a/test/Android.mk b/test/Android.mk
index da3ac70..87f0b63 100644
--- a/test/Android.mk
+++ b/test/Android.mk
@@ -45,6 +45,7 @@
 
 LOCAL_SRC_FILES := \
         1.0/Convolution2D.cpp \
+        1.0/FullyConnectedReshape.cpp \
         Tests.cpp \
         UtilsTests.cpp \
         Concurrent.cpp \