From f61c2705ad97273a9409a2aff427470ac131f596 Mon Sep 17 00:00:00 2001 From: Matthew Bentham Date: Tue, 23 Apr 2019 16:43:27 +0100 Subject: MLCE-117 Add a unit test for implicit flatten of FC layer input Change-Id: Ia4dd63927a54aa0cc24d5a378f30189c957f12e8 Signed-off-by: Matthew Bentham --- test/1.0/FullyConnectedReshape.cpp | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) create mode 100644 test/1.0/FullyConnectedReshape.cpp (limited to 'test/1.0') 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_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() -- cgit v1.2.1