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