blob: 250f88371e1e6741fc143d78516c54abef1d3607 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
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()
|