blob: d71ce1f2bfeea7e86ef0676e3bcd5f3eb1ec63c5 [file] [log] [blame]
Laurent Carlier749294b2020-06-01 09:03:17 +01001//
Teresa Charlin686f45c2020-07-23 17:59:00 +01002// Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
David Beckecb56cd2018-09-05 12:52:57 +01003// SPDX-License-Identifier: MIT
telsoa014fcda012018-03-09 14:13:49 +00004//
telsoa014fcda012018-03-09 14:13:49 +00005
Aron Virginas-Tar56055192018-11-12 18:10:43 +00006#include "NeonWorkloadFactoryHelper.hpp"
7
Colm Donelanc42a9872022-02-02 16:35:09 +00008#include <armnnTestUtils/TensorHelpers.hpp>
Sadik Armagana097d2a2021-11-24 15:47:28 +00009#include <UnitTests.hpp>
telsoa014fcda012018-03-09 14:13:49 +000010
Aron Virginas-Tarc9cc8042018-11-01 16:15:57 +000011#include <neon/NeonLayerSupport.hpp>
12#include <neon/NeonWorkloadFactory.hpp>
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +010013
Aron Virginas-Tarc9cc8042018-11-01 16:15:57 +000014#include <backendsCommon/test/ActivationFixture.hpp>
15#include <backendsCommon/test/LayerTests.hpp>
telsoa014fcda012018-03-09 14:13:49 +000016
Sadik Armagan1625efc2021-06-10 18:24:34 +010017#include <doctest/doctest.h>
telsoa014fcda012018-03-09 14:13:49 +000018
Sadik Armagan1625efc2021-06-10 18:24:34 +010019TEST_SUITE("Compute_ArmComputeNeon")
20{
Sadik Armagan581742d2019-08-12 14:11:37 +010021using namespace armnn;
22
23using FactoryType = NeonWorkloadFactory;
telsoa014fcda012018-03-09 14:13:49 +000024
25// ============================================================================
26// UNIT tests
27
Mike Kelly56858022020-01-27 12:14:47 +000028// BatchToSpace
Keith Davis33a626f2020-08-27 15:38:12 +010029ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcFloat321, BatchToSpaceNdNhwcTest1<DataType::Float32>)
30ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcFloat322, BatchToSpaceNdNhwcTest2<DataType::Float32>)
31ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcFloat323, BatchToSpaceNdNhwcTest3<DataType::Float32>)
Mike Kelly56858022020-01-27 12:14:47 +000032
Keith Davis33a626f2020-08-27 15:38:12 +010033ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwFloat321, BatchToSpaceNdNchwTest1<DataType::Float32>)
34ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwFloat322, BatchToSpaceNdNchwTest2<DataType::Float32>)
35ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwFloat323, BatchToSpaceNdNchwTest3<DataType::Float32>)
Mike Kelly56858022020-01-27 12:14:47 +000036
Keith Davis33a626f2020-08-27 15:38:12 +010037ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcInt1, BatchToSpaceNdNhwcTest1<DataType::QAsymmS8>)
38ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcInt2, BatchToSpaceNdNhwcTest2<DataType::QAsymmS8>)
39ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcInt3, BatchToSpaceNdNhwcTest3<DataType::QAsymmS8>)
Sadik Armagan303980c2020-04-17 12:45:14 +010040
Keith Davis33a626f2020-08-27 15:38:12 +010041ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwInt1, BatchToSpaceNdNchwTest1<DataType::QAsymmS8>)
42ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwInt2, BatchToSpaceNdNchwTest2<DataType::QAsymmS8>)
43ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwInt3, BatchToSpaceNdNchwTest3<DataType::QAsymmS8>)
Sadik Armagan303980c2020-04-17 12:45:14 +010044
Keith Davis33a626f2020-08-27 15:38:12 +010045ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcUint1, BatchToSpaceNdNhwcTest1<DataType::QAsymmU8>)
46ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcUint2, BatchToSpaceNdNhwcTest2<DataType::QAsymmU8>)
47ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcUint3, BatchToSpaceNdNhwcTest3<DataType::QAsymmU8>)
Mike Kelly56858022020-01-27 12:14:47 +000048
Keith Davis33a626f2020-08-27 15:38:12 +010049ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwUint1, BatchToSpaceNdNchwTest1<DataType::QAsymmU8>)
50ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwUint2, BatchToSpaceNdNchwTest2<DataType::QAsymmU8>)
51ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwUint3, BatchToSpaceNdNchwTest3<DataType::QAsymmU8>)
Mike Kelly56858022020-01-27 12:14:47 +000052
telsoa014fcda012018-03-09 14:13:49 +000053// Convolution
Keith Davisf500d6c2020-08-31 08:32:55 +010054ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution1d, Convolution1dTest, true)
telsoa014fcda012018-03-09 14:13:49 +000055
Keith Davisf500d6c2020-08-31 08:32:55 +010056ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution2d, SimpleConvolution2d3x5Test, true, DataLayout::NCHW)
57ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution2dNhwc, SimpleConvolution2d3x5Test, true, DataLayout::NHWC)
58ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution2d3x3Uint8, SimpleConvolution2d3x3Uint8Test, true, DataLayout::NCHW)
59ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution2d3x3Uint8Nhwc, SimpleConvolution2d3x3Uint8Test, true, DataLayout::NHWC)
60ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedConvolution2d, SimpleConvolution2d3x5Test, false, DataLayout::NCHW)
61ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedConvolution2dNhwc, SimpleConvolution2d3x5Test, false, DataLayout::NHWC)
62ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedConvolution2dStride2x2Nhwc,
Sadik Armagan581742d2019-08-12 14:11:37 +010063 SimpleConvolution2d3x3Stride2x2Test, false, DataLayout::NHWC)
jimfly010a088a62018-10-25 17:05:05 +010064
Keith Davisf500d6c2020-08-31 08:32:55 +010065ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedConvolution2dSquare, SimpleConvolution2d3x3Test, false, DataLayout::NCHW)
66ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution2dAsymmetricPadding,
67 Convolution2dAsymmetricPaddingTest,
68 DataLayout::NCHW)
narpra015f703182018-10-26 16:24:58 +010069
Keith Davisf500d6c2020-08-31 08:32:55 +010070ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedConvolution2dSquareNhwc, SimpleConvolution2d3x3Test, false, DataLayout::NHWC)
71ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution2dAsymmetricPaddingNhwc,
narpra015f703182018-10-26 16:24:58 +010072 Convolution2dAsymmetricPaddingTest,
Sadik Armagan581742d2019-08-12 14:11:37 +010073 DataLayout::NHWC)
telsoa014fcda012018-03-09 14:13:49 +000074
Keith Davisf500d6c2020-08-31 08:32:55 +010075ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution2dSquareNhwc, SimpleConvolution2d3x3NhwcTest, false)
telsoa014fcda012018-03-09 14:13:49 +000076
Keith Davisf500d6c2020-08-31 08:32:55 +010077ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d3x3Dilation3x3,
Sadik Armagan581742d2019-08-12 14:11:37 +010078 Convolution2d3x3Dilation3x3Test<DataType::Float32, DataType::Float32>,
Jan Eilers59c66702019-07-11 10:22:36 +010079 false,
Sadik Armagan581742d2019-08-12 14:11:37 +010080 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +010081ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d3x3Dilation3x3Nhwc,
Sadik Armagan581742d2019-08-12 14:11:37 +010082 Convolution2d3x3Dilation3x3Test<DataType::Float32, DataType::Float32>,
Jan Eilers59c66702019-07-11 10:22:36 +010083 false,
Sadik Armagan581742d2019-08-12 14:11:37 +010084 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +010085ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d3x3Dilation3x3Int8,
Sadik Armagan303980c2020-04-17 12:45:14 +010086 Convolution2d3x3Dilation3x3Test<DataType::QAsymmS8, DataType::Signed32>,
87 false,
88 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +010089ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d3x3Dilation3x3NhwcInt8,
Sadik Armagan303980c2020-04-17 12:45:14 +010090 Convolution2d3x3Dilation3x3Test<DataType::QAsymmS8, DataType::Signed32>,
91 false,
92 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +010093ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d3x3Dilation3x3Uint8,
Derek Lambertif90c56d2020-01-10 17:14:08 +000094 Convolution2d3x3Dilation3x3Test<DataType::QAsymmU8, DataType::Signed32>,
Jan Eilers59c66702019-07-11 10:22:36 +010095 false,
Sadik Armagan581742d2019-08-12 14:11:37 +010096 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +010097ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d3x3Dilation3x3NhwcUint8,
Derek Lambertif90c56d2020-01-10 17:14:08 +000098 Convolution2d3x3Dilation3x3Test<DataType::QAsymmU8, DataType::Signed32>,
Jan Eilers59c66702019-07-11 10:22:36 +010099 false,
Sadik Armagan581742d2019-08-12 14:11:37 +0100100 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100101ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x3x3Dilation3x3,
Sadik Armagan581742d2019-08-12 14:11:37 +0100102 Convolution2d2x3x3Dilation3x3Test<DataType::Float32, DataType::Float32>,
Jan Eilers59c66702019-07-11 10:22:36 +0100103 false,
Sadik Armagan581742d2019-08-12 14:11:37 +0100104 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100105ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x3x3Dilation3x3Nhwc,
Sadik Armagan581742d2019-08-12 14:11:37 +0100106 Convolution2d2x3x3Dilation3x3Test<DataType::Float32, DataType::Float32>,
Jan Eilers59c66702019-07-11 10:22:36 +0100107 false,
Sadik Armagan581742d2019-08-12 14:11:37 +0100108 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100109ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x3x3Dilation3x3Int8,
Sadik Armagan303980c2020-04-17 12:45:14 +0100110 Convolution2d2x3x3Dilation3x3Test<DataType::QAsymmS8, DataType::Signed32>,
111 false,
112 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100113ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x3x3Dilation3x3NhwcInt8,
Sadik Armagan303980c2020-04-17 12:45:14 +0100114 Convolution2d2x3x3Dilation3x3Test<DataType::QAsymmS8, DataType::Signed32>,
115 false,
116 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100117ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x3x3Dilation3x3Uint8,
Derek Lambertif90c56d2020-01-10 17:14:08 +0000118 Convolution2d2x3x3Dilation3x3Test<DataType::QAsymmU8, DataType::Signed32>,
Jan Eilers59c66702019-07-11 10:22:36 +0100119 false,
Sadik Armagan581742d2019-08-12 14:11:37 +0100120 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100121ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x3x3Dilation3x3NhwcUint8,
Derek Lambertif90c56d2020-01-10 17:14:08 +0000122 Convolution2d2x3x3Dilation3x3Test<DataType::QAsymmU8, DataType::Signed32>,
Jan Eilers59c66702019-07-11 10:22:36 +0100123 false,
Sadik Armagan581742d2019-08-12 14:11:37 +0100124 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100125ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x2Dilation2x2Padding2x2Stride3x3,
Jan Eilers59c66702019-07-11 10:22:36 +0100126 Convolution2d2x2Dilation2x2Padding2x2Stride3x3Test
Sadik Armagan581742d2019-08-12 14:11:37 +0100127 <DataType::Float32, DataType::Float32>,
Jan Eilers59c66702019-07-11 10:22:36 +0100128 false,
Sadik Armagan581742d2019-08-12 14:11:37 +0100129 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100130ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x2Dilation2x2Padding2x2Stride3x3Nhwc,
Jan Eilers59c66702019-07-11 10:22:36 +0100131 Convolution2d2x2Dilation2x2Padding2x2Stride3x3Test
Sadik Armagan581742d2019-08-12 14:11:37 +0100132 <DataType::Float32, DataType::Float32>,
Jan Eilers59c66702019-07-11 10:22:36 +0100133 false,
Sadik Armagan581742d2019-08-12 14:11:37 +0100134 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100135ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x2Dilation2x2Padding2x2Stride3x3Int8,
Sadik Armagan303980c2020-04-17 12:45:14 +0100136 Convolution2d2x2Dilation2x2Padding2x2Stride3x3Test
137 <DataType::QAsymmS8, DataType::Signed32>,
138 false,
139 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100140ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x2Dilation2x2Padding2x2Stride3x3NhwcInt8,
Sadik Armagan303980c2020-04-17 12:45:14 +0100141 Convolution2d2x2Dilation2x2Padding2x2Stride3x3Test
142 <DataType::QAsymmS8, DataType::Signed32>,
143 false,
144 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100145ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x2Dilation2x2Padding2x2Stride3x3Uint8,
Jan Eilers59c66702019-07-11 10:22:36 +0100146 Convolution2d2x2Dilation2x2Padding2x2Stride3x3Test
Derek Lambertif90c56d2020-01-10 17:14:08 +0000147 <DataType::QAsymmU8, DataType::Signed32>,
Jan Eilers59c66702019-07-11 10:22:36 +0100148 false,
Sadik Armagan581742d2019-08-12 14:11:37 +0100149 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100150ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x2Dilation2x2Padding2x2Stride3x3NhwcUint8,
Jan Eilers59c66702019-07-11 10:22:36 +0100151 Convolution2d2x2Dilation2x2Padding2x2Stride3x3Test
Derek Lambertif90c56d2020-01-10 17:14:08 +0000152 <DataType::QAsymmU8, DataType::Signed32>,
Jan Eilers59c66702019-07-11 10:22:36 +0100153 false,
Sadik Armagan581742d2019-08-12 14:11:37 +0100154 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100155ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dMult4,
Kevin May665a964a2019-08-21 16:53:50 +0100156 DepthwiseConvolution2dMult4Test<armnn::DataType::Float32, armnn::DataType::Float32>,
157 false,
158 armnn::DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100159ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dMult2,
Kevin May665a964a2019-08-21 16:53:50 +0100160 DepthwiseConvolution2dMult2Test<armnn::DataType::Float32, armnn::DataType::Float32>,
161 false,
162 armnn::DataLayout::NCHW)
Jan Eilers59c66702019-07-11 10:22:36 +0100163
Keith Davisf500d6c2020-08-31 08:32:55 +0100164ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2dPerAxisQuantTestNchw, Convolution2dPerAxisQuantTest, DataLayout::NCHW);
165ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2dPerAxisQuantTestNhwc, Convolution2dPerAxisQuantTest, DataLayout::NHWC);
Aron Virginas-Tar21fc28b2019-11-26 14:04:54 +0000166
Teresa Charlinec5f7d12021-10-22 17:15:00 +0100167// Convolution 3d - NDHWC
168ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution3d3x3x3Float32,
169 SimpleConvolution3d3x3x3Float32Test,
170 false,
171 DataLayout::NDHWC)
172ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution3d3x3x3Int8,
173 SimpleConvolution3d3x3x3Int8Test,
174 false,
175 DataLayout::NDHWC)
176ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution3d3x3x3Uint8,
177 SimpleConvolution3d3x3x3Uint8Test,
178 false,
179 DataLayout::NDHWC)
180
181ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution3d2x2x2Strides3x5x5Float32,
182 Convolution3d2x2x2Strides3x5x5Float32Test,
183 false,
184 DataLayout::NDHWC)
185ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution3d2x2x2Strides3x5x5TestInt8,
186 Convolution3d2x2x2Strides3x5x5Int8Test,
187 true,
188 DataLayout::NDHWC)
189ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution3d2x2x2Strides3x5x5TestUint8,
190 Convolution3d2x2x2Strides3x5x5Uint8Test,
191 false,
192 DataLayout::NDHWC)
193
194ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution3dPaddingSame3x3x3Float32,
195 Convolution3dPaddingSame3x3x3Float32Test,
196 false,
197 DataLayout::NDHWC)
198ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution3dPaddingSame3x3x3TestInt8,
199 Convolution3dPaddingSame3x3x3Int8Test,
200 false,
201 DataLayout::NDHWC)
202ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution3dPaddingSame3x3x3TestUint8,
203 Convolution3dPaddingSame3x3x3Uint8Test,
204 false,
205 DataLayout::NDHWC)
206
207ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution3d2x2x2Stride3x3x3SmallTestFloat32,
208 Convolution3d2x2x2Stride3x3x3SmallFloat32Test,
209 false,
210 DataLayout::NDHWC)
211#ifdef __ARM_FEATURE_FP16_VECTOR_ARITHMETIC
212ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution3d2x3x3TestFloat16,
213 Convolution3d2x3x3Float16Test,
214 true,
215 DataLayout::NDHWC)
216ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution3d2x2x2SmallTestFloat16,
217 Convolution3d2x2x2SmallFloat16Test,
218 false,
219 DataLayout::NDHWC)
220#endif
221
Aron Virginas-Tar2f00b742019-09-30 13:28:08 +0100222// DepthToSpace
223ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwFloat32_1, DepthToSpaceTest1<DataType::Float32>, DataLayout::NCHW);
224ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwFloat32_2, DepthToSpaceTest2<DataType::Float32>, DataLayout::NCHW);
225ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwFloat32_3, DepthToSpaceTest3<DataType::Float32>, DataLayout::NCHW);
226ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwFloat32_4, DepthToSpaceTest4<DataType::Float32>, DataLayout::NCHW);
227
Aron Virginas-Tar99836d32019-09-30 16:34:31 +0100228ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwFloat16_1, DepthToSpaceTest1<DataType::Float16>, DataLayout::NCHW);
229ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwFloat16_2, DepthToSpaceTest2<DataType::Float16>, DataLayout::NCHW);
230ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwFloat16_3, DepthToSpaceTest3<DataType::Float16>, DataLayout::NCHW);
231ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwFloat16_4, DepthToSpaceTest4<DataType::Float16>, DataLayout::NCHW);
232
Sadik Armagan303980c2020-04-17 12:45:14 +0100233ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwInt8_1, DepthToSpaceTest1<DataType::QAsymmS8>, DataLayout::NCHW);
234ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwInt8_2, DepthToSpaceTest2<DataType::QAsymmS8>, DataLayout::NCHW);
235ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwInt8_3, DepthToSpaceTest3<DataType::QAsymmS8>, DataLayout::NCHW);
236ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwInt8_4, DepthToSpaceTest4<DataType::QAsymmS8>, DataLayout::NCHW);
237
Derek Lambertif90c56d2020-01-10 17:14:08 +0000238ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwUint8_1, DepthToSpaceTest1<DataType::QAsymmU8>, DataLayout::NCHW);
239ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwUint8_2, DepthToSpaceTest2<DataType::QAsymmU8>, DataLayout::NCHW);
240ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwUint8_3, DepthToSpaceTest3<DataType::QAsymmU8>, DataLayout::NCHW);
241ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwUint8_4, DepthToSpaceTest4<DataType::QAsymmU8>, DataLayout::NCHW);
Aron Virginas-Tar2f00b742019-09-30 13:28:08 +0100242
Derek Lambertif90c56d2020-01-10 17:14:08 +0000243ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwInt16_1, DepthToSpaceTest1<DataType::QSymmS16>, DataLayout::NCHW);
244ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwInt16_2, DepthToSpaceTest2<DataType::QSymmS16>, DataLayout::NCHW);
245ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwInt16_3, DepthToSpaceTest3<DataType::QSymmS16>, DataLayout::NCHW);
246ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwInt16_4, DepthToSpaceTest4<DataType::QSymmS16>, DataLayout::NCHW);
Aron Virginas-Tar2f00b742019-09-30 13:28:08 +0100247
248ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcFloat32_1, DepthToSpaceTest1<DataType::Float32>, DataLayout::NHWC);
249ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcFloat32_2, DepthToSpaceTest2<DataType::Float32>, DataLayout::NHWC);
250ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcFloat32_3, DepthToSpaceTest3<DataType::Float32>, DataLayout::NHWC);
251ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcFloat32_4, DepthToSpaceTest4<DataType::Float32>, DataLayout::NHWC);
252
Aron Virginas-Tar99836d32019-09-30 16:34:31 +0100253ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcFloat16_1, DepthToSpaceTest1<DataType::Float16>, DataLayout::NHWC);
254ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcFloat16_2, DepthToSpaceTest2<DataType::Float16>, DataLayout::NHWC);
255ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcFloat16_3, DepthToSpaceTest3<DataType::Float16>, DataLayout::NHWC);
256ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcFloat16_4, DepthToSpaceTest4<DataType::Float16>, DataLayout::NHWC);
257
Sadik Armagan303980c2020-04-17 12:45:14 +0100258ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcInt8_1, DepthToSpaceTest1<DataType::QAsymmS8>, DataLayout::NHWC);
259ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcInt8_2, DepthToSpaceTest2<DataType::QAsymmS8>, DataLayout::NHWC);
260ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcInt8_3, DepthToSpaceTest3<DataType::QAsymmS8>, DataLayout::NHWC);
261ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcInt8_4, DepthToSpaceTest4<DataType::QAsymmS8>, DataLayout::NHWC);
262
Derek Lambertif90c56d2020-01-10 17:14:08 +0000263ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcUint8_1, DepthToSpaceTest1<DataType::QAsymmU8>, DataLayout::NHWC);
264ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcUint8_2, DepthToSpaceTest2<DataType::QAsymmU8>, DataLayout::NHWC);
265ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcUint8_3, DepthToSpaceTest3<DataType::QAsymmU8>, DataLayout::NHWC);
266ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcUint8_4, DepthToSpaceTest4<DataType::QAsymmU8>, DataLayout::NHWC);
Aron Virginas-Tar2f00b742019-09-30 13:28:08 +0100267
Derek Lambertif90c56d2020-01-10 17:14:08 +0000268ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcInt16_1, DepthToSpaceTest1<DataType::QSymmS16>, DataLayout::NHWC);
269ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcInt16_2, DepthToSpaceTest2<DataType::QSymmS16>, DataLayout::NHWC);
270ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcInt16_3, DepthToSpaceTest3<DataType::QSymmS16>, DataLayout::NHWC);
271ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcInt16_4, DepthToSpaceTest4<DataType::QSymmS16>, DataLayout::NHWC);
Aron Virginas-Tar2f00b742019-09-30 13:28:08 +0100272
telsoa014fcda012018-03-09 14:13:49 +0000273// Depthwise Convolution
Jan Eilers53ef7952021-06-02 12:01:25 +0100274ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2d, DepthwiseConvolution2dTest, true, DataLayout::NCHW)
275ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dUint8, DepthwiseConvolution2dUint8Test, true, DataLayout::NCHW)
276
277ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedDepthwiseConvolution2d, DepthwiseConvolution2dTest, false, DataLayout::NCHW)
278
Keith Davisf500d6c2020-08-31 08:32:55 +0100279ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dDepthMul1,
Sadik Armagan581742d2019-08-12 14:11:37 +0100280 DepthwiseConvolution2dDepthMul1Test, true, DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100281ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedDepthwiseConvolution2dDepthMul1,
Sadik Armagan581742d2019-08-12 14:11:37 +0100282 DepthwiseConvolution2dDepthMul1Test, false, DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100283ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dDepthMul1Uint8,
Sadik Armagan581742d2019-08-12 14:11:37 +0100284 DepthwiseConvolution2dDepthMul1Uint8Test, true, DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100285ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedDepthwiseConvolution2dDepthMul1Uint8,
Sadik Armagan581742d2019-08-12 14:11:37 +0100286 DepthwiseConvolution2dDepthMul1Uint8Test, false, DataLayout::NCHW)
jimfly01b9c89632018-10-26 16:50:13 +0100287
288// NHWC Depthwise Convolution
Keith Davisf500d6c2020-08-31 08:32:55 +0100289ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dDepthMul1NHhwc,
Sadik Armagan581742d2019-08-12 14:11:37 +0100290 DepthwiseConvolution2dDepthMul1Test, true, DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100291ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedDepthwiseConvolution2dDepthMul1Nhwc,
Sadik Armagan581742d2019-08-12 14:11:37 +0100292 DepthwiseConvolution2dDepthMul1Test, false, DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100293ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dDepthMul1Uint8Nhwc,
Sadik Armagan581742d2019-08-12 14:11:37 +0100294 DepthwiseConvolution2dDepthMul1Uint8Test, true, DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100295ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedDepthwiseConvolution2dDepthMul1Uint8Nhwc,
Sadik Armagan581742d2019-08-12 14:11:37 +0100296 DepthwiseConvolution2dDepthMul1Uint8Test, false, DataLayout::NHWC)
jimfly01b9c89632018-10-26 16:50:13 +0100297
Keith Davisf500d6c2020-08-31 08:32:55 +0100298ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dDepthNhwc, DepthwiseConvolution2dDepthNhwcTest, false)
299ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleDepthwiseConvolution2d3x3Dilation3x3Nhwc,
Bruno Goncalves22972f02019-04-26 21:03:24 -0300300 SimpleDepthwiseConvolution2d3x3Dilation3x3NhwcTest)
jimfly01b9c89632018-10-26 16:50:13 +0100301
telsoa014fcda012018-03-09 14:13:49 +0000302
Keith Davisf500d6c2020-08-31 08:32:55 +0100303ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dAsymmetric,
Sadik Armagan581742d2019-08-12 14:11:37 +0100304 DepthwiseConvolution2dAsymmetricTest, true, DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100305ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedDepthwiseConvolution2dAsymmetric,
Sadik Armagan581742d2019-08-12 14:11:37 +0100306 DepthwiseConvolution2dAsymmetricTest, false, DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100307ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dAsymmetricNhwc,
Sadik Armagan581742d2019-08-12 14:11:37 +0100308 DepthwiseConvolution2dAsymmetricTest, true, DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100309ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedDepthwiseConvolution2dAsymmetricNhwc,
Sadik Armagan581742d2019-08-12 14:11:37 +0100310 DepthwiseConvolution2dAsymmetricTest, false, DataLayout::NHWC)
surmeh013537c2c2018-05-18 16:31:43 +0100311
Keith Davisf500d6c2020-08-31 08:32:55 +0100312ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dDepthMul64, DepthwiseConvolution2dDepthMul64Test);
Matthew Jackson99107922019-07-25 12:58:56 +0100313
Keith Davisf500d6c2020-08-31 08:32:55 +0100314ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dPerAxisQuantTestNchw, DepthwiseConvolution2dPerAxisQuantTest,
Aron Virginas-Tar24457442019-11-27 13:18:25 +0000315 DataLayout::NCHW);
Keith Davisf500d6c2020-08-31 08:32:55 +0100316ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dPerAxisQuantTestNhwc, DepthwiseConvolution2dPerAxisQuantTest,
Aron Virginas-Tar24457442019-11-27 13:18:25 +0000317 DataLayout::NHWC);
318
telsoa014fcda012018-03-09 14:13:49 +0000319namespace
320{
321
Sadik Armagan581742d2019-08-12 14:11:37 +0100322DepthwiseConvolution2dDescriptor MakeDepthwiseConv2dDesc(uint32_t strideX, uint32_t strideY,
telsoa014fcda012018-03-09 14:13:49 +0000323 uint32_t depthMultiplier = 1, uint32_t padLeft = 0, uint32_t padRight = 0,
324 uint32_t padTop = 0, uint32_t padBottom = 0)
325{
Jan Eilers8eb25602020-03-09 12:13:48 +0000326 IgnoreUnused(depthMultiplier);
telsoa01c577f2c2018-08-31 09:22:23 +0100327
Sadik Armagan581742d2019-08-12 14:11:37 +0100328 DepthwiseConvolution2dDescriptor desc;
telsoa01c577f2c2018-08-31 09:22:23 +0100329
telsoa014fcda012018-03-09 14:13:49 +0000330 desc.m_PadLeft = padLeft;
331 desc.m_PadRight = padRight;
telsoa01c577f2c2018-08-31 09:22:23 +0100332
telsoa014fcda012018-03-09 14:13:49 +0000333 desc.m_PadTop = padTop;
334 desc.m_PadBottom = padBottom;
335 desc.m_StrideX = strideX;
336 desc.m_StrideY = strideY;
telsoa01c577f2c2018-08-31 09:22:23 +0100337 desc.m_BiasEnabled = false;
338
telsoa014fcda012018-03-09 14:13:49 +0000339 return desc;
340}
341
Sadik Armagan581742d2019-08-12 14:11:37 +0100342TensorInfo CreateOutputTensorInfo(const TensorInfo& inputInfo,
343 const TensorInfo& weightsInfo,
344 const DepthwiseConvolution2dDescriptor& descriptor,
345 DataType dataType)
telsoa01c577f2c2018-08-31 09:22:23 +0100346{
Sadik Armagan581742d2019-08-12 14:11:37 +0100347 const TensorShape& inputShape = inputInfo.GetShape();
348 const TensorShape& filterShape = weightsInfo.GetShape();
telsoa01c577f2c2018-08-31 09:22:23 +0100349
350 unsigned int inWidth = inputShape[3];
351 unsigned int inHeight = inputShape[2];
352 unsigned int inBatchSize = inputShape[0];
353
Jan Eilers53ef7952021-06-02 12:01:25 +0100354 unsigned int filterWidth = filterShape[2];
telsoa01c577f2c2018-08-31 09:22:23 +0100355 unsigned int readWidth = (inWidth + descriptor.m_PadLeft + descriptor.m_PadRight) - (filterWidth);
356 unsigned int outWidth = 1u + (readWidth / descriptor.m_StrideX);
357
Jan Eilers53ef7952021-06-02 12:01:25 +0100358 unsigned int filterHeight = filterShape[1];
telsoa01c577f2c2018-08-31 09:22:23 +0100359 unsigned int readHeight = (inHeight + descriptor.m_PadTop + descriptor.m_PadBottom) - (filterHeight);
360 unsigned int outHeight = 1u + (readHeight / descriptor.m_StrideY);
telsoa01c577f2c2018-08-31 09:22:23 +0100361
Jan Eilers53ef7952021-06-02 12:01:25 +0100362 unsigned int outChannels = filterShape[3];
telsoa01c577f2c2018-08-31 09:22:23 +0100363 unsigned int outBatchSize = inBatchSize;
364
Sadik Armagan581742d2019-08-12 14:11:37 +0100365 TensorShape outputShape({outBatchSize, outChannels, outHeight, outWidth});
366 return TensorInfo(outputShape, dataType);
telsoa01c577f2c2018-08-31 09:22:23 +0100367}
telsoa014fcda012018-03-09 14:13:49 +0000368}
369
Sadik Armagan1625efc2021-06-10 18:24:34 +0100370TEST_CASE("DepthwiseConv2dUtils")
telsoa014fcda012018-03-09 14:13:49 +0000371{
Sadik Armagan581742d2019-08-12 14:11:37 +0100372 const DataType dataType = DataType::Float32;
telsoa01c577f2c2018-08-31 09:22:23 +0100373
Sadik Armagan581742d2019-08-12 14:11:37 +0100374 TensorInfo inputInfo({1, 1, 10, 10 }, dataType);
375 TensorInfo outputInfo;
Jan Eilers53ef7952021-06-02 12:01:25 +0100376 TensorInfo weightsInfo3x3({ 1, 3, 3, 1 }, dataType); // [1,H,W,I*M]
Sadik Armagan581742d2019-08-12 14:11:37 +0100377 TensorInfo biasesInfo;
telsoa01c577f2c2018-08-31 09:22:23 +0100378
Sadik Armagan581742d2019-08-12 14:11:37 +0100379 DepthwiseConvolution2dDescriptor descriptor;
380 NeonLayerSupport layerSupport;
telsoa014fcda012018-03-09 14:13:49 +0000381
382 // Strides supported: 1,2,3
telsoa01c577f2c2018-08-31 09:22:23 +0100383 descriptor = MakeDepthwiseConv2dDesc(1, 1);
384 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo3x3, descriptor, dataType);
Sadik Armagan1625efc2021-06-10 18:24:34 +0100385 CHECK(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100386 weightsInfo3x3, biasesInfo));
telsoa014fcda012018-03-09 14:13:49 +0000387
telsoa01c577f2c2018-08-31 09:22:23 +0100388 descriptor = MakeDepthwiseConv2dDesc(1, 2);
389 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo3x3, descriptor, dataType);
Sadik Armagan1625efc2021-06-10 18:24:34 +0100390 CHECK(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100391 weightsInfo3x3, biasesInfo));
telsoa01c577f2c2018-08-31 09:22:23 +0100392
393 descriptor = MakeDepthwiseConv2dDesc(1, 3);
394 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo3x3, descriptor, dataType);
Sadik Armagan1625efc2021-06-10 18:24:34 +0100395 CHECK(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100396 weightsInfo3x3, biasesInfo));
telsoa01c577f2c2018-08-31 09:22:23 +0100397
398 descriptor = MakeDepthwiseConv2dDesc(2, 1);
399 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo3x3, descriptor, dataType);
Sadik Armagan1625efc2021-06-10 18:24:34 +0100400 CHECK(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100401 weightsInfo3x3, biasesInfo));
telsoa01c577f2c2018-08-31 09:22:23 +0100402
403 descriptor = MakeDepthwiseConv2dDesc(2, 2);
404 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo3x3, descriptor, dataType);
Sadik Armagan1625efc2021-06-10 18:24:34 +0100405 CHECK(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100406 weightsInfo3x3, biasesInfo));
telsoa01c577f2c2018-08-31 09:22:23 +0100407
408 descriptor = MakeDepthwiseConv2dDesc(2, 3);
409 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo3x3, descriptor, dataType);
Sadik Armagan1625efc2021-06-10 18:24:34 +0100410 CHECK(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100411 weightsInfo3x3, biasesInfo));
telsoa01c577f2c2018-08-31 09:22:23 +0100412
413 descriptor = MakeDepthwiseConv2dDesc(3, 1);
414 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo3x3, descriptor, dataType);
Sadik Armagan1625efc2021-06-10 18:24:34 +0100415 CHECK(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100416 weightsInfo3x3, biasesInfo));
telsoa01c577f2c2018-08-31 09:22:23 +0100417
418 descriptor = MakeDepthwiseConv2dDesc(3, 2);
419 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo3x3, descriptor, dataType);
Sadik Armagan1625efc2021-06-10 18:24:34 +0100420 CHECK(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100421 weightsInfo3x3, biasesInfo));
telsoa01c577f2c2018-08-31 09:22:23 +0100422
423 descriptor = MakeDepthwiseConv2dDesc(3, 3);
424 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo3x3, descriptor, dataType);
Sadik Armagan1625efc2021-06-10 18:24:34 +0100425 CHECK(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100426 weightsInfo3x3, biasesInfo));
telsoa01c577f2c2018-08-31 09:22:23 +0100427
428 // Supported stride 4
429 descriptor = MakeDepthwiseConv2dDesc(4, 1);
430 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo3x3, descriptor, dataType);
Sadik Armagan1625efc2021-06-10 18:24:34 +0100431 CHECK(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100432 weightsInfo3x3, biasesInfo));
telsoa014fcda012018-03-09 14:13:49 +0000433
434 // Supported weights shape 1x1
Sadik Armagan581742d2019-08-12 14:11:37 +0100435 TensorInfo weightsInfo1x1({ 1, 1, 1, 1 }, DataType::Float32);
telsoa01c577f2c2018-08-31 09:22:23 +0100436 descriptor = MakeDepthwiseConv2dDesc(1, 1);
437 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo1x1, descriptor, dataType);
Sadik Armagan1625efc2021-06-10 18:24:34 +0100438 CHECK(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100439 weightsInfo1x1, biasesInfo));
telsoa014fcda012018-03-09 14:13:49 +0000440
441 // Supported shape 2x2
Jan Eilers53ef7952021-06-02 12:01:25 +0100442 TensorInfo weightsInfo2x2({ 1, 2, 2, 1 }, DataType::Float32);
telsoa01c577f2c2018-08-31 09:22:23 +0100443 descriptor = MakeDepthwiseConv2dDesc(1, 1);
444 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo2x2, descriptor, dataType);
Sadik Armagan1625efc2021-06-10 18:24:34 +0100445 CHECK(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100446 weightsInfo2x2, biasesInfo));
surmeh013537c2c2018-05-18 16:31:43 +0100447
448 // Asymmetric padding
telsoa01c577f2c2018-08-31 09:22:23 +0100449 descriptor = MakeDepthwiseConv2dDesc(1, 1, 1, 1, 2, 1, 2);
450 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo3x3, descriptor, dataType);
Sadik Armagan1625efc2021-06-10 18:24:34 +0100451 CHECK(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100452 weightsInfo3x3, biasesInfo));
telsoa014fcda012018-03-09 14:13:49 +0000453}
454
Narumol Prangnawarat01961a72019-05-30 16:47:12 +0100455// Dequantize
Jan Eilersf7107932019-11-01 11:09:36 +0000456// Fp16 is only supported if __ARM_FEATURE_FP16_VECTOR_ARITHMETIC is enabled
Narumol Prangnawarat01961a72019-05-30 16:47:12 +0100457ARMNN_AUTO_TEST_CASE(DequantizeSimpleUint8, DequantizeSimpleUint8Test)
458ARMNN_AUTO_TEST_CASE(DequantizeOffsetUint8, DequantizeOffsetUint8Test)
Jan Eilersf7107932019-11-01 11:09:36 +0000459ARMNN_AUTO_TEST_CASE(DequantizeSimpleInt16, DequantizeSimpleInt16Test)
Narumol Prangnawarat01961a72019-05-30 16:47:12 +0100460
telsoa014fcda012018-03-09 14:13:49 +0000461// Pooling
Finn Williams826a5432020-08-27 16:15:20 +0100462ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling2dSize3x3Stride2x4,
463 SimpleMaxPooling2dSize3x3Stride2x4Test, true)
464ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling2dSize3x3Stride2x4Uint8,
465 SimpleMaxPooling2dSize3x3Stride2x4Uint8Test, true)
James Conroy45a9b772018-10-31 11:47:53 +0000466
Finn Williams826a5432020-08-27 16:15:20 +0100467ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling2d, SimpleMaxPooling2dTest, DataLayout::NCHW)
468ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling2dNhwc, SimpleMaxPooling2dTest, DataLayout::NHWC)
469ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling2dUint8, SimpleMaxPooling2dUint8Test, DataLayout::NCHW)
470ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling2dUint8Nhwc, SimpleMaxPooling2dUint8Test, DataLayout::NHWC)
James Conroy45a9b772018-10-31 11:47:53 +0000471
Finn Williams826a5432020-08-27 16:15:20 +0100472ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleAveragePooling2d, SimpleAveragePooling2dTest, DataLayout::NCHW)
473ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleAveragePooling2dNhwc, SimpleAveragePooling2dTest, DataLayout::NCHW)
474ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleAveragePooling2dUint8,
475 SimpleAveragePooling2dUint8Test, DataLayout::NCHW)
476ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleAveragePooling2dUint8Nhwc,
477 SimpleAveragePooling2dUint8Test, DataLayout::NHWC)
surmeh01bceff2f2018-03-29 16:29:27 +0100478
Finn Williams826a5432020-08-27 16:15:20 +0100479ARMNN_AUTO_TEST_CASE_WITH_THF(LargeTensorsAveragePooling2d, LargeTensorsAveragePooling2dTest)
480ARMNN_AUTO_TEST_CASE_WITH_THF(LargeTensorsAveragePooling2dUint8, LargeTensorsAveragePooling2dUint8Test)
telsoa014fcda012018-03-09 14:13:49 +0000481
Finn Williams826a5432020-08-27 16:15:20 +0100482ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleL2Pooling2d, SimpleL2Pooling2dTest, DataLayout::NCHW)
483ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleL2Pooling2dNeon, SimpleL2Pooling2dTest, DataLayout::NHWC)
484ARMNN_AUTO_TEST_CASE_WITH_THF(UNSUPPORTED_SimpleL2Pooling2dUint8, SimpleL2Pooling2dUint8Test, DataLayout::NCHW)
James Conroy45a9b772018-10-31 11:47:53 +0000485
Finn Williams826a5432020-08-27 16:15:20 +0100486ARMNN_AUTO_TEST_CASE_WITH_THF(L2Pooling2dSize3Stride1, L2Pooling2dSize3Stride1Test)
487ARMNN_AUTO_TEST_CASE_WITH_THF(UNSUPPORTED_L2Pooling2dSize3Stride1Uint8, L2Pooling2dSize3Stride1Uint8Test)
488ARMNN_AUTO_TEST_CASE_WITH_THF(L2Pooling2dSize3Stride3, L2Pooling2dSize3Stride3Test)
489ARMNN_AUTO_TEST_CASE_WITH_THF(UNSUPPORTED_L2Pooling2dSize3Stride3Uint8, L2Pooling2dSize3Stride3Uint8Test)
490ARMNN_AUTO_TEST_CASE_WITH_THF(L2Pooling2dSize3Stride4, L2Pooling2dSize3Stride4Test)
491ARMNN_AUTO_TEST_CASE_WITH_THF(UNSUPPORTED_L2Pooling2dSize3Stride4Uint8, L2Pooling2dSize3Stride4Uint8Test)
492ARMNN_AUTO_TEST_CASE_WITH_THF(L2Pooling2dSize7, L2Pooling2dSize7Test)
493ARMNN_AUTO_TEST_CASE_WITH_THF(UNSUPPORTED_L2Pooling2dSize7Uint8, L2Pooling2dSize7Uint8Test)
494ARMNN_AUTO_TEST_CASE_WITH_THF(L2Pooling2dSize9, L2Pooling2dSize9Test)
495ARMNN_AUTO_TEST_CASE_WITH_THF(UNSUPPORTED_L2Pooling2dSize9Uint8, L2Pooling2dSize9Uint8Test)
telsoa014fcda012018-03-09 14:13:49 +0000496
497// Ignore padding values for pooling but count padding fields into the divisor
Finn Williams826a5432020-08-27 16:15:20 +0100498ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleMaxPooling2d, IgnorePaddingSimpleMaxPooling2dTest)
499ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleMaxPooling2dUint8, IgnorePaddingSimpleMaxPooling2dUint8Test)
500ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingMaxPooling2dSize3, IgnorePaddingMaxPooling2dSize3Test)
501ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingMaxPooling2dSize3Uint8, IgnorePaddingMaxPooling2dSize3Uint8Test)
telsoa014fcda012018-03-09 14:13:49 +0000502
Finn Williams826a5432020-08-27 16:15:20 +0100503ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleAveragePooling2d, IgnorePaddingSimpleAveragePooling2dTest)
504ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleAveragePooling2dUint8, IgnorePaddingSimpleAveragePooling2dUint8Test)
505ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleAveragePooling2dNoPadding,
506 IgnorePaddingSimpleAveragePooling2dNoPaddingTest)
507ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleAveragePooling2dNoPaddingUint8,
508 IgnorePaddingSimpleAveragePooling2dNoPaddingUint8Test)
509ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingAveragePooling2dSize3, IgnorePaddingAveragePooling2dSize3Test)
510ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingAveragePooling2dSize3Uint8,
511 IgnorePaddingAveragePooling2dSize3Uint8Test)
512ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingAveragePooling2dSize3x2Stride2x2,
513 IgnorePaddingAveragePooling2dSize3x2Stride2x2Test, false)
514ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingAveragePooling2dSize3x2Stride2x2NoPadding,
515 IgnorePaddingAveragePooling2dSize3x2Stride2x2Test,
516 true)
telsoa014fcda012018-03-09 14:13:49 +0000517
Finn Williams826a5432020-08-27 16:15:20 +0100518ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleL2Pooling2d, IgnorePaddingSimpleL2Pooling2dTest)
519ARMNN_AUTO_TEST_CASE_WITH_THF(UNSUPPORTED_IgnorePaddingSimpleL2Pooling2dUint8,
520 IgnorePaddingSimpleL2Pooling2dUint8Test)
521ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingL2Pooling2dSize3, IgnorePaddingL2Pooling2dSize3Test)
522ARMNN_AUTO_TEST_CASE_WITH_THF(UNSUPPORTED_IgnorePaddingL2Pooling2dSize3Uint8,
523 IgnorePaddingL2Pooling2dSize3Uint8Test)
telsoa014fcda012018-03-09 14:13:49 +0000524
525// Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100526ARMNN_AUTO_TEST_CASE_WITH_THF(ConstantLinearActivation, ConstantLinearActivationTest)
telsoa014fcda012018-03-09 14:13:49 +0000527
Teresa Charline32c8442020-03-05 14:41:50 +0000528// Sigmoid Activation / Logistic
Keith Davis33a626f2020-08-27 15:38:12 +0100529ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSigmoid, SimpleSigmoidTest)
530ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSigmoidUint8, SimpleSigmoidUint8Test)
Teresa Charline32c8442020-03-05 14:41:50 +0000531
532// BoundedReLU Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100533ARMNN_AUTO_TEST_CASE_WITH_THF(ReLu1, BoundedReLuUpperAndLowerBoundTest)
534ARMNN_AUTO_TEST_CASE_WITH_THF(ReLu6, BoundedReLuUpperBoundOnlyTest)
535ARMNN_AUTO_TEST_CASE_WITH_THF(ReLu1Uint8, BoundedReLuUint8UpperAndLowerBoundTest)
536ARMNN_AUTO_TEST_CASE_WITH_THF(ReLu6Uint8, BoundedReLuUint8UpperBoundOnlyTest)
Narumol Prangnawarat65d30962019-03-14 11:55:03 +0000537
Teresa Charline32c8442020-03-05 14:41:50 +0000538// ReLU Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100539ARMNN_AUTO_TEST_CASE_WITH_THF(ReLu, ReLuTest)
540ARMNN_AUTO_TEST_CASE_WITH_THF(ReLuUint8, ReLuUint8Test)
David Monahan3b3c3812020-02-25 09:03:29 +0000541
Teresa Charline32c8442020-03-05 14:41:50 +0000542// SoftReLU Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100543ARMNN_AUTO_TEST_CASE_WITH_THF(SoftReLu, SoftReLuTest)
Teresa Charline32c8442020-03-05 14:41:50 +0000544
545// LeakyReLU Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100546ARMNN_AUTO_TEST_CASE_WITH_THF(LeakyReLu, LeakyReLuTest)
Teresa Charline32c8442020-03-05 14:41:50 +0000547
548// Abs Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100549ARMNN_AUTO_TEST_CASE_WITH_THF(Abs, AbsTest)
Narumol Prangnawaratb98bbcf2019-04-15 14:14:27 +0100550
Sadik Armagan6095ba52019-09-13 17:07:19 +0100551// Sqrt Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100552ARMNN_AUTO_TEST_CASE_WITH_THF(Sqrt, SqrtTest)
553ARMNN_AUTO_TEST_CASE_WITH_THF(SqrtNN, SqrtNNTest)
Sadik Armagan6095ba52019-09-13 17:07:19 +0100554
Teresa Charline32c8442020-03-05 14:41:50 +0000555// Square Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100556ARMNN_AUTO_TEST_CASE_WITH_THF(Square, SquareTest)
Teresa Charline32c8442020-03-05 14:41:50 +0000557
558// Tanh Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100559ARMNN_AUTO_TEST_CASE_WITH_THF(Tanh, TanhTest)
Teresa Charline32c8442020-03-05 14:41:50 +0000560
561// Elu Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100562ARMNN_AUTO_TEST_CASE_WITH_THF(Elu, EluTest)
Teresa Charline32c8442020-03-05 14:41:50 +0000563
Narumol Prangnawarat65d30962019-03-14 11:55:03 +0000564// Softmax
Colm Donelanade8c1d2021-03-24 14:51:31 +0000565// Moved to NeonLayerTests_NDK_Bug.cpp
566//ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSoftmaxBeta1, SimpleSoftmaxTest, 1.0f)
567//ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSoftmaxBeta2, SimpleSoftmaxTest, 2.0f)
Sadik Armagan56785c72020-08-27 12:57:20 +0100568ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSoftmaxBeta1Uint8, SimpleSoftmaxUint8Test, 1.0f)
569ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSoftmaxBeta2Uint8, SimpleSoftmaxUint8Test, 2.0f)
telsoa014fcda012018-03-09 14:13:49 +0000570
Keith Davis69e653f2020-07-02 11:49:26 +0100571// LogSoftmax
Colm Donelanade8c1d2021-03-24 14:51:31 +0000572// Moved to NeonLayerTests_NDK_Bug.cpp
573//ARMNN_AUTO_TEST_CASE_WITH_THF(LogSoftmaxFloat32_1, LogSoftmaxTest1<DataType::Float32>)
Keith Davis69e653f2020-07-02 11:49:26 +0100574
Mike Kelly0be3a882020-01-24 11:27:50 +0000575// Space To Batch Nd
Sadik Armagan714fe5b2020-08-27 12:46:36 +0100576ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdSimpleFloat32, SpaceToBatchNdSimpleFloat32Test)
577ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiChannelsFloat32, SpaceToBatchNdMultiChannelsFloat32Test)
578ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiBlockFloat32, SpaceToBatchNdMultiBlockFloat32Test)
579ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdPaddingFloat32, SpaceToBatchNdPaddingFloat32Test)
Mike Kelly0be3a882020-01-24 11:27:50 +0000580
Sadik Armagan714fe5b2020-08-27 12:46:36 +0100581ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdSimpleUint8, SpaceToBatchNdSimpleUint8Test)
582ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiChannelsUint8, SpaceToBatchNdMultiChannelsUint8Test)
583ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiBlockUint8, SpaceToBatchNdMultiBlockUint8Test)
584ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdPaddingUint8, SpaceToBatchNdPaddingUint8Test)
Mike Kelly0be3a882020-01-24 11:27:50 +0000585
Sadik Armagan714fe5b2020-08-27 12:46:36 +0100586ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdSimpleNhwcFloat32, SpaceToBatchNdSimpleNhwcFloat32Test)
Finn Williams826a5432020-08-27 16:15:20 +0100587ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiChannelsNhwcFloat32,
588 SpaceToBatchNdMultiChannelsNhwcFloat32Test)
Sadik Armagan714fe5b2020-08-27 12:46:36 +0100589ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiBlockNhwcFloat32, SpaceToBatchNdMultiBlockNhwcFloat32Test)
590ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdPaddingNhwcFloat32, SpaceToBatchNdPaddingNhwcFloat32Test)
Mike Kelly0be3a882020-01-24 11:27:50 +0000591
Sadik Armagan714fe5b2020-08-27 12:46:36 +0100592ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdSimpleNhwcUint8, SpaceToBatchNdSimpleNhwcUint8Test)
Finn Williams826a5432020-08-27 16:15:20 +0100593ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiChannelsNhwcUint8,
594 SpaceToBatchNdMultiChannelsNhwcUint8Test)
Sadik Armagan714fe5b2020-08-27 12:46:36 +0100595ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiBlockNhwcUint8, SpaceToBatchNdMultiBlockNhwcUint8Test)
596ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdPaddingNhwcUint8, SpaceToBatchNdPaddingNhwcUint8Test)
Mike Kelly0be3a882020-01-24 11:27:50 +0000597
Ellen Norris-Thompson29794572019-06-26 16:40:36 +0100598// SpaceToDepth
Sadik Armagan5e719842020-08-28 12:17:33 +0100599ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToDepthNchwAsymmQ8, SpaceToDepthNchwAsymmQ8Test)
600ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToDepthNhwcsymmQ8, SpaceToDepthNhwcAsymmQ8Test)
Ellen Norris-Thompson29794572019-06-26 16:40:36 +0100601
Sadik Armagan5e719842020-08-28 12:17:33 +0100602ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToDepthNhwc1Float32, SpaceToDepthNhwcFloat32Test1)
603ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToDepthNchw1Float32, SpaceToDepthNchwFloat32Test1)
Ellen Norris-Thompson29794572019-06-26 16:40:36 +0100604
Sadik Armagan5e719842020-08-28 12:17:33 +0100605ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToDepthNhwc2Float32, SpaceToDepthNhwcFloat32Test2)
606ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToDepthNchw2Float32, SpaceToDepthNchwFloat32Test2)
Ellen Norris-Thompson29794572019-06-26 16:40:36 +0100607
Sadik Armagan5e719842020-08-28 12:17:33 +0100608ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToDepthNhwcQSymm16, SpaceToDepthNhwcQSymm16Test)
609ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToDepthNchwQSymm16, SpaceToDepthNchwQSymm16Test)
Ellen Norris-Thompson29794572019-06-26 16:40:36 +0100610
telsoa01c577f2c2018-08-31 09:22:23 +0100611// Splitter
Sadik Armagan5e719842020-08-28 12:17:33 +0100612ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSplitterFloat32, SplitterFloat32Test)
613ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSplitterUint8, SplitterUint8Test)
telsoa014fcda012018-03-09 14:13:49 +0000614
Sadik Armagan5e719842020-08-28 12:17:33 +0100615ARMNN_AUTO_TEST_CASE_WITH_THF(CopyViaSplitterFloat32, CopyViaSplitterFloat32Test)
616ARMNN_AUTO_TEST_CASE_WITH_THF(CopyViaSplitterUint8, CopyViaSplitterUint8Test)
telsoa014fcda012018-03-09 14:13:49 +0000617
Jim Flynn4ed6c832019-05-20 11:02:46 +0100618// Concat
Keith Davisf500d6c2020-08-31 08:32:55 +0100619ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConcat, ConcatTest)
620ARMNN_AUTO_TEST_CASE_WITH_THF(ConcatUint8, ConcatUint8Test)
621ARMNN_AUTO_TEST_CASE_WITH_THF(ConcatUint8DifferentInputOutputQParam,
Derek Lambertif90c56d2020-01-10 17:14:08 +0000622 ConcatDifferentInputOutputQParamTest<DataType::QAsymmU8>, false)
telsoa014fcda012018-03-09 14:13:49 +0000623
Narumol Prangnawarat250d3922020-03-30 16:11:04 +0100624// Convert from BFloat16 to Float32
Keith Davisf500d6c2020-08-31 08:32:55 +0100625ARMNN_AUTO_TEST_CASE_WITH_THF(ConvertBf16ToFp32, ConvertBf16ToFp32Test)
Narumol Prangnawarat250d3922020-03-30 16:11:04 +0100626
627// Convert from Float32 to BFloat16
Keith Davisf500d6c2020-08-31 08:32:55 +0100628ARMNN_AUTO_TEST_CASE_WITH_THF(ConvertFp32ToBf16, ConvertFp32ToBf16Test)
Narumol Prangnawarat250d3922020-03-30 16:11:04 +0100629
telsoa014fcda012018-03-09 14:13:49 +0000630// Fully Connected
Finn Williams7faf9a82020-08-27 10:37:36 +0100631ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleFullyConnected, FullyConnectedFloat32Test, false, false)
632ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleFullyConnectedWithBias, FullyConnectedFloat32Test, true, false)
633ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleFullyConnectedWithTranspose, FullyConnectedFloat32Test, false, true)
634ARMNN_AUTO_TEST_CASE_WITH_THF(FullyConnectedLarge, FullyConnectedLargeTest, false)
635ARMNN_AUTO_TEST_CASE_WITH_THF(FullyConnectedLargeTransposed, FullyConnectedLargeTest, true)
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000636ARMNN_AUTO_TEST_CASE_WITH_THF(FullyConnectedUint8, FullyConnectedTest<DataType::QAsymmU8>, false, true)
637ARMNN_AUTO_TEST_CASE_WITH_THF(FullyConnectedBiasedUint8, FullyConnectedTest<DataType::QAsymmU8>, true, true)
telsoa014fcda012018-03-09 14:13:49 +0000638
639// Add
Keith Davis33a626f2020-08-27 15:38:12 +0100640ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleAdd, AdditionTest)
641ARMNN_AUTO_TEST_CASE_WITH_THF(Add5d, Addition5dTest)
642ARMNN_AUTO_TEST_CASE_WITH_THF(AddBroadcast, AdditionBroadcastTest)
643ARMNN_AUTO_TEST_CASE_WITH_THF(AddBroadcast1Element, AdditionBroadcast1ElementTest)
telsoa014fcda012018-03-09 14:13:49 +0000644
David Beckbc392452018-09-10 14:47:28 +0100645// Sub
Keith Davis33a626f2020-08-27 15:38:12 +0100646ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSub, SubtractionTest)
647ARMNN_AUTO_TEST_CASE_WITH_THF(SubBroadcast1Element, SubtractionBroadcast1ElementTest)
648ARMNN_AUTO_TEST_CASE_WITH_THF(SubBroadcast, SubtractionBroadcastTest)
649ARMNN_AUTO_TEST_CASE_WITH_THF(SubtractionUint8, SubtractionUint8Test)
650ARMNN_AUTO_TEST_CASE_WITH_THF(SubBroadcastUint8, SubtractionBroadcastUint8Test)
651ARMNN_AUTO_TEST_CASE_WITH_THF(SubBroadcast1ElementUint8, SubtractionBroadcast1ElementUint8Test)
David Beckbc392452018-09-10 14:47:28 +0100652
Pablo Telloe61f0712020-01-23 10:37:17 +0000653// Div
Keith Davis33a626f2020-08-27 15:38:12 +0100654ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleDivision, DivisionTest)
655ARMNN_AUTO_TEST_CASE_WITH_THF(DivisionByZero, DivisionByZeroTest)
656ARMNN_AUTO_TEST_CASE_WITH_THF(DivisionBroadcast1Element, DivisionBroadcast1ElementTest)
657ARMNN_AUTO_TEST_CASE_WITH_THF(DivisionBroadcast1DVector, DivisionBroadcast1DVectorTest)
Pablo Telloe61f0712020-01-23 10:37:17 +0000658
telsoa014fcda012018-03-09 14:13:49 +0000659// Mul
Keith Davis33a626f2020-08-27 15:38:12 +0100660ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMultiplication, MultiplicationTest)
661ARMNN_AUTO_TEST_CASE_WITH_THF(MultiplicationBroadcast1Element, MultiplicationBroadcast1ElementTest)
662ARMNN_AUTO_TEST_CASE_WITH_THF(MultiplicationBroadcast1DVector, MultiplicationBroadcast1DVectorTest)
663ARMNN_AUTO_TEST_CASE_WITH_THF(MultiplicationUint8, MultiplicationUint8Test)
664ARMNN_AUTO_TEST_CASE_WITH_THF(MultiplicationBroadcast1ElementUint8, MultiplicationBroadcast1ElementUint8Test)
665ARMNN_AUTO_TEST_CASE_WITH_THF(MultiplicationBroadcast1DVectorUint8, MultiplicationBroadcast1DVectorUint8Test)
666ARMNN_AUTO_TEST_CASE_WITH_THF(Multiplication5d, Multiplication5dTest)
telsoa014fcda012018-03-09 14:13:49 +0000667
668// Batch Norm
Keith Davis33a626f2020-08-27 15:38:12 +0100669ARMNN_AUTO_TEST_CASE_WITH_THF(BatchNormFloat32, BatchNormFloat32Test)
670ARMNN_AUTO_TEST_CASE_WITH_THF(BatchNormFloat32Nhwc, BatchNormFloat32NhwcTest)
telsoa014fcda012018-03-09 14:13:49 +0000671
David Monahan97451b42020-12-03 09:48:06 +0000672// Rank
673ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize1Float16, RankDimSize1Test<DataType::Float16>)
674ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize1Float32, RankDimSize1Test<DataType::Float32>)
675ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize1QAsymmU8, RankDimSize1Test<DataType::QAsymmU8>)
676ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize1Signed32, RankDimSize1Test<DataType::Signed32>)
677ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize1QSymmS16, RankDimSize1Test<DataType::QSymmS16>)
678ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize1QAsymmS8, RankDimSize1Test<DataType::QAsymmS8>)
679ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize1BFloat16, RankDimSize1Test<DataType::BFloat16>)
680
681ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize2Float16, RankDimSize2Test<DataType::Float16>)
682ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize2Float32, RankDimSize2Test<DataType::Float32>)
683ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize2QAsymmU8, RankDimSize2Test<DataType::QAsymmU8>)
684ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize2Signed32, RankDimSize2Test<DataType::Signed32>)
685ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize2QSymmS16, RankDimSize2Test<DataType::QSymmS16>)
686ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize2QAsymmS8, RankDimSize2Test<DataType::QAsymmS8>)
687ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize2BFloat16, RankDimSize2Test<DataType::BFloat16>)
688
689ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize3Float16, RankDimSize3Test<DataType::Float16>)
690ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize3Float32, RankDimSize3Test<DataType::Float32>)
691ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize3QAsymmU8, RankDimSize3Test<DataType::QAsymmU8>)
692ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize3Signed32, RankDimSize3Test<DataType::Signed32>)
693ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize3QSymmS16, RankDimSize3Test<DataType::QSymmS16>)
694ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize3QAsymmS8, RankDimSize3Test<DataType::QAsymmS8>)
695ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize3BFloat16, RankDimSize3Test<DataType::BFloat16>)
696
697ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize4Float16, RankDimSize4Test<DataType::Float16>)
698ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize4Float32, RankDimSize4Test<DataType::Float32>)
699ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize4QAsymmU8, RankDimSize4Test<DataType::QAsymmU8>)
700ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize4Signed32, RankDimSize4Test<DataType::Signed32>)
701ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize4QSymmS16, RankDimSize4Test<DataType::QSymmS16>)
702ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize4QAsymmS8, RankDimSize4Test<DataType::QAsymmS8>)
703ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize4BFloat16, RankDimSize4Test<DataType::BFloat16>)
704
Sadik Armagan0d4863d2019-10-09 14:26:32 +0100705// InstanceNormalization
Finn Williamsc43de6a2020-08-27 11:13:25 +0100706ARMNN_AUTO_TEST_CASE_WITH_THF(InstanceNormFloat32Nchw, InstanceNormFloat32Test, DataLayout::NCHW);
707ARMNN_AUTO_TEST_CASE_WITH_THF(InstanceNormFloat32Nhwc, InstanceNormFloat32Test, DataLayout::NHWC);
Sadik Armagan0d4863d2019-10-09 14:26:32 +0100708
Finn Williamsc43de6a2020-08-27 11:13:25 +0100709ARMNN_AUTO_TEST_CASE_WITH_THF(InstanceNormFloat32Nchw2, InstanceNormFloat32Test2, DataLayout::NCHW);
710ARMNN_AUTO_TEST_CASE_WITH_THF(InstanceNormFloat32Nhwc2, InstanceNormFloat32Test2, DataLayout::NHWC);
Sadik Armagan0d4863d2019-10-09 14:26:32 +0100711
telsoa014fcda012018-03-09 14:13:49 +0000712// Constant
Keith Davisf500d6c2020-08-31 08:32:55 +0100713ARMNN_AUTO_TEST_CASE_WITH_THF(Constant, ConstantTest)
714ARMNN_AUTO_TEST_CASE_WITH_THF(ConstantUint8, ConstantUint8SimpleQuantizationScaleNoOffsetTest)
telsoa014fcda012018-03-09 14:13:49 +0000715
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100716// Concat
Keith Davisf500d6c2020-08-31 08:32:55 +0100717ARMNN_AUTO_TEST_CASE_WITH_THF(Concat1d, Concat1dTest)
718ARMNN_AUTO_TEST_CASE_WITH_THF(Concat1dUint8, Concat1dUint8Test)
telsoa014fcda012018-03-09 14:13:49 +0000719
Keith Davisf500d6c2020-08-31 08:32:55 +0100720ARMNN_AUTO_TEST_CASE_WITH_THF(Concat2dDim0, Concat2dDim0Test)
721ARMNN_AUTO_TEST_CASE_WITH_THF(Concat2dDim0Uint8, Concat2dDim0Uint8Test)
722ARMNN_AUTO_TEST_CASE_WITH_THF(Concat2dDim1, Concat2dDim1Test)
723ARMNN_AUTO_TEST_CASE_WITH_THF(Concat2dDim1Uint8, Concat2dDim1Uint8Test)
telsoa014fcda012018-03-09 14:13:49 +0000724
Keith Davisf500d6c2020-08-31 08:32:55 +0100725ARMNN_AUTO_TEST_CASE_WITH_THF(Concat2dDim0DiffInputDims, Concat2dDim0DiffInputDimsTest)
726ARMNN_AUTO_TEST_CASE_WITH_THF(Concat2dDim0DiffInputDimsUint8, Concat2dDim0DiffInputDimsUint8Test)
727ARMNN_AUTO_TEST_CASE_WITH_THF(Concat2dDim1DiffInputDims, Concat2dDim1DiffInputDimsTest)
728ARMNN_AUTO_TEST_CASE_WITH_THF(Concat2dDim1DiffInputDimsUint8, Concat2dDim1DiffInputDimsUint8Test)
telsoa014fcda012018-03-09 14:13:49 +0000729
Keith Davisf500d6c2020-08-31 08:32:55 +0100730ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim0, Concat3dDim0Test)
731ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim0Uint8, Concat3dDim0Uint8Test)
732ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim1, Concat3dDim1Test)
733ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim1Uint8, Concat3dDim1Uint8Test)
734ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim2, Concat3dDim2Test, false)
735ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim2Uint8, Concat3dDim2Uint8Test, false)
telsoa014fcda012018-03-09 14:13:49 +0000736
Keith Davisf500d6c2020-08-31 08:32:55 +0100737ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim0DiffInputDims, Concat3dDim0DiffInputDimsTest)
738ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim0DiffInputDimsUint8, Concat3dDim0DiffInputDimsUint8Test)
739ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim1DiffInputDims, Concat3dDim1DiffInputDimsTest)
740ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim1DiffInputDimsUint8, Concat3dDim1DiffInputDimsUint8Test)
741ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim2DiffInputDims, Concat3dDim2DiffInputDimsTest, false)
742ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim2DiffInputDimsUint8, Concat3dDim2DiffInputDimsUint8Test, false)
telsoa014fcda012018-03-09 14:13:49 +0000743
Keith Davisf500d6c2020-08-31 08:32:55 +0100744ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDim0, Concat4dDim0Test)
745ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDim1, Concat4dDim1Test)
746ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDim3, Concat4dDim3Test, false)
747ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDim0Uint8, Concat4dDim0Uint8Test)
748ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDim1Uint8, Concat4dDim1Uint8Test)
749ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDim3Uint8, Concat4dDim3Uint8Test, false)
narpra015cdda352018-11-19 15:30:27 +0000750
Keith Davisf500d6c2020-08-31 08:32:55 +0100751ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDiffShapeDim0, Concat4dDiffShapeDim0Test)
752ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDiffShapeDim1, Concat4dDiffShapeDim1Test)
753ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDiffShapeDim3, Concat4dDiffShapeDim3Test, false)
754ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDiffShapeDim0Uint8, Concat4dDiffShapeDim0Uint8Test)
755ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDiffShapeDim1Uint8, Concat4dDiffShapeDim1Uint8Test)
756ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDiffShapeDim3Uint8, Concat4dDiffShapeDim3Uint8Test, false)
Sadik Armagan0d4863d2019-10-09 14:26:32 +0100757
telsoa014fcda012018-03-09 14:13:49 +0000758// L2 Normalization
Finn Williamsc43de6a2020-08-27 11:13:25 +0100759ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization1d, L2Normalization1dTest, DataLayout::NCHW)
760ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization2d, L2Normalization2dTest, DataLayout::NCHW)
761ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization3d, L2Normalization3dTest, DataLayout::NCHW)
762ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization4d, L2Normalization4dTest, DataLayout::NCHW)
Matteo Martincigh539b44d2018-10-01 09:26:39 +0100763
Colm Donelanade8c1d2021-03-24 14:51:31 +0000764// Moved to NeonLayerTests_NDK_Bug.cpp
765//ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization1dNhwc, L2Normalization1dTest, DataLayout::NHWC)
Finn Williamsc43de6a2020-08-27 11:13:25 +0100766ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization2dNhwc, L2Normalization2dTest, DataLayout::NHWC)
767ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization3dNhwc, L2Normalization3dTest, DataLayout::NHWC)
768ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization4dNhwc, L2Normalization4dTest, DataLayout::NHWC)
telsoa014fcda012018-03-09 14:13:49 +0000769
Finn Williamsc43de6a2020-08-27 11:13:25 +0100770ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization2dShape, L2Normalization2dShapeTest);
Matthew Jackson82b15ed2019-07-25 16:14:30 +0100771
Finn Williamsc43de6a2020-08-27 11:13:25 +0100772ARMNN_AUTO_TEST_CASE_WITH_THF(L2NormalizationDefaultEpsilon, L2NormalizationDefaultEpsilonTest, DataLayout::NCHW)
773ARMNN_AUTO_TEST_CASE_WITH_THF(L2NormalizationNonDefaultEpsilon, L2NormalizationNonDefaultEpsilonTest, DataLayout::NCHW)
Ferran Balaguer0c64f462019-06-18 16:32:30 +0100774
telsoa014fcda012018-03-09 14:13:49 +0000775// Floor
Finn Williams7faf9a82020-08-27 10:37:36 +0100776ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleFloor, SimpleFloorTest<DataType::Float32>)
telsoa014fcda012018-03-09 14:13:49 +0000777
Teresa Charlinf540eb82020-04-10 19:24:55 +0100778// Gather
Finn Williamsc43de6a2020-08-27 11:13:25 +0100779ARMNN_AUTO_TEST_CASE_WITH_THF(Gather1dParamsFloat32, Gather1dParamsFloat32Test)
780ARMNN_AUTO_TEST_CASE_WITH_THF(Gather1dParamsUint8, Gather1dParamsUint8Test)
781ARMNN_AUTO_TEST_CASE_WITH_THF(GatherMultiDimParamsFloat32, GatherMultiDimParamsFloat32Test)
782ARMNN_AUTO_TEST_CASE_WITH_THF(GatherMultiDimParamsUint8, GatherMultiDimParamsUint8Test)
Teresa Charlinf540eb82020-04-10 19:24:55 +0100783
Teresa Charlinbd22c7d2022-04-26 18:14:12 +0100784// GatherNd
785ARMNN_AUTO_TEST_CASE_WITH_THF(GatherNd2dFloat32, SimpleGatherNd2dTest<DataType::Float32>)
786ARMNN_AUTO_TEST_CASE_WITH_THF(GatherNd3dFloat32, SimpleGatherNd3dTest<DataType::Float32>)
787ARMNN_AUTO_TEST_CASE_WITH_THF(GatherNd4dFloat32, SimpleGatherNd4dTest<DataType::Float32>)
788ARMNN_AUTO_TEST_CASE_WITH_THF(GatherNd2dInt8, SimpleGatherNd2dTest<DataType::QAsymmS8>)
789ARMNN_AUTO_TEST_CASE_WITH_THF(GatherNd3dInt8, SimpleGatherNd3dTest<DataType::QAsymmS8>)
790ARMNN_AUTO_TEST_CASE_WITH_THF(GatherNd4dInt8, SimpleGatherNd4dTest<DataType::QAsymmS8>)
791ARMNN_AUTO_TEST_CASE_WITH_THF(GatherNd2dInt32, SimpleGatherNd2dTest<DataType::Signed32>)
792ARMNN_AUTO_TEST_CASE_WITH_THF(GatherNd3dInt32, SimpleGatherNd3dTest<DataType::Signed32>)
793ARMNN_AUTO_TEST_CASE_WITH_THF(GatherNd4dInt32, SimpleGatherNd4dTest<DataType::Signed32>)
794
Teresa Charlincedd34f2020-03-30 11:17:30 +0100795// Equal
Keith Davisf500d6c2020-08-31 08:32:55 +0100796ARMNN_AUTO_TEST_CASE_WITH_THF(EqualSimple, EqualSimpleTest)
797ARMNN_AUTO_TEST_CASE_WITH_THF(EqualBroadcast1Element, EqualBroadcast1ElementTest)
798ARMNN_AUTO_TEST_CASE_WITH_THF(EqualBroadcast1dVector, EqualBroadcast1dVectorTest)
Teresa Charlincedd34f2020-03-30 11:17:30 +0100799
Keith Davisf500d6c2020-08-31 08:32:55 +0100800ARMNN_AUTO_TEST_CASE_WITH_THF(EqualSimpleUint8, EqualSimpleUint8Test)
801ARMNN_AUTO_TEST_CASE_WITH_THF(EqualBroadcast1ElementUint8, EqualBroadcast1ElementUint8Test)
802ARMNN_AUTO_TEST_CASE_WITH_THF(EqualBroadcast1dVectorUint8, EqualBroadcast1dVectorUint8Test)
Teresa Charlincedd34f2020-03-30 11:17:30 +0100803
kevmay01eed85922019-01-28 08:37:25 +0000804// Greater
Keith Davisf500d6c2020-08-31 08:32:55 +0100805ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterSimple, GreaterSimpleTest)
806ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterBroadcast1Element, GreaterBroadcast1ElementTest)
807ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterBroadcast1dVector, GreaterBroadcast1dVectorTest)
Aron Virginas-Tar3bc00ec2019-10-18 15:42:58 +0100808
Keith Davisf500d6c2020-08-31 08:32:55 +0100809ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterSimpleUint8, GreaterSimpleUint8Test)
810ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterBroadcast1ElementUint8, GreaterBroadcast1ElementUint8Test)
811ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterBroadcast1dVectorUint8, GreaterBroadcast1dVectorUint8Test)
kevmay01eed85922019-01-28 08:37:25 +0000812
Teresa Charlincedd34f2020-03-30 11:17:30 +0100813// GreaterOrEqual
Keith Davisf500d6c2020-08-31 08:32:55 +0100814ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterOrEqualSimple, GreaterOrEqualSimpleTest)
815ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterOrEqualBroadcast1Element, GreaterOrEqualBroadcast1ElementTest)
816ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterOrEqualBroadcast1dVector, GreaterOrEqualBroadcast1dVectorTest)
Teresa Charlincedd34f2020-03-30 11:17:30 +0100817
Keith Davisf500d6c2020-08-31 08:32:55 +0100818ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterOrEqualSimpleUint8, GreaterOrEqualSimpleUint8Test)
819ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterOrEqualBroadcast1ElementUint8, GreaterOrEqualBroadcast1ElementUint8Test)
820ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterOrEqualBroadcast1dVectorUint8, GreaterOrEqualBroadcast1dVectorUint8Test)
Teresa Charlincedd34f2020-03-30 11:17:30 +0100821
822// Less
Keith Davisf500d6c2020-08-31 08:32:55 +0100823ARMNN_AUTO_TEST_CASE_WITH_THF(LessSimple, LessSimpleTest)
824ARMNN_AUTO_TEST_CASE_WITH_THF(LessBroadcast1Element, LessBroadcast1ElementTest)
825ARMNN_AUTO_TEST_CASE_WITH_THF(LessBroadcast1dVector, LessBroadcast1dVectorTest)
Teresa Charlincedd34f2020-03-30 11:17:30 +0100826
Keith Davisf500d6c2020-08-31 08:32:55 +0100827ARMNN_AUTO_TEST_CASE_WITH_THF(LessSimpleUint8, LessSimpleUint8Test)
828ARMNN_AUTO_TEST_CASE_WITH_THF(LessBroadcast1ElementUint8, LessBroadcast1ElementUint8Test)
829ARMNN_AUTO_TEST_CASE_WITH_THF(LessBroadcast1dVectorUint8, LessBroadcast1dVectorUint8Test)
Teresa Charlincedd34f2020-03-30 11:17:30 +0100830
831// LessOrEqual
Keith Davisf500d6c2020-08-31 08:32:55 +0100832ARMNN_AUTO_TEST_CASE_WITH_THF(LessOrEqualSimple, LessOrEqualSimpleTest)
833ARMNN_AUTO_TEST_CASE_WITH_THF(LessOrEqualBroadcast1Element, LessOrEqualBroadcast1ElementTest)
834ARMNN_AUTO_TEST_CASE_WITH_THF(LessOrEqualBroadcast1dVector, LessOrEqualBroadcast1dVectorTest)
Teresa Charlincedd34f2020-03-30 11:17:30 +0100835
Keith Davisf500d6c2020-08-31 08:32:55 +0100836ARMNN_AUTO_TEST_CASE_WITH_THF(LessOrEqualSimpleUint8, LessOrEqualSimpleUint8Test)
837ARMNN_AUTO_TEST_CASE_WITH_THF(LessOrEqualBroadcast1ElementUint8, LessOrEqualBroadcast1ElementUint8Test)
838ARMNN_AUTO_TEST_CASE_WITH_THF(LessOrEqualBroadcast1dVectorUint8, LessOrEqualBroadcast1dVectorUint8Test)
Teresa Charlincedd34f2020-03-30 11:17:30 +0100839
840// NotEqual
Keith Davisf500d6c2020-08-31 08:32:55 +0100841ARMNN_AUTO_TEST_CASE_WITH_THF(NotEqualSimple, NotEqualSimpleTest)
842ARMNN_AUTO_TEST_CASE_WITH_THF(NotEqualBroadcast1Element, NotEqualBroadcast1ElementTest)
843ARMNN_AUTO_TEST_CASE_WITH_THF(NotEqualBroadcast1dVector, NotEqualBroadcast1dVectorTest)
Teresa Charlincedd34f2020-03-30 11:17:30 +0100844
Keith Davisf500d6c2020-08-31 08:32:55 +0100845ARMNN_AUTO_TEST_CASE_WITH_THF(NotEqualSimpleUint8, NotEqualSimpleUint8Test)
846ARMNN_AUTO_TEST_CASE_WITH_THF(NotEqualBroadcast1ElementUint8, NotEqualBroadcast1ElementUint8Test)
847ARMNN_AUTO_TEST_CASE_WITH_THF(NotEqualBroadcast1dVectorUint8, NotEqualBroadcast1dVectorUint8Test)
Teresa Charlincedd34f2020-03-30 11:17:30 +0100848
telsoa014fcda012018-03-09 14:13:49 +0000849// Reshape
Finn Williamsec36d3e2020-08-28 13:17:05 +0100850ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleReshapeFloat32, SimpleReshapeTest<armnn::DataType::Float32>)
851ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleReshapeInt8, SimpleReshapeTest<armnn::DataType::QAsymmS8>)
852ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleReshapeUint8, SimpleReshapeTest<armnn::DataType::QAsymmU8>)
853ARMNN_AUTO_TEST_CASE_WITH_THF(Reshape5d, Reshape5dTest<armnn::DataType::Float32>)
Narumol Prangnawarat0c95f4c2020-11-18 16:52:07 +0000854ARMNN_AUTO_TEST_CASE_WITH_THF(ReshapeBoolean, ReshapeBooleanTest)
telsoa014fcda012018-03-09 14:13:49 +0000855
Teresa Charlinf9de7712021-10-28 11:31:16 +0100856// Pad - Constant
Finn Williams826a5432020-08-27 16:15:20 +0100857ARMNN_AUTO_TEST_CASE_WITH_THF(PadFloat322d, PadFloat322dTest)
858ARMNN_AUTO_TEST_CASE_WITH_THF(PadFloat322dCustomPadding, PadFloat322dCustomPaddingTest)
859ARMNN_AUTO_TEST_CASE_WITH_THF(PadFloat323d, PadFloat323dTest)
860ARMNN_AUTO_TEST_CASE_WITH_THF(PadFloat324d, PadFloat324dTest)
Éanna Ó Catháin12055742019-01-25 10:01:40 +0000861
Finn Williams826a5432020-08-27 16:15:20 +0100862ARMNN_AUTO_TEST_CASE_WITH_THF(PadUint82d, PadUint82dTest)
863ARMNN_AUTO_TEST_CASE_WITH_THF(PadUint82dCustomPadding, PadUint82dCustomPaddingTest)
864ARMNN_AUTO_TEST_CASE_WITH_THF(PadUint83d, PadUint83dTest)
865ARMNN_AUTO_TEST_CASE_WITH_THF(PadUint84d, PadUint84dTest)
Teresa Charlinec8e1982019-07-02 16:24:09 +0100866
Finn Williams826a5432020-08-27 16:15:20 +0100867ARMNN_AUTO_TEST_CASE_WITH_THF(Pad2dQSymm16, Pad2dTestCommon<DataType::QSymmS16>, 2.0f, 0, 0.0f)
868ARMNN_AUTO_TEST_CASE_WITH_THF(Pad2dQSymm16CustomPadding, Pad2dTestCommon<DataType::QSymmS16>, 2.0f, 0, 1.0f)
869ARMNN_AUTO_TEST_CASE_WITH_THF(Pad3dQSymm16, Pad3dTestCommon<DataType::QSymmS16>, 2.0f, 0)
870ARMNN_AUTO_TEST_CASE_WITH_THF(Pad4dQSymm16, Pad4dTestCommon<DataType::QSymmS16>, 2.0f, 0)
FinnWilliamsArm8b797a82019-07-09 16:01:41 +0100871
Teresa Charlinf9de7712021-10-28 11:31:16 +0100872// Pad - Symmetric & Reflect
873ARMNN_AUTO_TEST_CASE_WITH_THF(PadSymmetric2dFloat32, PadSymmetric2dFloat32Test)
874ARMNN_AUTO_TEST_CASE_WITH_THF(PadReflect2dFloat32, PadReflect2dFloat32Test)
875ARMNN_AUTO_TEST_CASE_WITH_THF(PadSymmetric3dFloat32, PadSymmetric3dFloat32Test)
876ARMNN_AUTO_TEST_CASE_WITH_THF(PadReflect3dFloat32, PadReflect3dFloat32Test)
877ARMNN_AUTO_TEST_CASE_WITH_THF(PadSymmetric3dUint8, PadSymmetric3dUint8Test)
878ARMNN_AUTO_TEST_CASE_WITH_THF(PadReflect3dUint8, PadReflect3dUint8Test)
879ARMNN_AUTO_TEST_CASE_WITH_THF(PadSymmetric3dInt8, PadSymmetric3dInt8Test)
880ARMNN_AUTO_TEST_CASE_WITH_THF(PadReflect3dInt8, PadReflect3dInt8Test)
881ARMNN_AUTO_TEST_CASE_WITH_THF(PadSymmetric4dFloat32, PadSymmetric4dFloat32Test)
882ARMNN_AUTO_TEST_CASE_WITH_THF(PadReflect4dFloat32, PadReflect4dFloat32Test)
883ARMNN_AUTO_TEST_CASE_WITH_THF(PadSymmetric4dUint8, PadSymmetric4dUint8Test)
884ARMNN_AUTO_TEST_CASE_WITH_THF(PadReflect4dUint8, PadReflect4dUint8Test)
885ARMNN_AUTO_TEST_CASE_WITH_THF(PadSymmetric4dInt8, PadSymmetric4dInt8Test)
886ARMNN_AUTO_TEST_CASE_WITH_THF(PadReflect4dInt8, PadReflect4dInt8Test)
887
telsoa014fcda012018-03-09 14:13:49 +0000888// Permute
Finn Williams826a5432020-08-27 16:15:20 +0100889ARMNN_AUTO_TEST_CASE_WITH_THF(SimplePermuteFloat32, SimplePermuteTest<DataType::Float32>)
890ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteFloat32ValueSet1Test, PermuteValueSet1Test<DataType::Float32>)
891ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteFloat32ValueSet2Test, PermuteValueSet2Test<DataType::Float32>)
892ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteFloat32ValueSet3Test, PermuteValueSet3Test<DataType::Float32>)
893ARMNN_AUTO_TEST_CASE_WITH_THF(SimplePermuteQASymmS8, SimplePermuteTest<DataType::QAsymmS8>)
894ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteQASymmS8ValueSet1Test, PermuteValueSet1Test<DataType::QAsymmS8>)
895ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteQASymmS8ValueSet2Test, PermuteValueSet2Test<DataType::QAsymmS8>)
896ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteQASymmS8ValueSet3Test, PermuteValueSet3Test<DataType::QAsymmS8>)
897ARMNN_AUTO_TEST_CASE_WITH_THF(SimplePermuteQASymm8, SimplePermuteTest<DataType::QAsymmU8>)
898ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteQASymm8ValueSet1Test, PermuteValueSet1Test<DataType::QAsymmU8>)
899ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteQASymm8ValueSet2Test, PermuteValueSet2Test<DataType::QAsymmU8>)
900ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteQASymm8ValueSet3Test, PermuteValueSet3Test<DataType::QAsymmU8>)
surmeh01bceff2f2018-03-29 16:29:27 +0100901
Les Bellde9011b2018-10-03 10:37:52 +0100902// Lstm
Finn Williamsc43de6a2020-08-27 11:13:25 +0100903ARMNN_AUTO_TEST_CASE_WITH_THF(LstmLayerFloat32WithCifgWithPeepholeNoProjection,
904 LstmLayerFloat32WithCifgWithPeepholeNoProjectionTest)
905ARMNN_AUTO_TEST_CASE_WITH_THF(LstmLayerFloat32NoCifgNoPeepholeNoProjection,
906 LstmLayerFloat32NoCifgNoPeepholeNoProjectionTest)
907ARMNN_AUTO_TEST_CASE_WITH_THF(LstmLayerFloat32NoCifgWithPeepholeWithProjection,
908 LstmLayerFloat32NoCifgWithPeepholeWithProjectionTest)
Colm Donelanade8c1d2021-03-24 14:51:31 +0000909// Moved to NeonLayerTests_NDK_Bug.cpp
910//ARMNN_AUTO_TEST_CASE_WITH_THF(LstmLayerFloat32NoCifgWithPeepholeWithProjectionWithLayerNorm,
911// LstmLayerFloat32NoCifgWithPeepholeWithProjectionWithLayerNormTest)
Les Bellde9011b2018-10-03 10:37:52 +0100912
James Conroycc340932020-05-12 18:08:52 +0100913// QLstm
Finn Williamsc43de6a2020-08-27 11:13:25 +0100914ARMNN_AUTO_TEST_CASE_WITH_THF(QLstm, QLstmTest)
915ARMNN_AUTO_TEST_CASE_WITH_THF(QLstm1, QLstmTest1)
916ARMNN_AUTO_TEST_CASE_WITH_THF(QLstm2, QLstmTest2)
James Conroycc340932020-05-12 18:08:52 +0100917
918// QuantizedLstm
Finn Williamsc43de6a2020-08-27 11:13:25 +0100919ARMNN_AUTO_TEST_CASE_WITH_THF(QuantizedLstm, QuantizedLstmTest)
Francis Murtagh4fc3c482019-08-02 13:20:54 +0100920
Cathal Corbettfd5bec42022-03-03 15:13:23 +0000921// Unidirectional Sequence Lstm
922ARMNN_AUTO_TEST_CASE_WITH_THF(UnidirectionalSequenceLstmLayerFloat32TimeMajorSingleBatch,
923 UnidirectionalSequenceLstmLayerFloat32TimeMajorSingleBatchTest)
924ARMNN_AUTO_TEST_CASE_WITH_THF(UnidirectionalSequenceLstmLayerFloat32BatchMajorSingleBatch,
925 UnidirectionalSequenceLstmLayerFloat32BatchMajorSingleBatchTest)
926ARMNN_AUTO_TEST_CASE_WITH_THF(UnidirectionalSequenceLstmLayerFloat32,
927 UnidirectionalSequenceLstmLayerFloat32Test)
928ARMNN_AUTO_TEST_CASE_WITH_THF(UnidirectionalSequenceLstmLayerFloat32TimeMajor,
929 UnidirectionalSequenceLstmLayerFloat32TimeMajorTest)
930ARMNN_AUTO_TEST_CASE_WITH_THF(UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjection,
931 UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest)
Cathal Corbettd83ffe02022-03-22 15:46:02 +0000932// Moved to NeonLayerTests_NDK_Bug.cpp
933//ARMNN_AUTO_TEST_CASE_WITH_THF(
934// UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNorm,
935// UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest)
Cathal Corbettfd5bec42022-03-03 15:13:23 +0000936ARMNN_AUTO_TEST_CASE_WITH_THF(UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjection,
937 UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest)
938
Matthew Benthamfd899962018-12-31 15:49:42 +0000939// Mean
Finn Williams826a5432020-08-27 16:15:20 +0100940ARMNN_AUTO_TEST_CASE_WITH_THF(MeanSimpleFloat32, MeanSimpleTest<DataType::Float32>)
941ARMNN_AUTO_TEST_CASE_WITH_THF(MeanSimpleAxisFloat32, MeanSimpleAxisTest<DataType::Float32>)
942ARMNN_AUTO_TEST_CASE_WITH_THF(MeanKeepDimsFloat32, MeanKeepDimsTest<DataType::Float32>)
943ARMNN_AUTO_TEST_CASE_WITH_THF(MeanMultipleDimsFloat32, MeanMultipleDimsTest<DataType::Float32>)
944ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts1Float32, MeanVts1Test<DataType::Float32>)
945ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts2Float32, MeanVts2Test<DataType::Float32>)
946ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts3Float32, MeanVts3Test<DataType::Float32>)
Matthew Benthamfd899962018-12-31 15:49:42 +0000947
Finn Williams826a5432020-08-27 16:15:20 +0100948ARMNN_AUTO_TEST_CASE_WITH_THF(MeanSimpleQuantisedAsymmS8, MeanSimpleTest<DataType::QAsymmS8>)
949ARMNN_AUTO_TEST_CASE_WITH_THF(MeanSimpleAxisQuantisedAsymmS8, MeanSimpleAxisTest<DataType::QAsymmS8>)
950ARMNN_AUTO_TEST_CASE_WITH_THF(MeanKeepDimsQuantisedAsymmS8, MeanKeepDimsTest<DataType::QAsymmS8>)
951ARMNN_AUTO_TEST_CASE_WITH_THF(MeanMultipleDimsQuantisedAsymmS8, MeanMultipleDimsTest<DataType::QAsymmS8>)
952ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts1QuantisedAsymmS8, MeanVts1Test<DataType::QAsymmS8>)
953ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts2QuantisedAsymmS8, MeanVts2Test<DataType::QAsymmS8>)
954ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts3QuantisedAsymmS8, MeanVts3Test<DataType::QAsymmS8>)
Sadik Armagan303980c2020-04-17 12:45:14 +0100955
Finn Williams826a5432020-08-27 16:15:20 +0100956ARMNN_AUTO_TEST_CASE_WITH_THF(MeanSimpleQuantisedAsymm8, MeanSimpleTest<DataType::QAsymmU8>)
957ARMNN_AUTO_TEST_CASE_WITH_THF(MeanSimpleAxisQuantisedAsymm8, MeanSimpleAxisTest<DataType::QAsymmU8>)
958ARMNN_AUTO_TEST_CASE_WITH_THF(MeanKeepDimsQuantisedAsymm8, MeanKeepDimsTest<DataType::QAsymmU8>)
959ARMNN_AUTO_TEST_CASE_WITH_THF(MeanMultipleDimsQuantisedAsymm8, MeanMultipleDimsTest<DataType::QAsymmU8>)
960ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts1QuantisedAsymm8, MeanVts1Test<DataType::QAsymmU8>)
961ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts2QuantisedAsymm8, MeanVts2Test<DataType::QAsymmU8>)
962ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts3QuantisedAsymm8, MeanVts3Test<DataType::QAsymmU8>)
Matthew Benthamfd899962018-12-31 15:49:42 +0000963
Nattapat Chaimanowong4e6597a2018-12-20 14:14:06 +0000964// Max
Keith Davis33a626f2020-08-27 15:38:12 +0100965ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaximum, MaximumSimpleTest)
966ARMNN_AUTO_TEST_CASE_WITH_THF(MaximumBroadcast1Element, MaximumBroadcast1ElementTest)
967ARMNN_AUTO_TEST_CASE_WITH_THF(MaximumBroadcast1DVector, MaximumBroadcast1DVectorTest)
968ARMNN_AUTO_TEST_CASE_WITH_THF(MaximumUint8, MaximumUint8Test)
969ARMNN_AUTO_TEST_CASE_WITH_THF(MaximumBroadcast1ElementUint8, MaximumBroadcast1ElementUint8Test)
970ARMNN_AUTO_TEST_CASE_WITH_THF(MaximumBroadcast1DVectorUint8, MaximumBroadcast1DVectorUint8Test)
Nattapat Chaimanowong4e6597a2018-12-20 14:14:06 +0000971
Conor Kennedy54b21692019-01-09 07:57:38 +0000972// Min
Keith Davis33a626f2020-08-27 15:38:12 +0100973ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMinimum1, MinimumBroadcast1ElementTest1)
974ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMinimum2, MinimumBroadcast1ElementTest2)
975ARMNN_AUTO_TEST_CASE_WITH_THF(Minimum1DVectorUint8, MinimumBroadcast1DVectorUint8Test)
Conor Kennedy54b21692019-01-09 07:57:38 +0000976
narpra0155a97bc2018-10-02 14:35:53 +0100977// Normalization
Finn Williams826a5432020-08-27 16:15:20 +0100978ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleNormalizationAcross, SimpleNormalizationAcrossTest)
979ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleNormalizationWithin, SimpleNormalizationWithinTest)
980ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleNormalizationAcrossNhwc, SimpleNormalizationAcrossNhwcTest)
Sadik Armagan483c8112021-06-01 09:24:52 +0100981ARMNN_AUTO_TEST_CASE_WITH_THF(AcrossChannelNormalization, AcrossChannelNormalizationTest)
narpra0155a97bc2018-10-02 14:35:53 +0100982
Sadik Armaganc625f002018-12-17 11:32:16 +0000983// Resize Bilinear - NCHW data layout
Finn Williamsec36d3e2020-08-28 13:17:05 +0100984ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeBilinear, SimpleResizeBilinearTest<DataType::Float32>, DataLayout::NCHW)
985ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearNop, ResizeBilinearNopTest<DataType::Float32>, DataLayout::NCHW)
986ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearSqMin, ResizeBilinearSqMinTest<DataType::Float32>, DataLayout::NCHW)
987ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMin, ResizeBilinearMinTest<DataType::Float32>, DataLayout::NCHW)
988ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMag, ResizeBilinearMagTest<DataType::Float32>, DataLayout::NCHW)
Sadik Armaganc625f002018-12-17 11:32:16 +0000989
Finn Williamsec36d3e2020-08-28 13:17:05 +0100990ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeBilinearUint8,
991 SimpleResizeBilinearTest<DataType::QAsymmU8>,
992 DataLayout::NCHW)
993ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearNopUint8,
994 ResizeBilinearNopTest<DataType::QAsymmU8>,
995 DataLayout::NCHW)
996ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearSqMinUint8,
997 ResizeBilinearSqMinTest<DataType::QAsymmU8>,
998 DataLayout::NCHW)
999ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMinUint8,
1000 ResizeBilinearMinTest<DataType::QAsymmU8>,
1001 DataLayout::NCHW)
1002ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMagUint8,
1003 ResizeBilinearMagTest<DataType::QAsymmU8>,
1004 DataLayout::NCHW)
Conor Kennedydc1c0822019-02-01 07:23:17 +00001005
Finn Williamsec36d3e2020-08-28 13:17:05 +01001006ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeBilinear,
1007 HalfPixelCentersResizeBilinearTest<DataType::Float32>,
1008 DataLayout::NCHW)
1009ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeBilinear,
1010 AlignCornersResizeBilinearTest<DataType::Float32>,
1011 DataLayout::NCHW)
1012ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeBilinearInt8,
1013 HalfPixelCentersResizeBilinearTest<DataType::QAsymmS8>,
1014 DataLayout::NCHW)
1015ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeBilinearInt8,
1016 AlignCornersResizeBilinearTest<DataType::QAsymmS8>,
1017 DataLayout::NCHW)
1018ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeBilinearUint8,
1019 HalfPixelCentersResizeBilinearTest<DataType::QAsymmU8>,
1020 DataLayout::NCHW)
1021ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeBilinearUint8,
1022 AlignCornersResizeBilinearTest<DataType::QAsymmU8>,
1023 DataLayout::NCHW)
Teresa Charlin686f45c2020-07-23 17:59:00 +01001024
Sadik Armaganc625f002018-12-17 11:32:16 +00001025// Resize Bilinear - NHWC data layout
Finn Williamsec36d3e2020-08-28 13:17:05 +01001026ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearNopNhwc,
1027 ResizeBilinearNopTest<DataType::Float32>,
1028 DataLayout::NHWC)
1029ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeBilinearNhwc,
1030 SimpleResizeBilinearTest<DataType::Float32>,
1031 DataLayout::NHWC)
1032ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearSqMinNhwc,
1033 ResizeBilinearSqMinTest<DataType::Float32>,
1034 DataLayout::NHWC)
1035ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMinNhwc,
1036 ResizeBilinearMinTest<DataType::Float32>,
1037 DataLayout::NHWC)
1038ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMagNhwc,
1039 ResizeBilinearMagTest<DataType::Float32>,
1040 DataLayout::NHWC)
Ellen Norris-Thompsonc4223d12019-06-10 14:15:12 +01001041
Finn Williamsec36d3e2020-08-28 13:17:05 +01001042ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearNopUint8Nhwc,
1043 ResizeBilinearNopTest<DataType::QAsymmU8>,
1044 DataLayout::NHWC)
1045ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeBilinearUint8Nhwc,
1046 SimpleResizeBilinearTest<DataType::QAsymmU8>,
1047 DataLayout::NHWC)
1048ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearSqMinUint8Nhwc,
1049 ResizeBilinearSqMinTest<DataType::QAsymmU8>,
1050 DataLayout::NHWC)
1051ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMinUint8Nhwc,
1052 ResizeBilinearMinTest<DataType::QAsymmU8>,
1053 DataLayout::NHWC)
1054ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMagUint8Nhwc,
1055 ResizeBilinearMagTest<DataType::QAsymmU8>,
1056 DataLayout::NHWC)
Sadik Armaganc625f002018-12-17 11:32:16 +00001057
Finn Williamsec36d3e2020-08-28 13:17:05 +01001058ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeBilinearNhwc,
1059 HalfPixelCentersResizeBilinearTest<DataType::Float32>,
1060 DataLayout::NHWC)
1061ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeBilinearNhwc,
1062 AlignCornersResizeBilinearTest<DataType::Float32>,
1063 DataLayout::NHWC)
1064ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeBilinearInt8Nhwc,
1065 HalfPixelCentersResizeBilinearTest<DataType::QAsymmS8>,
1066 DataLayout::NHWC)
1067ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeBilinearInt8Nhwc,
1068 AlignCornersResizeBilinearTest<DataType::QAsymmS8>,
1069 DataLayout::NHWC)
1070ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeBilinearUint8Nhwc,
1071 HalfPixelCentersResizeBilinearTest<DataType::QAsymmU8>,
1072 DataLayout::NHWC)
1073ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeBilinearUint8Nhwc,
1074 AlignCornersResizeBilinearTest<DataType::QAsymmU8>,
1075 DataLayout::NHWC)
Teresa Charlin686f45c2020-07-23 17:59:00 +01001076
Ellen Norris-Thompson37e68682019-07-15 14:23:30 +01001077// Resize NearestNeighbor - NCHW
Finn Williamsec36d3e2020-08-28 13:17:05 +01001078ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeNearestNeighbor,
1079 SimpleResizeNearestNeighborTest<DataType::Float32>,
1080 DataLayout::NCHW)
1081ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborNop,
1082 ResizeNearestNeighborNopTest<DataType::Float32>,
1083 DataLayout::NCHW)
1084ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborSqMin,
1085 ResizeNearestNeighborSqMinTest<DataType::Float32>,
1086 DataLayout::NCHW)
1087ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMin,
1088 ResizeNearestNeighborMinTest<DataType::Float32>,
1089 DataLayout::NCHW)
1090ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMag,
1091 ResizeNearestNeighborMagTest<DataType::Float32>,
1092 DataLayout::NCHW, 0.1f, 50, 0.1f, 50)
Ellen Norris-Thompson37e68682019-07-15 14:23:30 +01001093
Finn Williamsec36d3e2020-08-28 13:17:05 +01001094ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeNearestNeighborUint8,
1095 SimpleResizeNearestNeighborTest<DataType::QAsymmU8>,
1096 DataLayout::NCHW)
1097ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborNopUint8,
1098 ResizeNearestNeighborNopTest<DataType::QAsymmU8>,
1099 DataLayout::NCHW)
1100ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborSqMinUint8,
1101 ResizeNearestNeighborSqMinTest<DataType::QAsymmU8>,
1102 DataLayout::NCHW)
1103ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMinUint8,
1104 ResizeNearestNeighborMinTest<DataType::QAsymmU8>,
1105 DataLayout::NCHW)
1106ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMagUint8,
1107 ResizeNearestNeighborMagTest<DataType::QAsymmU8>,
1108 DataLayout::NCHW, 0.1f, 50, 0.1f, 50)
Ellen Norris-Thompson37e68682019-07-15 14:23:30 +01001109
Finn Williamsec36d3e2020-08-28 13:17:05 +01001110ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeNearestNeighbour,
1111 HalfPixelCentersResizeNearestNeighbourTest<DataType::Float32>,
1112 DataLayout::NCHW)
1113ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeNearestNeighbour,
1114 AlignCornersResizeNearestNeighbourTest<DataType::Float32>,
1115 DataLayout::NCHW)
1116ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeNearestNeighbourInt8,
1117 HalfPixelCentersResizeNearestNeighbourTest<DataType::QAsymmS8>,
1118 DataLayout::NCHW)
1119ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeNearestNeighbourInt8,
1120 AlignCornersResizeNearestNeighbourTest<DataType::QAsymmS8>,
1121 DataLayout::NCHW)
1122ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeNearestNeighbourUint8,
1123 HalfPixelCentersResizeNearestNeighbourTest<DataType::QAsymmU8>,
1124 DataLayout::NCHW)
1125ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeNearestNeighbourUint8,
1126 AlignCornersResizeNearestNeighbourTest<DataType::QAsymmU8>,
1127 DataLayout::NCHW)
Teresa Charlin686f45c2020-07-23 17:59:00 +01001128
Ellen Norris-Thompson37e68682019-07-15 14:23:30 +01001129// Resize NearestNeighbor - NHWC
Finn Williamsec36d3e2020-08-28 13:17:05 +01001130ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborNopNhwc,
1131 ResizeNearestNeighborNopTest<DataType::Float32>,
1132 DataLayout::NHWC)
1133ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeNearestNeighborNhwc,
1134 SimpleResizeNearestNeighborTest<DataType::Float32>,
1135 DataLayout::NHWC)
1136ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborSqMinNhwc,
1137 ResizeNearestNeighborSqMinTest<DataType::Float32>,
1138 DataLayout::NHWC)
1139ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMinNhwc,
1140 ResizeNearestNeighborMinTest<DataType::Float32>,
1141 DataLayout::NHWC)
1142ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMagNhwc,
1143 ResizeNearestNeighborMagTest<DataType::Float32>,
1144 DataLayout::NHWC, 0.1f, 50, 0.1f, 50)
Ellen Norris-Thompson37e68682019-07-15 14:23:30 +01001145
Finn Williamsec36d3e2020-08-28 13:17:05 +01001146ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborNopUint8Nhwc,
1147 ResizeNearestNeighborNopTest<DataType::QAsymmU8>,
1148 DataLayout::NHWC)
1149ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeNearestNeighborUint8Nhwc,
1150 SimpleResizeNearestNeighborTest<DataType::QAsymmU8>,
1151 DataLayout::NHWC)
1152ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborSqMinUint8Nhwc,
1153 ResizeNearestNeighborSqMinTest<DataType::QAsymmU8>,
1154 DataLayout::NHWC)
1155ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMinUint8Nhwc,
1156 ResizeNearestNeighborMinTest<DataType::QAsymmU8>,
1157 DataLayout::NHWC)
1158ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMagUint8Nhwc,
1159 ResizeNearestNeighborMagTest<DataType::QAsymmU8>,
1160 DataLayout::NHWC, 0.1f, 50, 0.1f, 50)
Ellen Norris-Thompson37e68682019-07-15 14:23:30 +01001161
Finn Williamsec36d3e2020-08-28 13:17:05 +01001162ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeNearestNeighbourNhwc,
1163 HalfPixelCentersResizeNearestNeighbourTest<DataType::Float32>,
1164 DataLayout::NHWC)
1165ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeNearestNeighbourNhwc,
1166 AlignCornersResizeNearestNeighbourTest<DataType::Float32>,
1167 DataLayout::NHWC)
1168ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeNearestNeighbourInt8Nhwc,
1169 HalfPixelCentersResizeNearestNeighbourTest<DataType::QAsymmS8>,
1170 DataLayout::NHWC)
1171ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeNearestNeighbourInt8Nhwc,
1172 AlignCornersResizeNearestNeighbourTest<DataType::QAsymmS8>,
1173 DataLayout::NHWC)
1174ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeNearestNeighbourUint8Nhwc,
1175 HalfPixelCentersResizeNearestNeighbourTest<DataType::QAsymmU8>,
1176 DataLayout::NHWC)
1177ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeNearestNeighbourUint8Nhwc,
1178 AlignCornersResizeNearestNeighbourTest<DataType::QAsymmU8>,
1179 DataLayout::NHWC)
Teresa Charlin686f45c2020-07-23 17:59:00 +01001180
josh minor036f02d2019-11-15 14:53:22 -06001181// Slice
1182ARMNN_AUTO_TEST_CASE(Slice4dFloat32, Slice4dFloat32Test)
1183ARMNN_AUTO_TEST_CASE(Slice3dFloat32, Slice3dFloat32Test)
1184ARMNN_AUTO_TEST_CASE(Slice2dFloat32, Slice2dFloat32Test)
1185ARMNN_AUTO_TEST_CASE(Slice1dFloat32, Slice1dFloat32Test)
1186ARMNN_AUTO_TEST_CASE(Slice4dUint8, Slice4dUint8Test)
1187ARMNN_AUTO_TEST_CASE(Slice3dUint8, Slice3dUint8Test)
1188ARMNN_AUTO_TEST_CASE(Slice2dUint8, Slice2dUint8Test)
1189ARMNN_AUTO_TEST_CASE(Slice1dUint8, Slice1dUint8Test)
1190ARMNN_AUTO_TEST_CASE(Slice4dInt16, Slice4dInt16Test)
1191ARMNN_AUTO_TEST_CASE(Slice3dInt16, Slice3dInt16Test)
1192ARMNN_AUTO_TEST_CASE(Slice2dInt16, Slice2dInt16Test)
1193ARMNN_AUTO_TEST_CASE(Slice1dInt16, Slice1dInt16Test)
1194
FinnWilliamsArm1fa19192019-08-02 17:26:31 +01001195// Strided Slice
Sadik Armagan5e719842020-08-28 12:17:33 +01001196ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice4dFloat32, StridedSlice4dFloat32Test)
1197ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice4dReverseFloat32, StridedSlice4dReverseFloat32Test)
1198ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSliceSimpleStrideFloat32, StridedSliceSimpleStrideFloat32Test)
1199ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSliceSimpleRangeMaskFloat32, StridedSliceSimpleRangeMaskFloat32Test)
1200ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSliceShrinkAxisMaskFloat32, StridedSliceShrinkAxisMaskFloat32Test)
Finn Williams826a5432020-08-27 16:15:20 +01001201ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSliceShrinkAxisMaskCTSFloat32,
1202 StridedSliceShrinkAxisMaskCTSFloat32Test)
Sadik Armagan5e719842020-08-28 12:17:33 +01001203ARMNN_AUTO_TEST_CASE_WITH_THF(
1204 StridedSliceShrinkAxisMaskBitPosition0Dim3Float32, StridedSliceShrinkAxisMaskBitPosition0Dim3Float32Test)
1205ARMNN_AUTO_TEST_CASE_WITH_THF(
1206 StridedSliceShrinkAxisMaskBitPosition0Float32, StridedSliceShrinkAxisMaskBitPosition0Float32Test)
1207ARMNN_AUTO_TEST_CASE_WITH_THF(
1208 StridedSliceShrinkAxisMaskBitPosition1Float32, StridedSliceShrinkAxisMaskBitPosition1Float32Test)
1209ARMNN_AUTO_TEST_CASE_WITH_THF(
1210 StridedSliceShrinkAxisMaskBitPosition2Float32, StridedSliceShrinkAxisMaskBitPosition2Float32Test)
1211ARMNN_AUTO_TEST_CASE_WITH_THF(
1212 StridedSliceShrinkAxisMaskBitPosition3Float32, StridedSliceShrinkAxisMaskBitPosition3Float32Test)
1213ARMNN_AUTO_TEST_CASE_WITH_THF(
1214 StridedSliceShrinkAxisMaskBitPosition0And1Float32, StridedSliceShrinkAxisMaskBitPosition0And1Float32Test)
1215ARMNN_AUTO_TEST_CASE_WITH_THF(
1216 StridedSliceShrinkAxisMaskBitPosition0And2Float32, StridedSliceShrinkAxisMaskBitPosition0And2Float32Test)
1217ARMNN_AUTO_TEST_CASE_WITH_THF(
1218 StridedSliceShrinkAxisMaskBitPosition0And3Float32, StridedSliceShrinkAxisMaskBitPosition0And3Float32Test)
1219ARMNN_AUTO_TEST_CASE_WITH_THF(
1220 StridedSliceShrinkAxisMaskBitPosition0And1And3Float32, StridedSliceShrinkAxisMaskBitPosition0And1And3Float32Test)
1221ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice3dFloat32, StridedSlice3dFloat32Test)
1222ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice3dReverseFloat32, StridedSlice3dReverseFloat32Test)
1223ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice2dFloat32, StridedSlice2dFloat32Test)
1224ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice2dReverseFloat32, StridedSlice2dReverseFloat32Test)
FinnWilliamsArm1fa19192019-08-02 17:26:31 +01001225
Sadik Armagan5e719842020-08-28 12:17:33 +01001226ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice4dUint8, StridedSlice4dUint8Test)
1227ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice4dReverseUint8, StridedSlice4dReverseUint8Test)
1228ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSliceSimpleStrideUint8, StridedSliceSimpleStrideUint8Test)
1229ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSliceSimpleRangeMaskUint8, StridedSliceSimpleRangeMaskUint8Test)
1230ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSliceShrinkAxisMaskUint8, StridedSliceShrinkAxisMaskUint8Test)
1231ARMNN_AUTO_TEST_CASE_WITH_THF(
1232 StridedSliceShrinkAxisMaskBitPosition0Dim3Uint8, StridedSliceShrinkAxisMaskBitPosition0Dim3Uint8Test)
1233ARMNN_AUTO_TEST_CASE_WITH_THF(
1234 StridedSliceShrinkAxisMaskBitPosition0Uint8, StridedSliceShrinkAxisMaskBitPosition0Uint8Test)
1235ARMNN_AUTO_TEST_CASE_WITH_THF(
1236 StridedSliceShrinkAxisMaskBitPosition1Uint8, StridedSliceShrinkAxisMaskBitPosition1Uint8Test)
1237ARMNN_AUTO_TEST_CASE_WITH_THF(
1238 StridedSliceShrinkAxisMaskBitPosition2Uint8, StridedSliceShrinkAxisMaskBitPosition2Uint8Test)
1239ARMNN_AUTO_TEST_CASE_WITH_THF(
1240 StridedSliceShrinkAxisMaskBitPosition3Uint8, StridedSliceShrinkAxisMaskBitPosition3Uint8Test)
1241ARMNN_AUTO_TEST_CASE_WITH_THF(
1242 StridedSliceShrinkAxisMaskBitPosition0And1Uint8, StridedSliceShrinkAxisMaskBitPosition0And1Uint8Test)
1243ARMNN_AUTO_TEST_CASE_WITH_THF(
1244 StridedSliceShrinkAxisMaskBitPosition0And2Uint8, StridedSliceShrinkAxisMaskBitPosition0And2Uint8Test)
1245ARMNN_AUTO_TEST_CASE_WITH_THF(
1246 StridedSliceShrinkAxisMaskBitPosition0And3Uint8, StridedSliceShrinkAxisMaskBitPosition0And3Uint8Test)
1247ARMNN_AUTO_TEST_CASE_WITH_THF(
1248 StridedSliceShrinkAxisMaskBitPosition0And1And3Uint8, StridedSliceShrinkAxisMaskBitPosition0And1And3Uint8Test)
1249ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice3dUint8, StridedSlice3dUint8Test)
1250ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice3dReverseUint8, StridedSlice3dReverseUint8Test)
1251ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice2dUint8, StridedSlice2dUint8Test)
1252ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice2dReverseUint8, StridedSlice2dReverseUint8Test)
FinnWilliamsArm1fa19192019-08-02 17:26:31 +01001253
Sadik Armaganfabc2892019-05-31 09:05:11 +01001254// Quantize
Finn Williamsec36d3e2020-08-28 13:17:05 +01001255ARMNN_AUTO_TEST_CASE_WITH_THF(QuantizeSimpleUint8, QuantizeSimpleUint8Test)
1256ARMNN_AUTO_TEST_CASE_WITH_THF(QuantizeClampUint8, QuantizeClampUint8Test)
Sadik Armaganfabc2892019-05-31 09:05:11 +01001257
Nikhil Raj9b461482019-07-03 15:58:31 +01001258// PReLU
Finn Williams826a5432020-08-27 16:15:20 +01001259ARMNN_AUTO_TEST_CASE_WITH_THF(PreluFloat32, PreluTest<DataType::Float32>)
1260ARMNN_AUTO_TEST_CASE_WITH_THF(PreluUint8, PreluTest<DataType::QAsymmU8>)
Nikhil Raj9b461482019-07-03 15:58:31 +01001261
Matthew Jackson87f65ea2019-08-01 10:01:34 +01001262// Stack
Sadik Armagan5e719842020-08-28 12:17:33 +01001263ARMNN_AUTO_TEST_CASE_WITH_THF(Stack0Axis, StackAxis0Float32Test)
1264ARMNN_AUTO_TEST_CASE_WITH_THF(StackOutput4DAxis1, StackOutput4DAxis1Float32Test)
1265ARMNN_AUTO_TEST_CASE_WITH_THF(StackOutput4DAxis2, StackOutput4DAxis2Float32Test)
1266ARMNN_AUTO_TEST_CASE_WITH_THF(StackOutput4DAxis3, StackOutput4DAxis3Float32Test)
1267ARMNN_AUTO_TEST_CASE_WITH_THF(StackOutput3DInputs3, StackOutput3DInputs3Float32Test)
1268ARMNN_AUTO_TEST_CASE_WITH_THF(StackOutput5D, StackOutput5DFloat32Test)
Matthew Jackson87f65ea2019-08-01 10:01:34 +01001269
Mike Kellyc9ea45a2020-02-28 18:11:58 +00001270// Transpose
Finn Williamsec36d3e2020-08-28 13:17:05 +01001271ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleTransposeFloat32, SimpleTransposeTest<DataType::Float32>)
1272ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeFloat32ValueSet1Test, TransposeValueSet1Test<DataType::Float32>)
1273ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeFloat32ValueSet2Test, TransposeValueSet2Test<DataType::Float32>)
1274ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeFloat32ValueSet3Test, TransposeValueSet3Test<DataType::Float32>)
1275ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleTransposeQASymms8, SimpleTransposeTest<DataType::QAsymmS8>)
1276ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeQASymmS8ValueSet1Test, TransposeValueSet1Test<DataType::QAsymmS8>)
1277ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeQASymmS8ValueSet2Test, TransposeValueSet2Test<DataType::QAsymmS8>)
1278ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeQASymmS8ValueSet3Test, TransposeValueSet3Test<DataType::QAsymmS8>)
1279ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleTransposeQASymm8, SimpleTransposeTest<DataType::QAsymmU8>)
1280ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeQASymm8ValueSet1Test, TransposeValueSet1Test<DataType::QAsymmU8>)
1281ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeQASymm8ValueSet2Test, TransposeValueSet2Test<DataType::QAsymmU8>)
1282ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeQASymm8ValueSet3Test, TransposeValueSet3Test<DataType::QAsymmU8>)
1283ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleTransposeQSymm16, SimpleTransposeTest<DataType::QSymmS16>)
1284ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeQSymm16ValueSet1Test, TransposeValueSet1Test<DataType::QSymmS16>)
1285ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeQSymm16ValueSet2Test, TransposeValueSet2Test<DataType::QSymmS16>)
1286ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeQSymm16ValueSet3Test, TransposeValueSet3Test<DataType::QSymmS16>)
Mike Kellyc9ea45a2020-02-28 18:11:58 +00001287
Sadik Armagan581742d2019-08-12 14:11:37 +01001288// TransposeConvolution2d
Finn Williamsec36d3e2020-08-28 13:17:05 +01001289ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleTransposeConvolution2dFloatNchw,
1290 SimpleTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1291 true,
1292 DataLayout::NCHW)
1293ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleTransposeConvolution2dFloatNhwc,
1294 SimpleTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1295 true,
1296 DataLayout::NHWC)
1297ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleTransposeConvolution2dUint8Nchw,
1298 SimpleTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1299 true,
1300 DataLayout::NCHW)
1301ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleTransposeConvolution2dUint8Nhwc,
1302 SimpleTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1303 true,
1304 DataLayout::NHWC)
Sadik Armagan581742d2019-08-12 14:11:37 +01001305
Finn Williamsec36d3e2020-08-28 13:17:05 +01001306ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedSimpleTransposeConvolution2dFloatNchw,
1307 SimpleTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1308 false,
1309 DataLayout::NCHW)
1310ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedSimpleTransposeConvolution2dFloatNhwc,
1311 SimpleTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1312 true,
1313 DataLayout::NHWC)
1314ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedSimpleTransposeConvolution2dUint8Nchw,
1315 SimpleTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1316 true,
1317 DataLayout::NCHW)
1318ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedSimpleTransposeConvolution2dUint8Nhwc,
1319 SimpleTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1320 true,
1321 DataLayout::NHWC)
Sadik Armagan581742d2019-08-12 14:11:37 +01001322
Finn Williamsec36d3e2020-08-28 13:17:05 +01001323ARMNN_AUTO_TEST_CASE_WITH_THF(PaddedTransposeConvolution2dFloatNchw,
1324 PaddedTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1325 true,
1326 DataLayout::NCHW)
1327ARMNN_AUTO_TEST_CASE_WITH_THF(PaddedTransposeConvolution2dFloatNhwc,
1328 PaddedTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1329 true,
1330 DataLayout::NHWC)
1331ARMNN_AUTO_TEST_CASE_WITH_THF(PaddedTransposeConvolution2dUint8Nchw,
1332 PaddedTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1333 true,
1334 DataLayout::NCHW)
1335ARMNN_AUTO_TEST_CASE_WITH_THF(PaddedTransposeConvolution2dUint8Nhwc,
1336 PaddedTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1337 true,
1338 DataLayout::NHWC)
Sadik Armagan581742d2019-08-12 14:11:37 +01001339
Finn Williamsec36d3e2020-08-28 13:17:05 +01001340ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedPaddedTransposeConvolution2dFloatNchw,
1341 PaddedTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1342 false,
1343 DataLayout::NCHW)
1344ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedPaddedTransposeConvolution2dFloatNhwc,
1345 PaddedTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1346 true,
1347 DataLayout::NHWC)
1348ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedPaddedTransposeConvolution2dInt8Nchw,
1349 PaddedTransposeConvolution2dTest<DataType::QAsymmS8, DataType::Signed32>,
1350 true,
1351 DataLayout::NCHW)
1352ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedPaddedTransposeConvolution2dInt8Nhwc,
1353 PaddedTransposeConvolution2dTest<DataType::QAsymmS8, DataType::Signed32>,
1354 true,
1355 DataLayout::NHWC)
1356ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedPaddedTransposeConvolution2dUint8Nchw,
1357 PaddedTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1358 true,
1359 DataLayout::NCHW)
1360ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedPaddedTransposeConvolution2dUint8Nhwc,
1361 PaddedTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1362 true,
1363 DataLayout::NHWC)
Sadik Armagan581742d2019-08-12 14:11:37 +01001364
Finn Williamsec36d3e2020-08-28 13:17:05 +01001365ARMNN_AUTO_TEST_CASE_WITH_THF(StridedTransposeConvolution2dFloatNchw,
1366 StridedTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1367 true,
1368 DataLayout::NCHW)
1369ARMNN_AUTO_TEST_CASE_WITH_THF(StridedTransposeConvolution2dFloatNhwc,
1370 StridedTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1371 true,
1372 DataLayout::NHWC)
1373ARMNN_AUTO_TEST_CASE_WITH_THF(StridedTransposeConvolution2dUint8Nchw,
1374 StridedTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1375 true,
1376 DataLayout::NCHW)
1377ARMNN_AUTO_TEST_CASE_WITH_THF(StridedTransposeConvolution2dUint8Nhwc,
1378 StridedTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1379 true,
1380 DataLayout::NHWC)
Sadik Armagan581742d2019-08-12 14:11:37 +01001381
Finn Williamsec36d3e2020-08-28 13:17:05 +01001382ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedStridedTransposeConvolution2dFloatNchw,
1383 StridedTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1384 false,
1385 DataLayout::NCHW)
1386ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedStridedTransposeConvolution2dFloatNhwc,
1387 StridedTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1388 true,
1389 DataLayout::NHWC)
1390ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedStridedTransposeConvolution2dUint8Nchw,
1391 StridedTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1392 true,
1393 DataLayout::NCHW)
1394ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedStridedTransposeConvolution2dUint8Nhwc,
1395 StridedTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1396 true,
1397 DataLayout::NHWC)
Sadik Armagan581742d2019-08-12 14:11:37 +01001398
Finn Williamsec36d3e2020-08-28 13:17:05 +01001399ARMNN_AUTO_TEST_CASE_WITH_THF(MultiChannelTransposeConvolution2dFloatNchw,
1400 MultiChannelTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1401 DataLayout::NCHW)
1402ARMNN_AUTO_TEST_CASE_WITH_THF(MultiChannelTransposeConvolution2dFloatNhwc,
1403 MultiChannelTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1404 DataLayout::NHWC)
1405ARMNN_AUTO_TEST_CASE_WITH_THF(MultiChannelTransposeConvolution2dUint8Nchw,
1406 MultiChannelTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1407 DataLayout::NCHW)
1408ARMNN_AUTO_TEST_CASE_WITH_THF(MultiChannelTransposeConvolution2dUint8Nhwc,
1409 MultiChannelTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1410 DataLayout::NHWC)
Aron Virginas-Tar7599aa42019-08-20 15:30:04 +01001411
Aron Virginas-Tar914e4db2019-09-09 13:36:45 +01001412// Abs
Keith Davis33a626f2020-08-27 15:38:12 +01001413ARMNN_AUTO_TEST_CASE_WITH_THF(Abs2d, Abs2dTest<DataType::Float32>)
1414ARMNN_AUTO_TEST_CASE_WITH_THF(Abs3d, Abs3dTest<DataType::Float32>)
1415ARMNN_AUTO_TEST_CASE_WITH_THF(AbsZero, AbsZeroTest<DataType::Float32>)
Aron Virginas-Tar914e4db2019-09-09 13:36:45 +01001416
Keith Davis33a626f2020-08-27 15:38:12 +01001417ARMNN_AUTO_TEST_CASE_WITH_THF(Abs2dSigned32, Abs2dTest<DataType::Signed32>)
1418ARMNN_AUTO_TEST_CASE_WITH_THF(Abs3dSigned32, Abs3dTest<DataType::Signed32>)
1419ARMNN_AUTO_TEST_CASE_WITH_THF(AbsZeroSigned32, AbsZeroTest<DataType::Signed32>)
Kevin Mayec52c3a2020-04-24 09:42:31 +01001420
Aron Virginas-Tar0dd3b432019-09-10 13:55:09 +01001421// Rsqrt
Keith Davis33a626f2020-08-27 15:38:12 +01001422ARMNN_AUTO_TEST_CASE_WITH_THF(Rsqrt2d, Rsqrt2dTest<DataType::Float32>)
1423ARMNN_AUTO_TEST_CASE_WITH_THF(Rsqrt3d, Rsqrt3dTest<DataType::Float32>)
1424ARMNN_AUTO_TEST_CASE_WITH_THF(RsqrtZero, RsqrtZeroTest<DataType::Float32>)
1425ARMNN_AUTO_TEST_CASE_WITH_THF(RsqrtNegative, RsqrtNegativeTest<DataType::Float32>)
Aron Virginas-Tar0dd3b432019-09-10 13:55:09 +01001426
Teresa Charlin03027232022-05-09 17:27:08 +01001427// Sqrt
1428ARMNN_AUTO_TEST_CASE_WITH_THF(Sqrt2d, Sqrt2dTest<DataType::Float32>)
1429ARMNN_AUTO_TEST_CASE_WITH_THF(Sqrt3d, Sqrt3dTest<DataType::Float32>)
1430ARMNN_AUTO_TEST_CASE_WITH_THF(SqrtZero, SqrtZeroTest<DataType::Float32>)
1431ARMNN_AUTO_TEST_CASE_WITH_THF(SqrtNegative, SqrtNegativeTest<DataType::Float32>)
1432
James Conroyd47a0642019-09-17 14:22:06 +01001433// ArgMinMax
Sadik Armagan43528ac2020-08-27 09:06:40 +01001434ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinFloat32, ArgMinSimpleTest<DataType::Float32>)
1435ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxFloat32, ArgMaxSimpleTest<DataType::Float32>)
1436ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinChannel, ArgMinChannelTest<DataType::Float32>)
1437ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxChannel, ArgMaxChannelTest<DataType::Float32>)
1438ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxHeight, ArgMaxHeightTest<DataType::Float32>)
1439ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinWidth, ArgMinWidthTest<DataType::Float32>)
James Conroyd47a0642019-09-17 14:22:06 +01001440
Sadik Armagan43528ac2020-08-27 09:06:40 +01001441ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinQAsymmS8, ArgMinSimpleTest<DataType::QAsymmS8>)
1442ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxQAsymmS8, ArgMaxSimpleTest<DataType::QAsymmS8>)
1443ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinChannelQAsymmS8, ArgMinChannelTest<DataType::QAsymmS8>)
1444ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxChannelQAsymmS8, ArgMaxChannelTest<DataType::QAsymmS8>)
1445ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxHeightQAsymmS8, ArgMaxHeightTest<DataType::QAsymmS8>)
1446ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinWidthQAsymmS8, ArgMinWidthTest<DataType::QAsymmS8>)
Sadik Armagan303980c2020-04-17 12:45:14 +01001447
Sadik Armagan43528ac2020-08-27 09:06:40 +01001448ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinQAsymm8, ArgMinSimpleTest<DataType::QAsymmU8>)
1449ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxQAsymm8, ArgMaxSimpleTest<DataType::QAsymmU8>)
1450ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinChannelQAsymm8, ArgMinChannelTest<DataType::QAsymmU8>)
1451ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxChannelQAsymm8, ArgMaxChannelTest<DataType::QAsymmU8>)
1452ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxHeightQAsymm8, ArgMaxHeightTest<DataType::QAsymmU8>)
1453ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinWidthQAsymm8, ArgMinWidthTest<DataType::QAsymmU8>)
Francis Murtagh52ec3462019-11-19 12:24:19 +00001454
Sadik Armaganac472102020-03-24 09:54:36 +00001455// Neg
Keith Davis33a626f2020-08-27 15:38:12 +01001456ARMNN_AUTO_TEST_CASE_WITH_THF(Neg2d, Neg2dTest<DataType::Float32>)
1457ARMNN_AUTO_TEST_CASE_WITH_THF(Neg3d, Neg3dTest<DataType::Float32>)
1458ARMNN_AUTO_TEST_CASE_WITH_THF(NegZero, NegZeroTest<DataType::Float32>)
1459ARMNN_AUTO_TEST_CASE_WITH_THF(NegNegative, NegNegativeTest<DataType::Float32>)
Sadik Armaganac472102020-03-24 09:54:36 +00001460
Sadik Armagan9fabf432020-05-27 13:40:58 +01001461// Exp
Keith Davis33a626f2020-08-27 15:38:12 +01001462ARMNN_AUTO_TEST_CASE_WITH_THF(Exp2d, Exp2dTest<DataType::Float32>)
Teresa Charlin50de4fa2021-05-31 18:47:33 +01001463ARMNN_AUTO_TEST_CASE_WITH_THF(Exp3d, Exp3dTest<DataType::Float32>)
Keith Davis33a626f2020-08-27 15:38:12 +01001464ARMNN_AUTO_TEST_CASE_WITH_THF(ExpZero, ExpZeroTest<DataType::Float32>)
1465ARMNN_AUTO_TEST_CASE_WITH_THF(ExpNegative, ExpNegativeTest<DataType::Float32>)
Sadik Armagan9fabf432020-05-27 13:40:58 +01001466
Teresa Charlin50de4fa2021-05-31 18:47:33 +01001467// Log
1468ARMNN_AUTO_TEST_CASE_WITH_THF(Log2d, Log2dTest<DataType::Float32>)
1469ARMNN_AUTO_TEST_CASE_WITH_THF(Log3d, Log3dTest<DataType::Float32>)
1470ARMNN_AUTO_TEST_CASE_WITH_THF(LogZero, LogZeroTest<DataType::Float32>)
1471ARMNN_AUTO_TEST_CASE_WITH_THF(LogNegative, LogNegativeTest<DataType::Float32>)
1472
1473// Sin
1474ARMNN_AUTO_TEST_CASE_WITH_THF(Sin2d, Sin2dTest<DataType::Float32>)
1475ARMNN_AUTO_TEST_CASE_WITH_THF(Sin3d, Sin3dTest<DataType::Float32>)
1476ARMNN_AUTO_TEST_CASE_WITH_THF(SinZero, SinZeroTest<DataType::Float32>)
1477ARMNN_AUTO_TEST_CASE_WITH_THF(SinNegative, SinNegativeTest<DataType::Float32>)
1478
Sadik Armagana792a052020-06-23 16:22:23 +01001479// Fill
Finn Williams87020072020-08-26 16:19:15 +01001480ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleFill, SimpleFillTest<DataType::Float32>)
1481ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleFillF16, SimpleFillTest<DataType::Float16>)
1482ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleFillS32, SimpleFillTest<DataType::Signed32>)
Sadik Armagana792a052020-06-23 16:22:23 +01001483
James Conroy177df1e2020-11-13 10:18:51 +00001484// Logical
1485ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalNot, LogicalNotTest)
1486ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalNotInt, LogicalNotIntTest)
1487
1488ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalAnd, LogicalAndTest)
1489ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalAndInt, LogicalAndIntTest)
1490ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalAndBroadcast1, LogicalAndBroadcast1Test)
1491ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalAndBroadcast2, LogicalAndBroadcast2Test)
1492ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalAndBroadcast3, LogicalAndBroadcast3Test)
1493
1494ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalOr, LogicalOrTest)
1495ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalOrInt, LogicalOrIntTest)
1496ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalOrBroadcast1, LogicalOrBroadcast1Test)
1497ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalOrBroadcast2, LogicalOrBroadcast2Test)
1498ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalOrBroadcast3, LogicalOrBroadcast3Test)
1499
Sadik Armagana2747482021-02-09 10:28:54 +00001500// ReduceSum
Colm Donelanade8c1d2021-03-24 14:51:31 +00001501// Moved to NeonLayerTests_NDK_Bug.cpp
1502//ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceSumFloat32, ReduceSumSimpleTest<DataType::Float32>)
Sadik Armagana2747482021-02-09 10:28:54 +00001503ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceSumSingleAxisFloat32_1, ReduceSumSingleAxisTest1<DataType::Float32>)
1504ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceSumSingleAxisFloat32_2, ReduceSumSingleAxisTest2<DataType::Float32>)
Colm Donelanade8c1d2021-03-24 14:51:31 +00001505// Moved to NeonLayerTests_NDK_Bug.cpp
1506//ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceSumSingleAxisFloat32_3, ReduceSumSingleAxisTest3<DataType::Float32>)
Sadik Armagana2747482021-02-09 10:28:54 +00001507
Teresa Charlin4e3e8312021-08-05 12:34:37 +01001508// ReduceProd
1509ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceProdSingleAxisFloat32_1, ReduceProdSingleAxisTest1<DataType::Float32>)
1510ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceProdSingleAxisFloat32_2, ReduceProdSingleAxisTest2<DataType::Float32>)
1511
Sadik Armagana2747482021-02-09 10:28:54 +00001512// ReduceMax
1513ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceMaxFloat32, ReduceMaxSimpleTest<DataType::Float32>)
1514ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceMaxNegativeAxisFloat32, ReduceMaxNegativeAxisTest<DataType::Float32>)
1515ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceMax2Float32, ReduceMaxSimpleTest2<DataType::Float32>)
1516
1517// ReduceMin
1518ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceMinFloat32, ReduceMinSimpleTest<DataType::Float32>)
1519ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceMinNegativeAxisFloat32, ReduceMinNegativeAxisTest<DataType::Float32>)
1520
Sadik Armagan48f011e2021-04-21 10:50:34 +01001521// Cast
1522ARMNN_AUTO_TEST_CASE_WITH_THF(CastInt32ToFloat, CastInt32ToFloat2dTest)
1523ARMNN_AUTO_TEST_CASE_WITH_THF(CastInt8AsymmToFloat, CastInt8AsymmToFloat2dTest)
1524ARMNN_AUTO_TEST_CASE_WITH_THF(CastUIntToFloat, CastUInt8ToFloat2dTest)
1525ARMNN_AUTO_TEST_CASE_WITH_THF(CastFloatToIn8, CastFloat32ToInt82dTest)
1526ARMNN_AUTO_TEST_CASE_WITH_THF(CastFloatToUInt8, CastFloat32ToUInt82dTest)
1527
Teresa Charline89dd692021-09-01 16:30:34 +01001528// ChannelShuffle
1529ARMNN_AUTO_TEST_CASE_WITH_THF(ChannelShuffle4DFloat32, ChannelShuffle4DTest<DataType::Float32>)
1530ARMNN_AUTO_TEST_CASE_WITH_THF(ChannelShuffle4DQAsymmU8, ChannelShuffle4DTest<DataType::QAsymmU8>)
Teresa Charlineb852bb2021-09-30 12:44:33 +01001531ARMNN_AUTO_TEST_CASE_WITH_THF(ChannelShuffle4DQAsymmS8, ChannelShuffle4DTest<DataType::QAsymmS8>)
Sadik Armagan48f011e2021-04-21 10:50:34 +01001532
Matteo Martincighdb16dd32019-08-27 16:41:11 +01001533#if defined(ARMNNREF_ENABLED)
Matteo Martincighe67edb22019-08-14 14:05:46 +01001534
1535// The ARMNN_COMPARE_REF_AUTO_TEST_CASE and the ARMNN_COMPARE_REF_FIXTURE_TEST_CASE test units are not available
1536// if the reference backend is not built
1537
telsoa014fcda012018-03-09 14:13:49 +00001538// ============================================================================
1539// COMPARE tests
1540
Keith Davisf500d6c2020-08-31 08:32:55 +01001541ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareConv2dWithReference, CompareConvolution2dTest)
telsoa014fcda012018-03-09 14:13:49 +00001542
Keith Davisf500d6c2020-08-31 08:32:55 +01001543ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareDepthwiseConv2dWithReferenceFloat32,
Nattapat Chaimanowong649dd952019-01-22 16:10:44 +00001544 CompareDepthwiseConvolution2dFloatTest,
Sadik Armagan581742d2019-08-12 14:11:37 +01001545 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +01001546ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareDepthwiseConv2dWithReferenceUint8,
Nattapat Chaimanowong649dd952019-01-22 16:10:44 +00001547 CompareDepthwiseConvolution2dUint8Test,
Sadik Armagan581742d2019-08-12 14:11:37 +01001548 DataLayout::NCHW)
jimfly017af00da2018-10-31 14:43:53 +00001549
Keith Davisf500d6c2020-08-31 08:32:55 +01001550ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareDepthwiseConv2dWithReferenceFloat32Nhwc,
Nattapat Chaimanowong649dd952019-01-22 16:10:44 +00001551 CompareDepthwiseConvolution2dFloatTest,
Sadik Armagan581742d2019-08-12 14:11:37 +01001552 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +01001553ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareDepthwiseConv2dWithReferenceUint8Nhwc,
Nattapat Chaimanowong649dd952019-01-22 16:10:44 +00001554 CompareDepthwiseConvolution2dUint8Test,
Sadik Armagan581742d2019-08-12 14:11:37 +01001555 DataLayout::NHWC)
telsoa014fcda012018-03-09 14:13:49 +00001556
Finn Williams826a5432020-08-27 16:15:20 +01001557ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareNormalizationWithinWithReference, CompareNormalizationTest,
1558 NormalizationAlgorithmChannel::Within,
1559 NormalizationAlgorithmMethod::LocalBrightness)
1560ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareNormalizationAcrossWithReference, CompareNormalizationTest,
1561 NormalizationAlgorithmChannel::Across,
1562 NormalizationAlgorithmMethod::LocalBrightness)
telsoa014fcda012018-03-09 14:13:49 +00001563
Finn Williams826a5432020-08-27 16:15:20 +01001564ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareMaxPooling2dWithReference, ComparePooling2dTest, PoolingAlgorithm::Max)
1565ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareMaxPooling2dWithReferenceUint8, ComparePooling2dUint8Test,
1566 PoolingAlgorithm::Max)
1567ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareAveragePooling2dWithReference, ComparePooling2dTest,
1568 PoolingAlgorithm::Average)
1569ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareAveragePooling2dWithReferenceUint8, ComparePooling2dUint8Test,
1570 PoolingAlgorithm::Average)
1571ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareL2Pooling2dWithReference, ComparePooling2dTest, PoolingAlgorithm::L2)
1572ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(UNSUPPORTED_CompareL2Pooling2dWithReferenceUint8, ComparePooling2dUint8Test,
1573 PoolingAlgorithm::L2)
telsoa014fcda012018-03-09 14:13:49 +00001574
Colm Donelanade8c1d2021-03-24 14:51:31 +00001575// Moved to NeonLayerTests_NDK_Bug.cpp
1576//ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareSoftmaxBeta1WithReference, CompareSoftmaxTest, 1.0f)
1577//ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareSoftmaxBeta2WithReference, CompareSoftmaxTest, 2.0f)
telsoa014fcda012018-03-09 14:13:49 +00001578
Sadik Armagan56785c72020-08-27 12:57:20 +01001579ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareSoftmaxUint8Beta1WithReference, CompareSoftmaxUint8Test, 1.0f)
1580ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareSoftmaxUint8Beta2WithReference, CompareSoftmaxUint8Test, 2.0f)
telsoa014fcda012018-03-09 14:13:49 +00001581
Keith Davis33a626f2020-08-27 15:38:12 +01001582ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareAddition, CompareAdditionTest)
telsoa014fcda012018-03-09 14:13:49 +00001583
Keith Davis33a626f2020-08-27 15:38:12 +01001584ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareMultiplicationWithReference, CompareMultiplicationTest)
telsoa014fcda012018-03-09 14:13:49 +00001585
Keith Davis33a626f2020-08-27 15:38:12 +01001586ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareBatchNorm, CompareBatchNormTest)
telsoa014fcda012018-03-09 14:13:49 +00001587
Keith Davis33a626f2020-08-27 15:38:12 +01001588ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareReLu1, CompareBoundedReLuTest, 1.0f, -1.0f)
1589ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareReLu6, CompareBoundedReLuTest, 6.0f, 0.0f)
telsoa014fcda012018-03-09 14:13:49 +00001590
1591// ============================================================================
1592// FIXTURE tests
1593
Keith Davis33a626f2020-08-27 15:38:12 +01001594ARMNN_COMPARE_REF_FIXTURE_TEST_CASE_WITH_THF(CompareSigmoidActivationWithReference, ActivationFixture,
Sadik Armagan581742d2019-08-12 14:11:37 +01001595 CompareActivationTest, ActivationFunction::Sigmoid, 5u)
telsoa014fcda012018-03-09 14:13:49 +00001596
Keith Davis33a626f2020-08-27 15:38:12 +01001597ARMNN_COMPARE_REF_FIXTURE_TEST_CASE_WITH_THF(CompareTanhActivationWithReference, ActivationFixture,
Sadik Armagan581742d2019-08-12 14:11:37 +01001598 CompareActivationTest, ActivationFunction::TanH, 5u)
telsoa014fcda012018-03-09 14:13:49 +00001599
Keith Davis33a626f2020-08-27 15:38:12 +01001600ARMNN_COMPARE_REF_FIXTURE_TEST_CASE_WITH_THF(CompareLinearActivationWithReference, ActivationFixture,
Sadik Armagan581742d2019-08-12 14:11:37 +01001601 CompareActivationTest, ActivationFunction::Linear, 5u)
telsoa014fcda012018-03-09 14:13:49 +00001602
Keith Davis33a626f2020-08-27 15:38:12 +01001603ARMNN_COMPARE_REF_FIXTURE_TEST_CASE_WITH_THF(CompareReLuActivationWithReference, ActivationFixture,
Sadik Armagan581742d2019-08-12 14:11:37 +01001604 CompareActivationTest, ActivationFunction::ReLu, 5u)
telsoa014fcda012018-03-09 14:13:49 +00001605
Keith Davis33a626f2020-08-27 15:38:12 +01001606ARMNN_COMPARE_REF_FIXTURE_TEST_CASE_WITH_THF(CompareBoundedReLuActivationWithReference, ActivationFixture,
Sadik Armagan581742d2019-08-12 14:11:37 +01001607 CompareActivationTest, ActivationFunction::BoundedReLu, 5u)
Keith Davis33a626f2020-08-27 15:38:12 +01001608ARMNN_COMPARE_REF_FIXTURE_TEST_CASE_WITH_THF(CompareBoundedReLuActivationWithReferenceUint8, ActivationFixture,
Sadik Armagan581742d2019-08-12 14:11:37 +01001609 CompareActivationUint8Test, ActivationFunction::BoundedReLu)
telsoa014fcda012018-03-09 14:13:49 +00001610
Keith Davis33a626f2020-08-27 15:38:12 +01001611ARMNN_COMPARE_REF_FIXTURE_TEST_CASE_WITH_THF(CompareSoftReLuActivationWithReference, ActivationFixture,
Sadik Armagan581742d2019-08-12 14:11:37 +01001612 CompareActivationTest, ActivationFunction::SoftReLu, 1u)
telsoa014fcda012018-03-09 14:13:49 +00001613
Keith Davis33a626f2020-08-27 15:38:12 +01001614ARMNN_COMPARE_REF_FIXTURE_TEST_CASE_WITH_THF(CompareLeakyReLuActivationWithReference, ActivationFixture,
Sadik Armagan581742d2019-08-12 14:11:37 +01001615 CompareActivationTest, ActivationFunction::LeakyReLu, 5u)
telsoa014fcda012018-03-09 14:13:49 +00001616
Keith Davis33a626f2020-08-27 15:38:12 +01001617ARMNN_COMPARE_REF_FIXTURE_TEST_CASE_WITH_THF(CompareAbsActivationWithReference, ActivationFixture,
Sadik Armagan581742d2019-08-12 14:11:37 +01001618 CompareActivationTest, ActivationFunction::Abs, 5u)
telsoa014fcda012018-03-09 14:13:49 +00001619
Keith Davis33a626f2020-08-27 15:38:12 +01001620ARMNN_COMPARE_REF_FIXTURE_TEST_CASE_WITH_THF(CompareSqrtActivationWithReference, PositiveActivationFixture,
Sadik Armagan581742d2019-08-12 14:11:37 +01001621 CompareActivationTest, ActivationFunction::Sqrt, 5u)
telsoa014fcda012018-03-09 14:13:49 +00001622
Keith Davis33a626f2020-08-27 15:38:12 +01001623ARMNN_COMPARE_REF_FIXTURE_TEST_CASE_WITH_THF(CompareSquareActivationWithReference, ActivationFixture,
Sadik Armagan581742d2019-08-12 14:11:37 +01001624 CompareActivationTest, ActivationFunction::Square, 5u)
Matteo Martincighe67edb22019-08-14 14:05:46 +01001625
Keith Davis33a626f2020-08-27 15:38:12 +01001626ARMNN_COMPARE_REF_FIXTURE_TEST_CASE_WITH_THF(CompareEluActivationWithReference, ActivationFixture,
Teresa Charline32c8442020-03-05 14:41:50 +00001627 CompareActivationTest, ActivationFunction::Elu, 5u)
1628
Matteo Martincighe67edb22019-08-14 14:05:46 +01001629#endif
1630
Sadik Armagan1625efc2021-06-10 18:24:34 +01001631}