ArmNN
 21.02
NeonLayerSupportTests.cpp
Go to the documentation of this file.
1 //
2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #include "NeonWorkloadFactoryHelper.hpp"
7 
8 #include <layers/ConvertFp16ToFp32Layer.hpp>
9 #include <layers/ConvertFp32ToFp16Layer.hpp>
10 #include <test/TensorHelpers.hpp>
11 
12 #include <backendsCommon/CpuTensorHandle.hpp>
13 #include <neon/NeonWorkloadFactory.hpp>
14 #include <backendsCommon/test/IsLayerSupportedTestImpl.hpp>
15 #include <backendsCommon/test/LayerTests.hpp>
16 
17 #include <boost/test/unit_test.hpp>
18 
19 #include <string>
20 
21 BOOST_AUTO_TEST_SUITE(NeonLayerSupport)
22 
23 BOOST_AUTO_TEST_CASE(IsLayerSupportedFloat16Neon)
24 {
25  armnn::NeonWorkloadFactory factory =
26  NeonWorkloadFactoryHelper::GetFactory(NeonWorkloadFactoryHelper::GetMemoryManager());
27  IsLayerSupportedTests<armnn::NeonWorkloadFactory, armnn::DataType::Float16>(&factory);
28 }
29 
30 BOOST_AUTO_TEST_CASE(IsLayerSupportedFloat32Neon)
31 {
32  armnn::NeonWorkloadFactory factory =
33  NeonWorkloadFactoryHelper::GetFactory(NeonWorkloadFactoryHelper::GetMemoryManager());
34  IsLayerSupportedTests<armnn::NeonWorkloadFactory, armnn::DataType::Float32>(&factory);
35 }
36 
37 BOOST_AUTO_TEST_CASE(IsLayerSupportedQAsymmU8Neon)
38 {
39  armnn::NeonWorkloadFactory factory =
40  NeonWorkloadFactoryHelper::GetFactory(NeonWorkloadFactoryHelper::GetMemoryManager());
41  IsLayerSupportedTests<armnn::NeonWorkloadFactory, armnn::DataType::QAsymmU8>(&factory);
42 }
43 
44 BOOST_AUTO_TEST_CASE(IsLayerSupportedQAsymmS8Neon)
45 {
46  armnn::NeonWorkloadFactory factory =
47  NeonWorkloadFactoryHelper::GetFactory(NeonWorkloadFactoryHelper::GetMemoryManager());
48  IsLayerSupportedTests<armnn::NeonWorkloadFactory, armnn::DataType::QAsymmS8>(&factory);
49 }
50 
51 BOOST_AUTO_TEST_CASE(IsLayerSupportedQSymmS8Neon)
52 {
53  armnn::NeonWorkloadFactory factory =
54  NeonWorkloadFactoryHelper::GetFactory(NeonWorkloadFactoryHelper::GetMemoryManager());
55  IsLayerSupportedTests<armnn::NeonWorkloadFactory, armnn::DataType::QSymmS8>(&factory);
56 }
57 
58 BOOST_AUTO_TEST_CASE(IsConvertFp16ToFp32SupportedNeon)
59 {
60  std::string reasonIfUnsupported;
61 
62  bool result = IsConvertLayerSupportedTests<armnn::NeonWorkloadFactory, armnn::ConvertFp16ToFp32Layer,
63  armnn::DataType::Float16, armnn::DataType::Float32>(reasonIfUnsupported);
64 
65  BOOST_CHECK(result);
66 }
67 
68 BOOST_AUTO_TEST_CASE(IsConvertFp32ToFp16SupportedNeon)
69 {
70  std::string reasonIfUnsupported;
71 
72  bool result = IsConvertLayerSupportedTests<armnn::NeonWorkloadFactory, armnn::ConvertFp32ToFp16Layer,
73  armnn::DataType::Float32, armnn::DataType::Float16>(reasonIfUnsupported);
74 
75  BOOST_CHECK(result);
76 }
77 
78 BOOST_AUTO_TEST_CASE(IsLogicalBinarySupportedNeon)
79 {
80  std::string reasonIfUnsupported;
81 
82  bool result = IsLogicalBinaryLayerSupportedTests<armnn::NeonWorkloadFactory,
83  armnn::DataType::Boolean, armnn::DataType::Boolean>(reasonIfUnsupported);
84 
85  BOOST_CHECK(result);
86 }
87 
88 BOOST_AUTO_TEST_CASE(IsLogicalBinaryBroadcastSupportedNeon)
89 {
90  std::string reasonIfUnsupported;
91 
92  bool result = IsLogicalBinaryLayerBroadcastSupportedTests<armnn::NeonWorkloadFactory,
93  armnn::DataType::Boolean, armnn::DataType::Boolean>(reasonIfUnsupported);
94 
95  BOOST_CHECK(result);
96 }
97 
98 BOOST_AUTO_TEST_CASE(IsMeanSupportedNeon)
99 {
100  std::string reasonIfUnsupported;
101 
102  bool result = IsMeanLayerSupportedTests<armnn::NeonWorkloadFactory,
103  armnn::DataType::Float32, armnn::DataType::Float32>(reasonIfUnsupported);
104 
105  BOOST_CHECK(result);
106 }
107 
108 BOOST_AUTO_TEST_CASE(IsConstantSupportedNeon)
109 {
110  std::string reasonIfUnsupported;
111 
112  bool result = IsConstantLayerSupportedTests<armnn::NeonWorkloadFactory,
113  armnn::DataType::Float16>(reasonIfUnsupported);
114  BOOST_CHECK(result);
115 
116  result = IsConstantLayerSupportedTests<armnn::NeonWorkloadFactory,
117  armnn::DataType::Float32>(reasonIfUnsupported);
118  BOOST_CHECK(result);
119 
120  result = IsConstantLayerSupportedTests<armnn::NeonWorkloadFactory,
121  armnn::DataType::QAsymmU8>(reasonIfUnsupported);
122  BOOST_CHECK(result);
123 
124  result = IsConstantLayerSupportedTests<armnn::NeonWorkloadFactory,
125  armnn::DataType::Boolean>(reasonIfUnsupported);
126  BOOST_CHECK(!result);
127 
128  result = IsConstantLayerSupportedTests<armnn::NeonWorkloadFactory,
129  armnn::DataType::QSymmS16>(reasonIfUnsupported);
130  BOOST_CHECK(result);
131 
132  result = IsConstantLayerSupportedTests<armnn::NeonWorkloadFactory,
133  armnn::DataType::QSymmS8>(reasonIfUnsupported);
134  BOOST_CHECK(result);
135 
136  result = IsConstantLayerSupportedTests<armnn::NeonWorkloadFactory,
137  armnn::DataType::QAsymmS8>(reasonIfUnsupported);
138  BOOST_CHECK(result);
139 
140  result = IsConstantLayerSupportedTests<armnn::NeonWorkloadFactory,
141  armnn::DataType::BFloat16>(reasonIfUnsupported);
142  BOOST_CHECK(result);
143 }
144 
145 BOOST_AUTO_TEST_SUITE_END()
BOOST_AUTO_TEST_SUITE
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
LayerTests.hpp
armnn::ConvertFp16ToFp32Layer
This layer converts data type Float 16 to Float 32.
Definition: ConvertFp16ToFp32Layer.hpp:14
TensorHelpers.hpp
NeonWorkloadFactoryHelper.hpp
armnn::ConvertFp32ToFp16Layer
This layer converts data type Float 32 to Float 16.
Definition: ConvertFp32ToFp16Layer.hpp:13
CpuTensorHandle.hpp
BOOST_AUTO_TEST_SUITE_END
BOOST_AUTO_TEST_SUITE_END()
IsLayerSupportedTestImpl.hpp
BOOST_AUTO_TEST_CASE
BOOST_AUTO_TEST_CASE(IsLayerSupportedFloat16Neon)
Definition: NeonLayerSupportTests.cpp:23
ConvertFp32ToFp16Layer.hpp
NeonWorkloadFactory.hpp
ConvertFp16ToFp32Layer.hpp