Matthew Bentham | f61c270 | 2019-04-23 16:43:27 +0100 | [diff] [blame^] | 1 | // |
| 2 | // Copyright © 2017 Arm Ltd. All rights reserved. |
| 3 | // SPDX-License-Identifier: MIT |
| 4 | // |
| 5 | |
| 6 | #include "../DriverTestHelpers.hpp" |
| 7 | #include "../../1.0/FullyConnected.hpp" |
| 8 | |
| 9 | #include <boost/test/unit_test.hpp> |
| 10 | |
| 11 | BOOST_AUTO_TEST_SUITE(FullyConnectedReshapeTests) |
| 12 | |
| 13 | BOOST_AUTO_TEST_CASE(TestFlattenFullyConnectedInput) |
| 14 | { |
| 15 | using armnn::TensorShape; |
| 16 | BOOST_TEST(FlattenFullyConnectedInput(TensorShape({97,1,1,2048}), TensorShape({512, 2048})) == |
| 17 | TensorShape({97, 2048})); |
| 18 | } |
| 19 | |
| 20 | BOOST_AUTO_TEST_SUITE_END() |