diff options
author | Matthew Bentham <matthew.bentham@arm.com> | 2019-04-23 16:43:27 +0100 |
---|---|---|
committer | derek.lamberti <derek.lamberti@arm.com> | 2019-04-25 12:12:48 +0000 |
commit | f61c2705ad97273a9409a2aff427470ac131f596 (patch) | |
tree | 8835dae435f174a54afe5bd41e2763d5eb1236bc /test | |
parent | 5404c01b4b0a2455aea0f5d1de0a45e2e859a466 (diff) | |
download | android-nn-driver-f61c2705ad97273a9409a2aff427470ac131f596.tar.gz |
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>
Diffstat (limited to 'test')
-rw-r--r-- | test/1.0/FullyConnectedReshape.cpp | 20 | ||||
-rw-r--r-- | test/Android.mk | 1 |
2 files changed, 21 insertions, 0 deletions
diff --git a/test/1.0/FullyConnectedReshape.cpp b/test/1.0/FullyConnectedReshape.cpp new file mode 100644 index 00000000..250f8837 --- /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() diff --git a/test/Android.mk b/test/Android.mk index da3ac706..87f0b636 100644 --- a/test/Android.mk +++ b/test/Android.mk @@ -45,6 +45,7 @@ endif # PLATFORM_VERSION == 9 LOCAL_SRC_FILES := \ 1.0/Convolution2D.cpp \ + 1.0/FullyConnectedReshape.cpp \ Tests.cpp \ UtilsTests.cpp \ Concurrent.cpp \ |