• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 //
2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #include "../DriverTestHelpers.hpp"
7 #include "../Convolution2D.hpp"
8 #include "../../1.1/HalPolicy.hpp"
9 
10 #include <boost/test/unit_test.hpp>
11 #include <log/log.h>
12 
13 #include <OperationsUtils.h>
14 
15 BOOST_AUTO_TEST_SUITE(Convolution2DTests)
16 
17 using namespace android::hardware;
18 using namespace driverTestHelpers;
19 using namespace armnn_driver;
20 
21 namespace driverTestHelpers
22 {
23 
SetModelFp16Flag(V1_1::Model & model,bool fp16Enabled)24 void SetModelFp16Flag(V1_1::Model& model, bool fp16Enabled)
25 {
26     // Set the fp16 flag in the given model
27     model.relaxComputationFloat32toFloat16 = fp16Enabled;
28 }
29 
30 } // namespace driverTestHelpers
31 
BOOST_AUTO_TEST_CASE(ConvValidPadding_Hal_1_1)32 BOOST_AUTO_TEST_CASE(ConvValidPadding_Hal_1_1)
33 {
34     PaddingTestImpl<hal_1_1::HalPolicy>(android::nn::kPaddingValid);
35 }
36 
BOOST_AUTO_TEST_CASE(ConvSamePadding_Hal_1_1)37 BOOST_AUTO_TEST_CASE(ConvSamePadding_Hal_1_1)
38 {
39     PaddingTestImpl<hal_1_1::HalPolicy>(android::nn::kPaddingSame);
40 }
41 
BOOST_AUTO_TEST_CASE(ConvValidPaddingFp16Flag_Hal_1_1)42 BOOST_AUTO_TEST_CASE(ConvValidPaddingFp16Flag_Hal_1_1)
43 {
44     PaddingTestImpl<hal_1_1::HalPolicy>(android::nn::kPaddingValid, true);
45 }
46 
BOOST_AUTO_TEST_CASE(ConvSamePaddingFp16Flag_Hal_1_1)47 BOOST_AUTO_TEST_CASE(ConvSamePaddingFp16Flag_Hal_1_1)
48 {
49     PaddingTestImpl<hal_1_1::HalPolicy>(android::nn::kPaddingSame, true);
50 }
51 
52 BOOST_AUTO_TEST_SUITE_END()
53