blob: d351870645c2412c16986b335a6c0c8c1f918645 [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
Aron Virginas-Tarc9cc8042018-11-01 16:15:57 +00008#include <test/TensorHelpers.hpp>
9#include <test/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 <reference/RefWorkloadFactory.hpp>
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +010015
Aron Virginas-Tarc9cc8042018-11-01 16:15:57 +000016#include <backendsCommon/test/ActivationFixture.hpp>
17#include <backendsCommon/test/LayerTests.hpp>
telsoa014fcda012018-03-09 14:13:49 +000018
arovir0143095f32018-10-09 18:04:24 +010019#include <boost/test/unit_test.hpp>
telsoa014fcda012018-03-09 14:13:49 +000020
21BOOST_AUTO_TEST_SUITE(Compute_ArmComputeNeon)
Sadik Armagan581742d2019-08-12 14:11:37 +010022
23using namespace armnn;
24
25using FactoryType = NeonWorkloadFactory;
telsoa014fcda012018-03-09 14:13:49 +000026
27// ============================================================================
28// UNIT tests
29
Mike Kelly56858022020-01-27 12:14:47 +000030// BatchToSpace
Keith Davis33a626f2020-08-27 15:38:12 +010031ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcFloat321, BatchToSpaceNdNhwcTest1<DataType::Float32>)
32ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcFloat322, BatchToSpaceNdNhwcTest2<DataType::Float32>)
33ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcFloat323, BatchToSpaceNdNhwcTest3<DataType::Float32>)
Mike Kelly56858022020-01-27 12:14:47 +000034
Keith Davis33a626f2020-08-27 15:38:12 +010035ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwFloat321, BatchToSpaceNdNchwTest1<DataType::Float32>)
36ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwFloat322, BatchToSpaceNdNchwTest2<DataType::Float32>)
37ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwFloat323, BatchToSpaceNdNchwTest3<DataType::Float32>)
Mike Kelly56858022020-01-27 12:14:47 +000038
Keith Davis33a626f2020-08-27 15:38:12 +010039ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcInt1, BatchToSpaceNdNhwcTest1<DataType::QAsymmS8>)
40ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcInt2, BatchToSpaceNdNhwcTest2<DataType::QAsymmS8>)
41ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcInt3, BatchToSpaceNdNhwcTest3<DataType::QAsymmS8>)
Sadik Armagan303980c2020-04-17 12:45:14 +010042
Keith Davis33a626f2020-08-27 15:38:12 +010043ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwInt1, BatchToSpaceNdNchwTest1<DataType::QAsymmS8>)
44ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwInt2, BatchToSpaceNdNchwTest2<DataType::QAsymmS8>)
45ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwInt3, BatchToSpaceNdNchwTest3<DataType::QAsymmS8>)
Sadik Armagan303980c2020-04-17 12:45:14 +010046
Keith Davis33a626f2020-08-27 15:38:12 +010047ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcUint1, BatchToSpaceNdNhwcTest1<DataType::QAsymmU8>)
48ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcUint2, BatchToSpaceNdNhwcTest2<DataType::QAsymmU8>)
49ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcUint3, BatchToSpaceNdNhwcTest3<DataType::QAsymmU8>)
Mike Kelly56858022020-01-27 12:14:47 +000050
Keith Davis33a626f2020-08-27 15:38:12 +010051ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwUint1, BatchToSpaceNdNchwTest1<DataType::QAsymmU8>)
52ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwUint2, BatchToSpaceNdNchwTest2<DataType::QAsymmU8>)
53ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwUint3, BatchToSpaceNdNchwTest3<DataType::QAsymmU8>)
Mike Kelly56858022020-01-27 12:14:47 +000054
telsoa014fcda012018-03-09 14:13:49 +000055// Convolution
Keith Davisf500d6c2020-08-31 08:32:55 +010056ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution1d, Convolution1dTest, true)
telsoa014fcda012018-03-09 14:13:49 +000057
Keith Davisf500d6c2020-08-31 08:32:55 +010058ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution2d, SimpleConvolution2d3x5Test, true, DataLayout::NCHW)
59ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution2dNhwc, SimpleConvolution2d3x5Test, true, DataLayout::NHWC)
60ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution2d3x3Uint8, SimpleConvolution2d3x3Uint8Test, true, DataLayout::NCHW)
61ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution2d3x3Uint8Nhwc, SimpleConvolution2d3x3Uint8Test, true, DataLayout::NHWC)
62ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedConvolution2d, SimpleConvolution2d3x5Test, false, DataLayout::NCHW)
63ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedConvolution2dNhwc, SimpleConvolution2d3x5Test, false, DataLayout::NHWC)
64ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedConvolution2dStride2x2Nhwc,
Sadik Armagan581742d2019-08-12 14:11:37 +010065 SimpleConvolution2d3x3Stride2x2Test, false, DataLayout::NHWC)
jimfly010a088a62018-10-25 17:05:05 +010066
Keith Davisf500d6c2020-08-31 08:32:55 +010067ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedConvolution2dSquare, SimpleConvolution2d3x3Test, false, DataLayout::NCHW)
68ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution2dAsymmetricPadding,
69 Convolution2dAsymmetricPaddingTest,
70 DataLayout::NCHW)
narpra015f703182018-10-26 16:24:58 +010071
Keith Davisf500d6c2020-08-31 08:32:55 +010072ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedConvolution2dSquareNhwc, SimpleConvolution2d3x3Test, false, DataLayout::NHWC)
73ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution2dAsymmetricPaddingNhwc,
narpra015f703182018-10-26 16:24:58 +010074 Convolution2dAsymmetricPaddingTest,
Sadik Armagan581742d2019-08-12 14:11:37 +010075 DataLayout::NHWC)
telsoa014fcda012018-03-09 14:13:49 +000076
Keith Davisf500d6c2020-08-31 08:32:55 +010077ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution2dSquareNhwc, SimpleConvolution2d3x3NhwcTest, false)
telsoa014fcda012018-03-09 14:13:49 +000078
Keith Davisf500d6c2020-08-31 08:32:55 +010079ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d3x3Dilation3x3,
Sadik Armagan581742d2019-08-12 14:11:37 +010080 Convolution2d3x3Dilation3x3Test<DataType::Float32, DataType::Float32>,
Jan Eilers59c66702019-07-11 10:22:36 +010081 false,
Sadik Armagan581742d2019-08-12 14:11:37 +010082 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +010083ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d3x3Dilation3x3Nhwc,
Sadik Armagan581742d2019-08-12 14:11:37 +010084 Convolution2d3x3Dilation3x3Test<DataType::Float32, DataType::Float32>,
Jan Eilers59c66702019-07-11 10:22:36 +010085 false,
Sadik Armagan581742d2019-08-12 14:11:37 +010086 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +010087ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d3x3Dilation3x3Int8,
Sadik Armagan303980c2020-04-17 12:45:14 +010088 Convolution2d3x3Dilation3x3Test<DataType::QAsymmS8, DataType::Signed32>,
89 false,
90 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +010091ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d3x3Dilation3x3NhwcInt8,
Sadik Armagan303980c2020-04-17 12:45:14 +010092 Convolution2d3x3Dilation3x3Test<DataType::QAsymmS8, DataType::Signed32>,
93 false,
94 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +010095ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d3x3Dilation3x3Uint8,
Derek Lambertif90c56d2020-01-10 17:14:08 +000096 Convolution2d3x3Dilation3x3Test<DataType::QAsymmU8, DataType::Signed32>,
Jan Eilers59c66702019-07-11 10:22:36 +010097 false,
Sadik Armagan581742d2019-08-12 14:11:37 +010098 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +010099ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d3x3Dilation3x3NhwcUint8,
Derek Lambertif90c56d2020-01-10 17:14:08 +0000100 Convolution2d3x3Dilation3x3Test<DataType::QAsymmU8, DataType::Signed32>,
Jan Eilers59c66702019-07-11 10:22:36 +0100101 false,
Sadik Armagan581742d2019-08-12 14:11:37 +0100102 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100103ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x3x3Dilation3x3,
Sadik Armagan581742d2019-08-12 14:11:37 +0100104 Convolution2d2x3x3Dilation3x3Test<DataType::Float32, DataType::Float32>,
Jan Eilers59c66702019-07-11 10:22:36 +0100105 false,
Sadik Armagan581742d2019-08-12 14:11:37 +0100106 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100107ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x3x3Dilation3x3Nhwc,
Sadik Armagan581742d2019-08-12 14:11:37 +0100108 Convolution2d2x3x3Dilation3x3Test<DataType::Float32, DataType::Float32>,
Jan Eilers59c66702019-07-11 10:22:36 +0100109 false,
Sadik Armagan581742d2019-08-12 14:11:37 +0100110 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100111ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x3x3Dilation3x3Int8,
Sadik Armagan303980c2020-04-17 12:45:14 +0100112 Convolution2d2x3x3Dilation3x3Test<DataType::QAsymmS8, DataType::Signed32>,
113 false,
114 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100115ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x3x3Dilation3x3NhwcInt8,
Sadik Armagan303980c2020-04-17 12:45:14 +0100116 Convolution2d2x3x3Dilation3x3Test<DataType::QAsymmS8, DataType::Signed32>,
117 false,
118 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100119ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x3x3Dilation3x3Uint8,
Derek Lambertif90c56d2020-01-10 17:14:08 +0000120 Convolution2d2x3x3Dilation3x3Test<DataType::QAsymmU8, DataType::Signed32>,
Jan Eilers59c66702019-07-11 10:22:36 +0100121 false,
Sadik Armagan581742d2019-08-12 14:11:37 +0100122 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100123ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x3x3Dilation3x3NhwcUint8,
Derek Lambertif90c56d2020-01-10 17:14:08 +0000124 Convolution2d2x3x3Dilation3x3Test<DataType::QAsymmU8, DataType::Signed32>,
Jan Eilers59c66702019-07-11 10:22:36 +0100125 false,
Sadik Armagan581742d2019-08-12 14:11:37 +0100126 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100127ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x2Dilation2x2Padding2x2Stride3x3,
Jan Eilers59c66702019-07-11 10:22:36 +0100128 Convolution2d2x2Dilation2x2Padding2x2Stride3x3Test
Sadik Armagan581742d2019-08-12 14:11:37 +0100129 <DataType::Float32, DataType::Float32>,
Jan Eilers59c66702019-07-11 10:22:36 +0100130 false,
Sadik Armagan581742d2019-08-12 14:11:37 +0100131 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100132ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x2Dilation2x2Padding2x2Stride3x3Nhwc,
Jan Eilers59c66702019-07-11 10:22:36 +0100133 Convolution2d2x2Dilation2x2Padding2x2Stride3x3Test
Sadik Armagan581742d2019-08-12 14:11:37 +0100134 <DataType::Float32, DataType::Float32>,
Jan Eilers59c66702019-07-11 10:22:36 +0100135 false,
Sadik Armagan581742d2019-08-12 14:11:37 +0100136 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100137ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x2Dilation2x2Padding2x2Stride3x3Int8,
Sadik Armagan303980c2020-04-17 12:45:14 +0100138 Convolution2d2x2Dilation2x2Padding2x2Stride3x3Test
139 <DataType::QAsymmS8, DataType::Signed32>,
140 false,
141 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100142ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x2Dilation2x2Padding2x2Stride3x3NhwcInt8,
Sadik Armagan303980c2020-04-17 12:45:14 +0100143 Convolution2d2x2Dilation2x2Padding2x2Stride3x3Test
144 <DataType::QAsymmS8, DataType::Signed32>,
145 false,
146 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100147ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x2Dilation2x2Padding2x2Stride3x3Uint8,
Jan Eilers59c66702019-07-11 10:22:36 +0100148 Convolution2d2x2Dilation2x2Padding2x2Stride3x3Test
Derek Lambertif90c56d2020-01-10 17:14:08 +0000149 <DataType::QAsymmU8, DataType::Signed32>,
Jan Eilers59c66702019-07-11 10:22:36 +0100150 false,
Sadik Armagan581742d2019-08-12 14:11:37 +0100151 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100152ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x2Dilation2x2Padding2x2Stride3x3NhwcUint8,
Jan Eilers59c66702019-07-11 10:22:36 +0100153 Convolution2d2x2Dilation2x2Padding2x2Stride3x3Test
Derek Lambertif90c56d2020-01-10 17:14:08 +0000154 <DataType::QAsymmU8, DataType::Signed32>,
Jan Eilers59c66702019-07-11 10:22:36 +0100155 false,
Sadik Armagan581742d2019-08-12 14:11:37 +0100156 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100157ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dMult4,
Kevin May665a964a2019-08-21 16:53:50 +0100158 DepthwiseConvolution2dMult4Test<armnn::DataType::Float32, armnn::DataType::Float32>,
159 false,
160 armnn::DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100161ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dMult2,
Kevin May665a964a2019-08-21 16:53:50 +0100162 DepthwiseConvolution2dMult2Test<armnn::DataType::Float32, armnn::DataType::Float32>,
163 false,
164 armnn::DataLayout::NCHW)
Jan Eilers59c66702019-07-11 10:22:36 +0100165
Keith Davisf500d6c2020-08-31 08:32:55 +0100166ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2dPerAxisQuantTestNchw, Convolution2dPerAxisQuantTest, DataLayout::NCHW);
167ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2dPerAxisQuantTestNhwc, Convolution2dPerAxisQuantTest, DataLayout::NHWC);
Aron Virginas-Tar21fc28b2019-11-26 14:04:54 +0000168
Aron Virginas-Tar2f00b742019-09-30 13:28:08 +0100169// DepthToSpace
170ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwFloat32_1, DepthToSpaceTest1<DataType::Float32>, DataLayout::NCHW);
171ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwFloat32_2, DepthToSpaceTest2<DataType::Float32>, DataLayout::NCHW);
172ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwFloat32_3, DepthToSpaceTest3<DataType::Float32>, DataLayout::NCHW);
173ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwFloat32_4, DepthToSpaceTest4<DataType::Float32>, DataLayout::NCHW);
174
Aron Virginas-Tar99836d32019-09-30 16:34:31 +0100175ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwFloat16_1, DepthToSpaceTest1<DataType::Float16>, DataLayout::NCHW);
176ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwFloat16_2, DepthToSpaceTest2<DataType::Float16>, DataLayout::NCHW);
177ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwFloat16_3, DepthToSpaceTest3<DataType::Float16>, DataLayout::NCHW);
178ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwFloat16_4, DepthToSpaceTest4<DataType::Float16>, DataLayout::NCHW);
179
Sadik Armagan303980c2020-04-17 12:45:14 +0100180ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwInt8_1, DepthToSpaceTest1<DataType::QAsymmS8>, DataLayout::NCHW);
181ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwInt8_2, DepthToSpaceTest2<DataType::QAsymmS8>, DataLayout::NCHW);
182ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwInt8_3, DepthToSpaceTest3<DataType::QAsymmS8>, DataLayout::NCHW);
183ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwInt8_4, DepthToSpaceTest4<DataType::QAsymmS8>, DataLayout::NCHW);
184
Derek Lambertif90c56d2020-01-10 17:14:08 +0000185ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwUint8_1, DepthToSpaceTest1<DataType::QAsymmU8>, DataLayout::NCHW);
186ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwUint8_2, DepthToSpaceTest2<DataType::QAsymmU8>, DataLayout::NCHW);
187ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwUint8_3, DepthToSpaceTest3<DataType::QAsymmU8>, DataLayout::NCHW);
188ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwUint8_4, DepthToSpaceTest4<DataType::QAsymmU8>, DataLayout::NCHW);
Aron Virginas-Tar2f00b742019-09-30 13:28:08 +0100189
Derek Lambertif90c56d2020-01-10 17:14:08 +0000190ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwInt16_1, DepthToSpaceTest1<DataType::QSymmS16>, DataLayout::NCHW);
191ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwInt16_2, DepthToSpaceTest2<DataType::QSymmS16>, DataLayout::NCHW);
192ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwInt16_3, DepthToSpaceTest3<DataType::QSymmS16>, DataLayout::NCHW);
193ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwInt16_4, DepthToSpaceTest4<DataType::QSymmS16>, DataLayout::NCHW);
Aron Virginas-Tar2f00b742019-09-30 13:28:08 +0100194
195ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcFloat32_1, DepthToSpaceTest1<DataType::Float32>, DataLayout::NHWC);
196ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcFloat32_2, DepthToSpaceTest2<DataType::Float32>, DataLayout::NHWC);
197ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcFloat32_3, DepthToSpaceTest3<DataType::Float32>, DataLayout::NHWC);
198ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcFloat32_4, DepthToSpaceTest4<DataType::Float32>, DataLayout::NHWC);
199
Aron Virginas-Tar99836d32019-09-30 16:34:31 +0100200ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcFloat16_1, DepthToSpaceTest1<DataType::Float16>, DataLayout::NHWC);
201ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcFloat16_2, DepthToSpaceTest2<DataType::Float16>, DataLayout::NHWC);
202ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcFloat16_3, DepthToSpaceTest3<DataType::Float16>, DataLayout::NHWC);
203ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcFloat16_4, DepthToSpaceTest4<DataType::Float16>, DataLayout::NHWC);
204
Sadik Armagan303980c2020-04-17 12:45:14 +0100205ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcInt8_1, DepthToSpaceTest1<DataType::QAsymmS8>, DataLayout::NHWC);
206ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcInt8_2, DepthToSpaceTest2<DataType::QAsymmS8>, DataLayout::NHWC);
207ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcInt8_3, DepthToSpaceTest3<DataType::QAsymmS8>, DataLayout::NHWC);
208ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcInt8_4, DepthToSpaceTest4<DataType::QAsymmS8>, DataLayout::NHWC);
209
Derek Lambertif90c56d2020-01-10 17:14:08 +0000210ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcUint8_1, DepthToSpaceTest1<DataType::QAsymmU8>, DataLayout::NHWC);
211ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcUint8_2, DepthToSpaceTest2<DataType::QAsymmU8>, DataLayout::NHWC);
212ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcUint8_3, DepthToSpaceTest3<DataType::QAsymmU8>, DataLayout::NHWC);
213ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcUint8_4, DepthToSpaceTest4<DataType::QAsymmU8>, DataLayout::NHWC);
Aron Virginas-Tar2f00b742019-09-30 13:28:08 +0100214
Derek Lambertif90c56d2020-01-10 17:14:08 +0000215ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcInt16_1, DepthToSpaceTest1<DataType::QSymmS16>, DataLayout::NHWC);
216ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcInt16_2, DepthToSpaceTest2<DataType::QSymmS16>, DataLayout::NHWC);
217ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcInt16_3, DepthToSpaceTest3<DataType::QSymmS16>, DataLayout::NHWC);
218ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcInt16_4, DepthToSpaceTest4<DataType::QSymmS16>, DataLayout::NHWC);
Aron Virginas-Tar2f00b742019-09-30 13:28:08 +0100219
telsoa014fcda012018-03-09 14:13:49 +0000220// Depthwise Convolution
Keith Davisf500d6c2020-08-31 08:32:55 +0100221ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dDepthMul1,
Sadik Armagan581742d2019-08-12 14:11:37 +0100222 DepthwiseConvolution2dDepthMul1Test, true, DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100223ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedDepthwiseConvolution2dDepthMul1,
Sadik Armagan581742d2019-08-12 14:11:37 +0100224 DepthwiseConvolution2dDepthMul1Test, false, DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100225ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dDepthMul1Uint8,
Sadik Armagan581742d2019-08-12 14:11:37 +0100226 DepthwiseConvolution2dDepthMul1Uint8Test, true, DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100227ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedDepthwiseConvolution2dDepthMul1Uint8,
Sadik Armagan581742d2019-08-12 14:11:37 +0100228 DepthwiseConvolution2dDepthMul1Uint8Test, false, DataLayout::NCHW)
jimfly01b9c89632018-10-26 16:50:13 +0100229
230// NHWC Depthwise Convolution
Keith Davisf500d6c2020-08-31 08:32:55 +0100231ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dDepthMul1NHhwc,
Sadik Armagan581742d2019-08-12 14:11:37 +0100232 DepthwiseConvolution2dDepthMul1Test, true, DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100233ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedDepthwiseConvolution2dDepthMul1Nhwc,
Sadik Armagan581742d2019-08-12 14:11:37 +0100234 DepthwiseConvolution2dDepthMul1Test, false, DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100235ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dDepthMul1Uint8Nhwc,
Sadik Armagan581742d2019-08-12 14:11:37 +0100236 DepthwiseConvolution2dDepthMul1Uint8Test, true, DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100237ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedDepthwiseConvolution2dDepthMul1Uint8Nhwc,
Sadik Armagan581742d2019-08-12 14:11:37 +0100238 DepthwiseConvolution2dDepthMul1Uint8Test, false, DataLayout::NHWC)
jimfly01b9c89632018-10-26 16:50:13 +0100239
Keith Davisf500d6c2020-08-31 08:32:55 +0100240ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dDepthNhwc, DepthwiseConvolution2dDepthNhwcTest, false)
241ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleDepthwiseConvolution2d3x3Dilation3x3Nhwc,
Bruno Goncalves22972f02019-04-26 21:03:24 -0300242 SimpleDepthwiseConvolution2d3x3Dilation3x3NhwcTest)
jimfly01b9c89632018-10-26 16:50:13 +0100243
telsoa014fcda012018-03-09 14:13:49 +0000244
Keith Davisf500d6c2020-08-31 08:32:55 +0100245ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dAsymmetric,
Sadik Armagan581742d2019-08-12 14:11:37 +0100246 DepthwiseConvolution2dAsymmetricTest, true, DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100247ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedDepthwiseConvolution2dAsymmetric,
Sadik Armagan581742d2019-08-12 14:11:37 +0100248 DepthwiseConvolution2dAsymmetricTest, false, DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100249ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dAsymmetricNhwc,
Sadik Armagan581742d2019-08-12 14:11:37 +0100250 DepthwiseConvolution2dAsymmetricTest, true, DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100251ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedDepthwiseConvolution2dAsymmetricNhwc,
Sadik Armagan581742d2019-08-12 14:11:37 +0100252 DepthwiseConvolution2dAsymmetricTest, false, DataLayout::NHWC)
surmeh013537c2c2018-05-18 16:31:43 +0100253
Keith Davisf500d6c2020-08-31 08:32:55 +0100254ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dDepthMul64, DepthwiseConvolution2dDepthMul64Test);
Matthew Jackson99107922019-07-25 12:58:56 +0100255
Keith Davisf500d6c2020-08-31 08:32:55 +0100256ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dPerAxisQuantTestNchw, DepthwiseConvolution2dPerAxisQuantTest,
Aron Virginas-Tar24457442019-11-27 13:18:25 +0000257 DataLayout::NCHW);
Keith Davisf500d6c2020-08-31 08:32:55 +0100258ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dPerAxisQuantTestNhwc, DepthwiseConvolution2dPerAxisQuantTest,
Aron Virginas-Tar24457442019-11-27 13:18:25 +0000259 DataLayout::NHWC);
260
telsoa014fcda012018-03-09 14:13:49 +0000261namespace
262{
263
Sadik Armagan581742d2019-08-12 14:11:37 +0100264DepthwiseConvolution2dDescriptor MakeDepthwiseConv2dDesc(uint32_t strideX, uint32_t strideY,
telsoa014fcda012018-03-09 14:13:49 +0000265 uint32_t depthMultiplier = 1, uint32_t padLeft = 0, uint32_t padRight = 0,
266 uint32_t padTop = 0, uint32_t padBottom = 0)
267{
Jan Eilers8eb25602020-03-09 12:13:48 +0000268 IgnoreUnused(depthMultiplier);
telsoa01c577f2c2018-08-31 09:22:23 +0100269
Sadik Armagan581742d2019-08-12 14:11:37 +0100270 DepthwiseConvolution2dDescriptor desc;
telsoa01c577f2c2018-08-31 09:22:23 +0100271
telsoa014fcda012018-03-09 14:13:49 +0000272 desc.m_PadLeft = padLeft;
273 desc.m_PadRight = padRight;
telsoa01c577f2c2018-08-31 09:22:23 +0100274
telsoa014fcda012018-03-09 14:13:49 +0000275 desc.m_PadTop = padTop;
276 desc.m_PadBottom = padBottom;
277 desc.m_StrideX = strideX;
278 desc.m_StrideY = strideY;
telsoa01c577f2c2018-08-31 09:22:23 +0100279 desc.m_BiasEnabled = false;
280
telsoa014fcda012018-03-09 14:13:49 +0000281 return desc;
282}
283
Sadik Armagan581742d2019-08-12 14:11:37 +0100284TensorInfo CreateOutputTensorInfo(const TensorInfo& inputInfo,
285 const TensorInfo& weightsInfo,
286 const DepthwiseConvolution2dDescriptor& descriptor,
287 DataType dataType)
telsoa01c577f2c2018-08-31 09:22:23 +0100288{
Sadik Armagan581742d2019-08-12 14:11:37 +0100289 const TensorShape& inputShape = inputInfo.GetShape();
290 const TensorShape& filterShape = weightsInfo.GetShape();
telsoa01c577f2c2018-08-31 09:22:23 +0100291
292 unsigned int inWidth = inputShape[3];
293 unsigned int inHeight = inputShape[2];
294 unsigned int inBatchSize = inputShape[0];
295
296 unsigned int filterWidth = filterShape[3];
297 unsigned int readWidth = (inWidth + descriptor.m_PadLeft + descriptor.m_PadRight) - (filterWidth);
298 unsigned int outWidth = 1u + (readWidth / descriptor.m_StrideX);
299
300 unsigned int filterHeight = filterShape[2];
301 unsigned int readHeight = (inHeight + descriptor.m_PadTop + descriptor.m_PadBottom) - (filterHeight);
302 unsigned int outHeight = 1u + (readHeight / descriptor.m_StrideY);
303 unsigned int depthMultiplier = filterShape[0];
304
305 unsigned int outChannels = filterShape[1] * depthMultiplier;
306 unsigned int outBatchSize = inBatchSize;
307
Sadik Armagan581742d2019-08-12 14:11:37 +0100308 TensorShape outputShape({outBatchSize, outChannels, outHeight, outWidth});
309 return TensorInfo(outputShape, dataType);
telsoa01c577f2c2018-08-31 09:22:23 +0100310}
telsoa014fcda012018-03-09 14:13:49 +0000311}
312
313BOOST_AUTO_TEST_CASE(DepthwiseConv2dUtils)
314{
Sadik Armagan581742d2019-08-12 14:11:37 +0100315 const DataType dataType = DataType::Float32;
telsoa01c577f2c2018-08-31 09:22:23 +0100316
Sadik Armagan581742d2019-08-12 14:11:37 +0100317 TensorInfo inputInfo({1, 1, 10, 10 }, dataType);
318 TensorInfo outputInfo;
319 TensorInfo weightsInfo3x3({ 1, 1, 3, 3 }, dataType);
320 TensorInfo biasesInfo;
telsoa01c577f2c2018-08-31 09:22:23 +0100321
Sadik Armagan581742d2019-08-12 14:11:37 +0100322 DepthwiseConvolution2dDescriptor descriptor;
323 NeonLayerSupport layerSupport;
telsoa014fcda012018-03-09 14:13:49 +0000324
325 // Strides supported: 1,2,3
telsoa01c577f2c2018-08-31 09:22:23 +0100326 descriptor = MakeDepthwiseConv2dDesc(1, 1);
327 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo3x3, descriptor, dataType);
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100328 BOOST_TEST(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
329 weightsInfo3x3, biasesInfo));
telsoa014fcda012018-03-09 14:13:49 +0000330
telsoa01c577f2c2018-08-31 09:22:23 +0100331 descriptor = MakeDepthwiseConv2dDesc(1, 2);
332 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo3x3, descriptor, dataType);
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100333 BOOST_TEST(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
334 weightsInfo3x3, biasesInfo));
telsoa01c577f2c2018-08-31 09:22:23 +0100335
336 descriptor = MakeDepthwiseConv2dDesc(1, 3);
337 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo3x3, descriptor, dataType);
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100338 BOOST_TEST(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
339 weightsInfo3x3, biasesInfo));
telsoa01c577f2c2018-08-31 09:22:23 +0100340
341 descriptor = MakeDepthwiseConv2dDesc(2, 1);
342 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo3x3, descriptor, dataType);
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100343 BOOST_TEST(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
344 weightsInfo3x3, biasesInfo));
telsoa01c577f2c2018-08-31 09:22:23 +0100345
346 descriptor = MakeDepthwiseConv2dDesc(2, 2);
347 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo3x3, descriptor, dataType);
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100348 BOOST_TEST(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
349 weightsInfo3x3, biasesInfo));
telsoa01c577f2c2018-08-31 09:22:23 +0100350
351 descriptor = MakeDepthwiseConv2dDesc(2, 3);
352 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo3x3, descriptor, dataType);
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100353 BOOST_TEST(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
354 weightsInfo3x3, biasesInfo));
telsoa01c577f2c2018-08-31 09:22:23 +0100355
356 descriptor = MakeDepthwiseConv2dDesc(3, 1);
357 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo3x3, descriptor, dataType);
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100358 BOOST_TEST(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
359 weightsInfo3x3, biasesInfo));
telsoa01c577f2c2018-08-31 09:22:23 +0100360
361 descriptor = MakeDepthwiseConv2dDesc(3, 2);
362 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo3x3, descriptor, dataType);
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100363 BOOST_TEST(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
364 weightsInfo3x3, biasesInfo));
telsoa01c577f2c2018-08-31 09:22:23 +0100365
366 descriptor = MakeDepthwiseConv2dDesc(3, 3);
367 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo3x3, descriptor, dataType);
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100368 BOOST_TEST(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
369 weightsInfo3x3, biasesInfo));
telsoa01c577f2c2018-08-31 09:22:23 +0100370
371 // Supported stride 4
372 descriptor = MakeDepthwiseConv2dDesc(4, 1);
373 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo3x3, descriptor, dataType);
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100374 BOOST_TEST(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
375 weightsInfo3x3, biasesInfo));
telsoa014fcda012018-03-09 14:13:49 +0000376
377 // Supported weights shape 1x1
Sadik Armagan581742d2019-08-12 14:11:37 +0100378 TensorInfo weightsInfo1x1({ 1, 1, 1, 1 }, DataType::Float32);
telsoa01c577f2c2018-08-31 09:22:23 +0100379 descriptor = MakeDepthwiseConv2dDesc(1, 1);
380 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo1x1, descriptor, dataType);
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100381 BOOST_TEST(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
382 weightsInfo1x1, biasesInfo));
telsoa014fcda012018-03-09 14:13:49 +0000383
384 // Supported shape 2x2
Sadik Armagan581742d2019-08-12 14:11:37 +0100385 TensorInfo weightsInfo2x2({ 1, 1, 2, 2 }, DataType::Float32);
telsoa01c577f2c2018-08-31 09:22:23 +0100386 descriptor = MakeDepthwiseConv2dDesc(1, 1);
387 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo2x2, descriptor, dataType);
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100388 BOOST_TEST(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
389 weightsInfo2x2, biasesInfo));
surmeh013537c2c2018-05-18 16:31:43 +0100390
391 // Asymmetric padding
telsoa01c577f2c2018-08-31 09:22:23 +0100392 descriptor = MakeDepthwiseConv2dDesc(1, 1, 1, 1, 2, 1, 2);
393 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo3x3, descriptor, dataType);
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100394 BOOST_TEST(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
395 weightsInfo3x3, biasesInfo));
telsoa014fcda012018-03-09 14:13:49 +0000396}
397
Narumol Prangnawarat01961a72019-05-30 16:47:12 +0100398// Dequantize
Jan Eilersf7107932019-11-01 11:09:36 +0000399// Fp16 is only supported if __ARM_FEATURE_FP16_VECTOR_ARITHMETIC is enabled
Narumol Prangnawarat01961a72019-05-30 16:47:12 +0100400ARMNN_AUTO_TEST_CASE(DequantizeSimpleUint8, DequantizeSimpleUint8Test)
401ARMNN_AUTO_TEST_CASE(DequantizeOffsetUint8, DequantizeOffsetUint8Test)
Jan Eilersf7107932019-11-01 11:09:36 +0000402ARMNN_AUTO_TEST_CASE(DequantizeSimpleInt16, DequantizeSimpleInt16Test)
Narumol Prangnawarat01961a72019-05-30 16:47:12 +0100403
telsoa014fcda012018-03-09 14:13:49 +0000404// Pooling
Finn Williams826a5432020-08-27 16:15:20 +0100405ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling2dSize3x3Stride2x4,
406 SimpleMaxPooling2dSize3x3Stride2x4Test, true)
407ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling2dSize3x3Stride2x4Uint8,
408 SimpleMaxPooling2dSize3x3Stride2x4Uint8Test, true)
James Conroy45a9b772018-10-31 11:47:53 +0000409
Finn Williams826a5432020-08-27 16:15:20 +0100410ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling2d, SimpleMaxPooling2dTest, DataLayout::NCHW)
411ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling2dNhwc, SimpleMaxPooling2dTest, DataLayout::NHWC)
412ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling2dUint8, SimpleMaxPooling2dUint8Test, DataLayout::NCHW)
413ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling2dUint8Nhwc, SimpleMaxPooling2dUint8Test, DataLayout::NHWC)
James Conroy45a9b772018-10-31 11:47:53 +0000414
Finn Williams826a5432020-08-27 16:15:20 +0100415ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleAveragePooling2d, SimpleAveragePooling2dTest, DataLayout::NCHW)
416ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleAveragePooling2dNhwc, SimpleAveragePooling2dTest, DataLayout::NCHW)
417ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleAveragePooling2dUint8,
418 SimpleAveragePooling2dUint8Test, DataLayout::NCHW)
419ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleAveragePooling2dUint8Nhwc,
420 SimpleAveragePooling2dUint8Test, DataLayout::NHWC)
surmeh01bceff2f2018-03-29 16:29:27 +0100421
Finn Williams826a5432020-08-27 16:15:20 +0100422ARMNN_AUTO_TEST_CASE_WITH_THF(LargeTensorsAveragePooling2d, LargeTensorsAveragePooling2dTest)
423ARMNN_AUTO_TEST_CASE_WITH_THF(LargeTensorsAveragePooling2dUint8, LargeTensorsAveragePooling2dUint8Test)
telsoa014fcda012018-03-09 14:13:49 +0000424
Finn Williams826a5432020-08-27 16:15:20 +0100425ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleL2Pooling2d, SimpleL2Pooling2dTest, DataLayout::NCHW)
426ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleL2Pooling2dNeon, SimpleL2Pooling2dTest, DataLayout::NHWC)
427ARMNN_AUTO_TEST_CASE_WITH_THF(UNSUPPORTED_SimpleL2Pooling2dUint8, SimpleL2Pooling2dUint8Test, DataLayout::NCHW)
James Conroy45a9b772018-10-31 11:47:53 +0000428
Finn Williams826a5432020-08-27 16:15:20 +0100429ARMNN_AUTO_TEST_CASE_WITH_THF(L2Pooling2dSize3Stride1, L2Pooling2dSize3Stride1Test)
430ARMNN_AUTO_TEST_CASE_WITH_THF(UNSUPPORTED_L2Pooling2dSize3Stride1Uint8, L2Pooling2dSize3Stride1Uint8Test)
431ARMNN_AUTO_TEST_CASE_WITH_THF(L2Pooling2dSize3Stride3, L2Pooling2dSize3Stride3Test)
432ARMNN_AUTO_TEST_CASE_WITH_THF(UNSUPPORTED_L2Pooling2dSize3Stride3Uint8, L2Pooling2dSize3Stride3Uint8Test)
433ARMNN_AUTO_TEST_CASE_WITH_THF(L2Pooling2dSize3Stride4, L2Pooling2dSize3Stride4Test)
434ARMNN_AUTO_TEST_CASE_WITH_THF(UNSUPPORTED_L2Pooling2dSize3Stride4Uint8, L2Pooling2dSize3Stride4Uint8Test)
435ARMNN_AUTO_TEST_CASE_WITH_THF(L2Pooling2dSize7, L2Pooling2dSize7Test)
436ARMNN_AUTO_TEST_CASE_WITH_THF(UNSUPPORTED_L2Pooling2dSize7Uint8, L2Pooling2dSize7Uint8Test)
437ARMNN_AUTO_TEST_CASE_WITH_THF(L2Pooling2dSize9, L2Pooling2dSize9Test)
438ARMNN_AUTO_TEST_CASE_WITH_THF(UNSUPPORTED_L2Pooling2dSize9Uint8, L2Pooling2dSize9Uint8Test)
telsoa014fcda012018-03-09 14:13:49 +0000439
440// Ignore padding values for pooling but count padding fields into the divisor
Finn Williams826a5432020-08-27 16:15:20 +0100441ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleMaxPooling2d, IgnorePaddingSimpleMaxPooling2dTest)
442ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleMaxPooling2dUint8, IgnorePaddingSimpleMaxPooling2dUint8Test)
443ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingMaxPooling2dSize3, IgnorePaddingMaxPooling2dSize3Test)
444ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingMaxPooling2dSize3Uint8, IgnorePaddingMaxPooling2dSize3Uint8Test)
telsoa014fcda012018-03-09 14:13:49 +0000445
Finn Williams826a5432020-08-27 16:15:20 +0100446ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleAveragePooling2d, IgnorePaddingSimpleAveragePooling2dTest)
447ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleAveragePooling2dUint8, IgnorePaddingSimpleAveragePooling2dUint8Test)
448ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleAveragePooling2dNoPadding,
449 IgnorePaddingSimpleAveragePooling2dNoPaddingTest)
450ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleAveragePooling2dNoPaddingUint8,
451 IgnorePaddingSimpleAveragePooling2dNoPaddingUint8Test)
452ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingAveragePooling2dSize3, IgnorePaddingAveragePooling2dSize3Test)
453ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingAveragePooling2dSize3Uint8,
454 IgnorePaddingAveragePooling2dSize3Uint8Test)
455ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingAveragePooling2dSize3x2Stride2x2,
456 IgnorePaddingAveragePooling2dSize3x2Stride2x2Test, false)
457ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingAveragePooling2dSize3x2Stride2x2NoPadding,
458 IgnorePaddingAveragePooling2dSize3x2Stride2x2Test,
459 true)
telsoa014fcda012018-03-09 14:13:49 +0000460
Finn Williams826a5432020-08-27 16:15:20 +0100461ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleL2Pooling2d, IgnorePaddingSimpleL2Pooling2dTest)
462ARMNN_AUTO_TEST_CASE_WITH_THF(UNSUPPORTED_IgnorePaddingSimpleL2Pooling2dUint8,
463 IgnorePaddingSimpleL2Pooling2dUint8Test)
464ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingL2Pooling2dSize3, IgnorePaddingL2Pooling2dSize3Test)
465ARMNN_AUTO_TEST_CASE_WITH_THF(UNSUPPORTED_IgnorePaddingL2Pooling2dSize3Uint8,
466 IgnorePaddingL2Pooling2dSize3Uint8Test)
telsoa014fcda012018-03-09 14:13:49 +0000467
468// Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100469ARMNN_AUTO_TEST_CASE_WITH_THF(ConstantLinearActivation, ConstantLinearActivationTest)
telsoa014fcda012018-03-09 14:13:49 +0000470
Teresa Charline32c8442020-03-05 14:41:50 +0000471// Sigmoid Activation / Logistic
Keith Davis33a626f2020-08-27 15:38:12 +0100472ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSigmoid, SimpleSigmoidTest)
473ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSigmoidUint8, SimpleSigmoidUint8Test)
Teresa Charline32c8442020-03-05 14:41:50 +0000474
475// BoundedReLU Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100476ARMNN_AUTO_TEST_CASE_WITH_THF(ReLu1, BoundedReLuUpperAndLowerBoundTest)
477ARMNN_AUTO_TEST_CASE_WITH_THF(ReLu6, BoundedReLuUpperBoundOnlyTest)
478ARMNN_AUTO_TEST_CASE_WITH_THF(ReLu1Uint8, BoundedReLuUint8UpperAndLowerBoundTest)
479ARMNN_AUTO_TEST_CASE_WITH_THF(ReLu6Uint8, BoundedReLuUint8UpperBoundOnlyTest)
Narumol Prangnawarat65d30962019-03-14 11:55:03 +0000480
Teresa Charline32c8442020-03-05 14:41:50 +0000481// ReLU Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100482ARMNN_AUTO_TEST_CASE_WITH_THF(ReLu, ReLuTest)
483ARMNN_AUTO_TEST_CASE_WITH_THF(ReLuUint8, ReLuUint8Test)
David Monahan3b3c3812020-02-25 09:03:29 +0000484
Teresa Charline32c8442020-03-05 14:41:50 +0000485// SoftReLU Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100486ARMNN_AUTO_TEST_CASE_WITH_THF(SoftReLu, SoftReLuTest)
Teresa Charline32c8442020-03-05 14:41:50 +0000487
488// LeakyReLU Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100489ARMNN_AUTO_TEST_CASE_WITH_THF(LeakyReLu, LeakyReLuTest)
Teresa Charline32c8442020-03-05 14:41:50 +0000490
491// Abs Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100492ARMNN_AUTO_TEST_CASE_WITH_THF(Abs, AbsTest)
Narumol Prangnawaratb98bbcf2019-04-15 14:14:27 +0100493
Sadik Armagan6095ba52019-09-13 17:07:19 +0100494// Sqrt Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100495ARMNN_AUTO_TEST_CASE_WITH_THF(Sqrt, SqrtTest)
496ARMNN_AUTO_TEST_CASE_WITH_THF(SqrtNN, SqrtNNTest)
Sadik Armagan6095ba52019-09-13 17:07:19 +0100497
Teresa Charline32c8442020-03-05 14:41:50 +0000498// Square Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100499ARMNN_AUTO_TEST_CASE_WITH_THF(Square, SquareTest)
Teresa Charline32c8442020-03-05 14:41:50 +0000500
501// Tanh Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100502ARMNN_AUTO_TEST_CASE_WITH_THF(Tanh, TanhTest)
Teresa Charline32c8442020-03-05 14:41:50 +0000503
504// Elu Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100505ARMNN_AUTO_TEST_CASE_WITH_THF(Elu, EluTest)
Teresa Charline32c8442020-03-05 14:41:50 +0000506
Narumol Prangnawarat65d30962019-03-14 11:55:03 +0000507// Softmax
Sadik Armagan56785c72020-08-27 12:57:20 +0100508ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSoftmaxBeta1, SimpleSoftmaxTest, 1.0f)
509ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSoftmaxBeta2, SimpleSoftmaxTest, 2.0f)
510ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSoftmaxBeta1Uint8, SimpleSoftmaxUint8Test, 1.0f)
511ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSoftmaxBeta2Uint8, SimpleSoftmaxUint8Test, 2.0f)
telsoa014fcda012018-03-09 14:13:49 +0000512
Keith Davis69e653f2020-07-02 11:49:26 +0100513// LogSoftmax
Finn Williamsc43de6a2020-08-27 11:13:25 +0100514ARMNN_AUTO_TEST_CASE_WITH_THF(LogSoftmaxFloat32_1, LogSoftmaxTest1<DataType::Float32>)
Keith Davis69e653f2020-07-02 11:49:26 +0100515
Mike Kelly0be3a882020-01-24 11:27:50 +0000516// Space To Batch Nd
Sadik Armagan714fe5b2020-08-27 12:46:36 +0100517ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdSimpleFloat32, SpaceToBatchNdSimpleFloat32Test)
518ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiChannelsFloat32, SpaceToBatchNdMultiChannelsFloat32Test)
519ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiBlockFloat32, SpaceToBatchNdMultiBlockFloat32Test)
520ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdPaddingFloat32, SpaceToBatchNdPaddingFloat32Test)
Mike Kelly0be3a882020-01-24 11:27:50 +0000521
Sadik Armagan714fe5b2020-08-27 12:46:36 +0100522ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdSimpleUint8, SpaceToBatchNdSimpleUint8Test)
523ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiChannelsUint8, SpaceToBatchNdMultiChannelsUint8Test)
524ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiBlockUint8, SpaceToBatchNdMultiBlockUint8Test)
525ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdPaddingUint8, SpaceToBatchNdPaddingUint8Test)
Mike Kelly0be3a882020-01-24 11:27:50 +0000526
Sadik Armagan714fe5b2020-08-27 12:46:36 +0100527ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdSimpleNhwcFloat32, SpaceToBatchNdSimpleNhwcFloat32Test)
Finn Williams826a5432020-08-27 16:15:20 +0100528ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiChannelsNhwcFloat32,
529 SpaceToBatchNdMultiChannelsNhwcFloat32Test)
Sadik Armagan714fe5b2020-08-27 12:46:36 +0100530ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiBlockNhwcFloat32, SpaceToBatchNdMultiBlockNhwcFloat32Test)
531ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdPaddingNhwcFloat32, SpaceToBatchNdPaddingNhwcFloat32Test)
Mike Kelly0be3a882020-01-24 11:27:50 +0000532
Sadik Armagan714fe5b2020-08-27 12:46:36 +0100533ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdSimpleNhwcUint8, SpaceToBatchNdSimpleNhwcUint8Test)
Finn Williams826a5432020-08-27 16:15:20 +0100534ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiChannelsNhwcUint8,
535 SpaceToBatchNdMultiChannelsNhwcUint8Test)
Sadik Armagan714fe5b2020-08-27 12:46:36 +0100536ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiBlockNhwcUint8, SpaceToBatchNdMultiBlockNhwcUint8Test)
537ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdPaddingNhwcUint8, SpaceToBatchNdPaddingNhwcUint8Test)
Mike Kelly0be3a882020-01-24 11:27:50 +0000538
Ellen Norris-Thompson29794572019-06-26 16:40:36 +0100539// SpaceToDepth
Sadik Armagan5e719842020-08-28 12:17:33 +0100540ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToDepthNchwAsymmQ8, SpaceToDepthNchwAsymmQ8Test)
541ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToDepthNhwcsymmQ8, SpaceToDepthNhwcAsymmQ8Test)
Ellen Norris-Thompson29794572019-06-26 16:40:36 +0100542
Sadik Armagan5e719842020-08-28 12:17:33 +0100543ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToDepthNhwc1Float32, SpaceToDepthNhwcFloat32Test1)
544ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToDepthNchw1Float32, SpaceToDepthNchwFloat32Test1)
Ellen Norris-Thompson29794572019-06-26 16:40:36 +0100545
Sadik Armagan5e719842020-08-28 12:17:33 +0100546ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToDepthNhwc2Float32, SpaceToDepthNhwcFloat32Test2)
547ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToDepthNchw2Float32, SpaceToDepthNchwFloat32Test2)
Ellen Norris-Thompson29794572019-06-26 16:40:36 +0100548
Sadik Armagan5e719842020-08-28 12:17:33 +0100549ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToDepthNhwcQSymm16, SpaceToDepthNhwcQSymm16Test)
550ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToDepthNchwQSymm16, SpaceToDepthNchwQSymm16Test)
Ellen Norris-Thompson29794572019-06-26 16:40:36 +0100551
telsoa01c577f2c2018-08-31 09:22:23 +0100552// Splitter
Sadik Armagan5e719842020-08-28 12:17:33 +0100553ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSplitterFloat32, SplitterFloat32Test)
554ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSplitterUint8, SplitterUint8Test)
telsoa014fcda012018-03-09 14:13:49 +0000555
Sadik Armagan5e719842020-08-28 12:17:33 +0100556ARMNN_AUTO_TEST_CASE_WITH_THF(CopyViaSplitterFloat32, CopyViaSplitterFloat32Test)
557ARMNN_AUTO_TEST_CASE_WITH_THF(CopyViaSplitterUint8, CopyViaSplitterUint8Test)
telsoa014fcda012018-03-09 14:13:49 +0000558
Jim Flynn4ed6c832019-05-20 11:02:46 +0100559// Concat
Keith Davisf500d6c2020-08-31 08:32:55 +0100560ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConcat, ConcatTest)
561ARMNN_AUTO_TEST_CASE_WITH_THF(ConcatUint8, ConcatUint8Test)
562ARMNN_AUTO_TEST_CASE_WITH_THF(ConcatUint8DifferentInputOutputQParam,
Derek Lambertif90c56d2020-01-10 17:14:08 +0000563 ConcatDifferentInputOutputQParamTest<DataType::QAsymmU8>, false)
telsoa014fcda012018-03-09 14:13:49 +0000564
Narumol Prangnawarat250d3922020-03-30 16:11:04 +0100565// Convert from BFloat16 to Float32
Keith Davisf500d6c2020-08-31 08:32:55 +0100566ARMNN_AUTO_TEST_CASE_WITH_THF(ConvertBf16ToFp32, ConvertBf16ToFp32Test)
Narumol Prangnawarat250d3922020-03-30 16:11:04 +0100567
568// Convert from Float32 to BFloat16
Keith Davisf500d6c2020-08-31 08:32:55 +0100569ARMNN_AUTO_TEST_CASE_WITH_THF(ConvertFp32ToBf16, ConvertFp32ToBf16Test)
Narumol Prangnawarat250d3922020-03-30 16:11:04 +0100570
telsoa014fcda012018-03-09 14:13:49 +0000571// Fully Connected
Finn Williams7faf9a82020-08-27 10:37:36 +0100572ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleFullyConnected, FullyConnectedFloat32Test, false, false)
573ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleFullyConnectedWithBias, FullyConnectedFloat32Test, true, false)
574ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleFullyConnectedWithTranspose, FullyConnectedFloat32Test, false, true)
575ARMNN_AUTO_TEST_CASE_WITH_THF(FullyConnectedLarge, FullyConnectedLargeTest, false)
576ARMNN_AUTO_TEST_CASE_WITH_THF(FullyConnectedLargeTransposed, FullyConnectedLargeTest, true)
577ARMNN_AUTO_TEST_CASE_WITH_THF(FullyConnectedUint8, FullyConnectedTest<DataType::QAsymmU8>, false)
578ARMNN_AUTO_TEST_CASE_WITH_THF(FullyConnectedBiasedUint8, FullyConnectedTest<DataType::QAsymmU8>, true)
telsoa014fcda012018-03-09 14:13:49 +0000579
580// Add
Keith Davis33a626f2020-08-27 15:38:12 +0100581ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleAdd, AdditionTest)
582ARMNN_AUTO_TEST_CASE_WITH_THF(Add5d, Addition5dTest)
583ARMNN_AUTO_TEST_CASE_WITH_THF(AddBroadcast, AdditionBroadcastTest)
584ARMNN_AUTO_TEST_CASE_WITH_THF(AddBroadcast1Element, AdditionBroadcast1ElementTest)
telsoa014fcda012018-03-09 14:13:49 +0000585
David Beckbc392452018-09-10 14:47:28 +0100586// Sub
Keith Davis33a626f2020-08-27 15:38:12 +0100587ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSub, SubtractionTest)
588ARMNN_AUTO_TEST_CASE_WITH_THF(SubBroadcast1Element, SubtractionBroadcast1ElementTest)
589ARMNN_AUTO_TEST_CASE_WITH_THF(SubBroadcast, SubtractionBroadcastTest)
590ARMNN_AUTO_TEST_CASE_WITH_THF(SubtractionUint8, SubtractionUint8Test)
591ARMNN_AUTO_TEST_CASE_WITH_THF(SubBroadcastUint8, SubtractionBroadcastUint8Test)
592ARMNN_AUTO_TEST_CASE_WITH_THF(SubBroadcast1ElementUint8, SubtractionBroadcast1ElementUint8Test)
David Beckbc392452018-09-10 14:47:28 +0100593
Pablo Telloe61f0712020-01-23 10:37:17 +0000594// Div
Keith Davis33a626f2020-08-27 15:38:12 +0100595ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleDivision, DivisionTest)
596ARMNN_AUTO_TEST_CASE_WITH_THF(DivisionByZero, DivisionByZeroTest)
597ARMNN_AUTO_TEST_CASE_WITH_THF(DivisionBroadcast1Element, DivisionBroadcast1ElementTest)
598ARMNN_AUTO_TEST_CASE_WITH_THF(DivisionBroadcast1DVector, DivisionBroadcast1DVectorTest)
Pablo Telloe61f0712020-01-23 10:37:17 +0000599
telsoa014fcda012018-03-09 14:13:49 +0000600// Mul
Keith Davis33a626f2020-08-27 15:38:12 +0100601ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMultiplication, MultiplicationTest)
602ARMNN_AUTO_TEST_CASE_WITH_THF(MultiplicationBroadcast1Element, MultiplicationBroadcast1ElementTest)
603ARMNN_AUTO_TEST_CASE_WITH_THF(MultiplicationBroadcast1DVector, MultiplicationBroadcast1DVectorTest)
604ARMNN_AUTO_TEST_CASE_WITH_THF(MultiplicationUint8, MultiplicationUint8Test)
605ARMNN_AUTO_TEST_CASE_WITH_THF(MultiplicationBroadcast1ElementUint8, MultiplicationBroadcast1ElementUint8Test)
606ARMNN_AUTO_TEST_CASE_WITH_THF(MultiplicationBroadcast1DVectorUint8, MultiplicationBroadcast1DVectorUint8Test)
607ARMNN_AUTO_TEST_CASE_WITH_THF(Multiplication5d, Multiplication5dTest)
telsoa014fcda012018-03-09 14:13:49 +0000608
609// Batch Norm
Keith Davis33a626f2020-08-27 15:38:12 +0100610ARMNN_AUTO_TEST_CASE_WITH_THF(BatchNormFloat32, BatchNormFloat32Test)
611ARMNN_AUTO_TEST_CASE_WITH_THF(BatchNormFloat32Nhwc, BatchNormFloat32NhwcTest)
telsoa014fcda012018-03-09 14:13:49 +0000612
David Monahan97451b42020-12-03 09:48:06 +0000613// Rank
614ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize1Float16, RankDimSize1Test<DataType::Float16>)
615ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize1Float32, RankDimSize1Test<DataType::Float32>)
616ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize1QAsymmU8, RankDimSize1Test<DataType::QAsymmU8>)
617ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize1Signed32, RankDimSize1Test<DataType::Signed32>)
618ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize1QSymmS16, RankDimSize1Test<DataType::QSymmS16>)
619ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize1QAsymmS8, RankDimSize1Test<DataType::QAsymmS8>)
620ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize1BFloat16, RankDimSize1Test<DataType::BFloat16>)
621
622ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize2Float16, RankDimSize2Test<DataType::Float16>)
623ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize2Float32, RankDimSize2Test<DataType::Float32>)
624ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize2QAsymmU8, RankDimSize2Test<DataType::QAsymmU8>)
625ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize2Signed32, RankDimSize2Test<DataType::Signed32>)
626ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize2QSymmS16, RankDimSize2Test<DataType::QSymmS16>)
627ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize2QAsymmS8, RankDimSize2Test<DataType::QAsymmS8>)
628ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize2BFloat16, RankDimSize2Test<DataType::BFloat16>)
629
630ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize3Float16, RankDimSize3Test<DataType::Float16>)
631ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize3Float32, RankDimSize3Test<DataType::Float32>)
632ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize3QAsymmU8, RankDimSize3Test<DataType::QAsymmU8>)
633ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize3Signed32, RankDimSize3Test<DataType::Signed32>)
634ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize3QSymmS16, RankDimSize3Test<DataType::QSymmS16>)
635ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize3QAsymmS8, RankDimSize3Test<DataType::QAsymmS8>)
636ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize3BFloat16, RankDimSize3Test<DataType::BFloat16>)
637
638ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize4Float16, RankDimSize4Test<DataType::Float16>)
639ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize4Float32, RankDimSize4Test<DataType::Float32>)
640ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize4QAsymmU8, RankDimSize4Test<DataType::QAsymmU8>)
641ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize4Signed32, RankDimSize4Test<DataType::Signed32>)
642ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize4QSymmS16, RankDimSize4Test<DataType::QSymmS16>)
643ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize4QAsymmS8, RankDimSize4Test<DataType::QAsymmS8>)
644ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize4BFloat16, RankDimSize4Test<DataType::BFloat16>)
645
Sadik Armagan0d4863d2019-10-09 14:26:32 +0100646// InstanceNormalization
Finn Williamsc43de6a2020-08-27 11:13:25 +0100647ARMNN_AUTO_TEST_CASE_WITH_THF(InstanceNormFloat32Nchw, InstanceNormFloat32Test, DataLayout::NCHW);
648ARMNN_AUTO_TEST_CASE_WITH_THF(InstanceNormFloat32Nhwc, InstanceNormFloat32Test, DataLayout::NHWC);
Sadik Armagan0d4863d2019-10-09 14:26:32 +0100649
Finn Williamsc43de6a2020-08-27 11:13:25 +0100650ARMNN_AUTO_TEST_CASE_WITH_THF(InstanceNormFloat32Nchw2, InstanceNormFloat32Test2, DataLayout::NCHW);
651ARMNN_AUTO_TEST_CASE_WITH_THF(InstanceNormFloat32Nhwc2, InstanceNormFloat32Test2, DataLayout::NHWC);
Sadik Armagan0d4863d2019-10-09 14:26:32 +0100652
telsoa014fcda012018-03-09 14:13:49 +0000653// Constant
Keith Davisf500d6c2020-08-31 08:32:55 +0100654ARMNN_AUTO_TEST_CASE_WITH_THF(Constant, ConstantTest)
655ARMNN_AUTO_TEST_CASE_WITH_THF(ConstantUint8, ConstantUint8SimpleQuantizationScaleNoOffsetTest)
telsoa014fcda012018-03-09 14:13:49 +0000656
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100657// Concat
Keith Davisf500d6c2020-08-31 08:32:55 +0100658ARMNN_AUTO_TEST_CASE_WITH_THF(Concat1d, Concat1dTest)
659ARMNN_AUTO_TEST_CASE_WITH_THF(Concat1dUint8, Concat1dUint8Test)
telsoa014fcda012018-03-09 14:13:49 +0000660
Keith Davisf500d6c2020-08-31 08:32:55 +0100661ARMNN_AUTO_TEST_CASE_WITH_THF(Concat2dDim0, Concat2dDim0Test)
662ARMNN_AUTO_TEST_CASE_WITH_THF(Concat2dDim0Uint8, Concat2dDim0Uint8Test)
663ARMNN_AUTO_TEST_CASE_WITH_THF(Concat2dDim1, Concat2dDim1Test)
664ARMNN_AUTO_TEST_CASE_WITH_THF(Concat2dDim1Uint8, Concat2dDim1Uint8Test)
telsoa014fcda012018-03-09 14:13:49 +0000665
Keith Davisf500d6c2020-08-31 08:32:55 +0100666ARMNN_AUTO_TEST_CASE_WITH_THF(Concat2dDim0DiffInputDims, Concat2dDim0DiffInputDimsTest)
667ARMNN_AUTO_TEST_CASE_WITH_THF(Concat2dDim0DiffInputDimsUint8, Concat2dDim0DiffInputDimsUint8Test)
668ARMNN_AUTO_TEST_CASE_WITH_THF(Concat2dDim1DiffInputDims, Concat2dDim1DiffInputDimsTest)
669ARMNN_AUTO_TEST_CASE_WITH_THF(Concat2dDim1DiffInputDimsUint8, Concat2dDim1DiffInputDimsUint8Test)
telsoa014fcda012018-03-09 14:13:49 +0000670
Keith Davisf500d6c2020-08-31 08:32:55 +0100671ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim0, Concat3dDim0Test)
672ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim0Uint8, Concat3dDim0Uint8Test)
673ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim1, Concat3dDim1Test)
674ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim1Uint8, Concat3dDim1Uint8Test)
675ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim2, Concat3dDim2Test, false)
676ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim2Uint8, Concat3dDim2Uint8Test, false)
telsoa014fcda012018-03-09 14:13:49 +0000677
Keith Davisf500d6c2020-08-31 08:32:55 +0100678ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim0DiffInputDims, Concat3dDim0DiffInputDimsTest)
679ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim0DiffInputDimsUint8, Concat3dDim0DiffInputDimsUint8Test)
680ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim1DiffInputDims, Concat3dDim1DiffInputDimsTest)
681ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim1DiffInputDimsUint8, Concat3dDim1DiffInputDimsUint8Test)
682ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim2DiffInputDims, Concat3dDim2DiffInputDimsTest, false)
683ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim2DiffInputDimsUint8, Concat3dDim2DiffInputDimsUint8Test, false)
telsoa014fcda012018-03-09 14:13:49 +0000684
Keith Davisf500d6c2020-08-31 08:32:55 +0100685ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDim0, Concat4dDim0Test)
686ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDim1, Concat4dDim1Test)
687ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDim3, Concat4dDim3Test, false)
688ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDim0Uint8, Concat4dDim0Uint8Test)
689ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDim1Uint8, Concat4dDim1Uint8Test)
690ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDim3Uint8, Concat4dDim3Uint8Test, false)
narpra015cdda352018-11-19 15:30:27 +0000691
Keith Davisf500d6c2020-08-31 08:32:55 +0100692ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDiffShapeDim0, Concat4dDiffShapeDim0Test)
693ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDiffShapeDim1, Concat4dDiffShapeDim1Test)
694ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDiffShapeDim3, Concat4dDiffShapeDim3Test, false)
695ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDiffShapeDim0Uint8, Concat4dDiffShapeDim0Uint8Test)
696ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDiffShapeDim1Uint8, Concat4dDiffShapeDim1Uint8Test)
697ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDiffShapeDim3Uint8, Concat4dDiffShapeDim3Uint8Test, false)
Sadik Armagan0d4863d2019-10-09 14:26:32 +0100698
telsoa014fcda012018-03-09 14:13:49 +0000699// L2 Normalization
Finn Williamsc43de6a2020-08-27 11:13:25 +0100700ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization1d, L2Normalization1dTest, DataLayout::NCHW)
701ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization2d, L2Normalization2dTest, DataLayout::NCHW)
702ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization3d, L2Normalization3dTest, DataLayout::NCHW)
703ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization4d, L2Normalization4dTest, DataLayout::NCHW)
Matteo Martincigh539b44d2018-10-01 09:26:39 +0100704
Finn Williamsc43de6a2020-08-27 11:13:25 +0100705ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization1dNhwc, L2Normalization1dTest, DataLayout::NHWC)
706ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization2dNhwc, L2Normalization2dTest, DataLayout::NHWC)
707ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization3dNhwc, L2Normalization3dTest, DataLayout::NHWC)
708ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization4dNhwc, L2Normalization4dTest, DataLayout::NHWC)
telsoa014fcda012018-03-09 14:13:49 +0000709
Finn Williamsc43de6a2020-08-27 11:13:25 +0100710ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization2dShape, L2Normalization2dShapeTest);
Matthew Jackson82b15ed2019-07-25 16:14:30 +0100711
Finn Williamsc43de6a2020-08-27 11:13:25 +0100712ARMNN_AUTO_TEST_CASE_WITH_THF(L2NormalizationDefaultEpsilon, L2NormalizationDefaultEpsilonTest, DataLayout::NCHW)
713ARMNN_AUTO_TEST_CASE_WITH_THF(L2NormalizationNonDefaultEpsilon, L2NormalizationNonDefaultEpsilonTest, DataLayout::NCHW)
Ferran Balaguer0c64f462019-06-18 16:32:30 +0100714
telsoa014fcda012018-03-09 14:13:49 +0000715// Floor
Finn Williams7faf9a82020-08-27 10:37:36 +0100716ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleFloor, SimpleFloorTest<DataType::Float32>)
telsoa014fcda012018-03-09 14:13:49 +0000717
Teresa Charlinf540eb82020-04-10 19:24:55 +0100718// Gather
Finn Williamsc43de6a2020-08-27 11:13:25 +0100719ARMNN_AUTO_TEST_CASE_WITH_THF(Gather1dParamsFloat32, Gather1dParamsFloat32Test)
720ARMNN_AUTO_TEST_CASE_WITH_THF(Gather1dParamsUint8, Gather1dParamsUint8Test)
721ARMNN_AUTO_TEST_CASE_WITH_THF(GatherMultiDimParamsFloat32, GatherMultiDimParamsFloat32Test)
722ARMNN_AUTO_TEST_CASE_WITH_THF(GatherMultiDimParamsUint8, GatherMultiDimParamsUint8Test)
Teresa Charlinf540eb82020-04-10 19:24:55 +0100723
Teresa Charlincedd34f2020-03-30 11:17:30 +0100724// Equal
Keith Davisf500d6c2020-08-31 08:32:55 +0100725ARMNN_AUTO_TEST_CASE_WITH_THF(EqualSimple, EqualSimpleTest)
726ARMNN_AUTO_TEST_CASE_WITH_THF(EqualBroadcast1Element, EqualBroadcast1ElementTest)
727ARMNN_AUTO_TEST_CASE_WITH_THF(EqualBroadcast1dVector, EqualBroadcast1dVectorTest)
Teresa Charlincedd34f2020-03-30 11:17:30 +0100728
Keith Davisf500d6c2020-08-31 08:32:55 +0100729ARMNN_AUTO_TEST_CASE_WITH_THF(EqualSimpleUint8, EqualSimpleUint8Test)
730ARMNN_AUTO_TEST_CASE_WITH_THF(EqualBroadcast1ElementUint8, EqualBroadcast1ElementUint8Test)
731ARMNN_AUTO_TEST_CASE_WITH_THF(EqualBroadcast1dVectorUint8, EqualBroadcast1dVectorUint8Test)
Teresa Charlincedd34f2020-03-30 11:17:30 +0100732
kevmay01eed85922019-01-28 08:37:25 +0000733// Greater
Keith Davisf500d6c2020-08-31 08:32:55 +0100734ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterSimple, GreaterSimpleTest)
735ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterBroadcast1Element, GreaterBroadcast1ElementTest)
736ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterBroadcast1dVector, GreaterBroadcast1dVectorTest)
Aron Virginas-Tar3bc00ec2019-10-18 15:42:58 +0100737
Keith Davisf500d6c2020-08-31 08:32:55 +0100738ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterSimpleUint8, GreaterSimpleUint8Test)
739ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterBroadcast1ElementUint8, GreaterBroadcast1ElementUint8Test)
740ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterBroadcast1dVectorUint8, GreaterBroadcast1dVectorUint8Test)
kevmay01eed85922019-01-28 08:37:25 +0000741
Teresa Charlincedd34f2020-03-30 11:17:30 +0100742// GreaterOrEqual
Keith Davisf500d6c2020-08-31 08:32:55 +0100743ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterOrEqualSimple, GreaterOrEqualSimpleTest)
744ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterOrEqualBroadcast1Element, GreaterOrEqualBroadcast1ElementTest)
745ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterOrEqualBroadcast1dVector, GreaterOrEqualBroadcast1dVectorTest)
Teresa Charlincedd34f2020-03-30 11:17:30 +0100746
Keith Davisf500d6c2020-08-31 08:32:55 +0100747ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterOrEqualSimpleUint8, GreaterOrEqualSimpleUint8Test)
748ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterOrEqualBroadcast1ElementUint8, GreaterOrEqualBroadcast1ElementUint8Test)
749ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterOrEqualBroadcast1dVectorUint8, GreaterOrEqualBroadcast1dVectorUint8Test)
Teresa Charlincedd34f2020-03-30 11:17:30 +0100750
751// Less
Keith Davisf500d6c2020-08-31 08:32:55 +0100752ARMNN_AUTO_TEST_CASE_WITH_THF(LessSimple, LessSimpleTest)
753ARMNN_AUTO_TEST_CASE_WITH_THF(LessBroadcast1Element, LessBroadcast1ElementTest)
754ARMNN_AUTO_TEST_CASE_WITH_THF(LessBroadcast1dVector, LessBroadcast1dVectorTest)
Teresa Charlincedd34f2020-03-30 11:17:30 +0100755
Keith Davisf500d6c2020-08-31 08:32:55 +0100756ARMNN_AUTO_TEST_CASE_WITH_THF(LessSimpleUint8, LessSimpleUint8Test)
757ARMNN_AUTO_TEST_CASE_WITH_THF(LessBroadcast1ElementUint8, LessBroadcast1ElementUint8Test)
758ARMNN_AUTO_TEST_CASE_WITH_THF(LessBroadcast1dVectorUint8, LessBroadcast1dVectorUint8Test)
Teresa Charlincedd34f2020-03-30 11:17:30 +0100759
760// LessOrEqual
Keith Davisf500d6c2020-08-31 08:32:55 +0100761ARMNN_AUTO_TEST_CASE_WITH_THF(LessOrEqualSimple, LessOrEqualSimpleTest)
762ARMNN_AUTO_TEST_CASE_WITH_THF(LessOrEqualBroadcast1Element, LessOrEqualBroadcast1ElementTest)
763ARMNN_AUTO_TEST_CASE_WITH_THF(LessOrEqualBroadcast1dVector, LessOrEqualBroadcast1dVectorTest)
Teresa Charlincedd34f2020-03-30 11:17:30 +0100764
Keith Davisf500d6c2020-08-31 08:32:55 +0100765ARMNN_AUTO_TEST_CASE_WITH_THF(LessOrEqualSimpleUint8, LessOrEqualSimpleUint8Test)
766ARMNN_AUTO_TEST_CASE_WITH_THF(LessOrEqualBroadcast1ElementUint8, LessOrEqualBroadcast1ElementUint8Test)
767ARMNN_AUTO_TEST_CASE_WITH_THF(LessOrEqualBroadcast1dVectorUint8, LessOrEqualBroadcast1dVectorUint8Test)
Teresa Charlincedd34f2020-03-30 11:17:30 +0100768
769// NotEqual
Keith Davisf500d6c2020-08-31 08:32:55 +0100770ARMNN_AUTO_TEST_CASE_WITH_THF(NotEqualSimple, NotEqualSimpleTest)
771ARMNN_AUTO_TEST_CASE_WITH_THF(NotEqualBroadcast1Element, NotEqualBroadcast1ElementTest)
772ARMNN_AUTO_TEST_CASE_WITH_THF(NotEqualBroadcast1dVector, NotEqualBroadcast1dVectorTest)
Teresa Charlincedd34f2020-03-30 11:17:30 +0100773
Keith Davisf500d6c2020-08-31 08:32:55 +0100774ARMNN_AUTO_TEST_CASE_WITH_THF(NotEqualSimpleUint8, NotEqualSimpleUint8Test)
775ARMNN_AUTO_TEST_CASE_WITH_THF(NotEqualBroadcast1ElementUint8, NotEqualBroadcast1ElementUint8Test)
776ARMNN_AUTO_TEST_CASE_WITH_THF(NotEqualBroadcast1dVectorUint8, NotEqualBroadcast1dVectorUint8Test)
Teresa Charlincedd34f2020-03-30 11:17:30 +0100777
telsoa014fcda012018-03-09 14:13:49 +0000778// Reshape
Finn Williamsec36d3e2020-08-28 13:17:05 +0100779ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleReshapeFloat32, SimpleReshapeTest<armnn::DataType::Float32>)
780ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleReshapeInt8, SimpleReshapeTest<armnn::DataType::QAsymmS8>)
781ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleReshapeUint8, SimpleReshapeTest<armnn::DataType::QAsymmU8>)
782ARMNN_AUTO_TEST_CASE_WITH_THF(Reshape5d, Reshape5dTest<armnn::DataType::Float32>)
Narumol Prangnawarat0c95f4c2020-11-18 16:52:07 +0000783ARMNN_AUTO_TEST_CASE_WITH_THF(ReshapeBoolean, ReshapeBooleanTest)
telsoa014fcda012018-03-09 14:13:49 +0000784
Éanna Ó Catháin12055742019-01-25 10:01:40 +0000785// Pad
Finn Williams826a5432020-08-27 16:15:20 +0100786ARMNN_AUTO_TEST_CASE_WITH_THF(PadFloat322d, PadFloat322dTest)
787ARMNN_AUTO_TEST_CASE_WITH_THF(PadFloat322dCustomPadding, PadFloat322dCustomPaddingTest)
788ARMNN_AUTO_TEST_CASE_WITH_THF(PadFloat323d, PadFloat323dTest)
789ARMNN_AUTO_TEST_CASE_WITH_THF(PadFloat324d, PadFloat324dTest)
Éanna Ó Catháin12055742019-01-25 10:01:40 +0000790
Finn Williams826a5432020-08-27 16:15:20 +0100791ARMNN_AUTO_TEST_CASE_WITH_THF(PadUint82d, PadUint82dTest)
792ARMNN_AUTO_TEST_CASE_WITH_THF(PadUint82dCustomPadding, PadUint82dCustomPaddingTest)
793ARMNN_AUTO_TEST_CASE_WITH_THF(PadUint83d, PadUint83dTest)
794ARMNN_AUTO_TEST_CASE_WITH_THF(PadUint84d, PadUint84dTest)
Teresa Charlinec8e1982019-07-02 16:24:09 +0100795
Finn Williams826a5432020-08-27 16:15:20 +0100796ARMNN_AUTO_TEST_CASE_WITH_THF(Pad2dQSymm16, Pad2dTestCommon<DataType::QSymmS16>, 2.0f, 0, 0.0f)
797ARMNN_AUTO_TEST_CASE_WITH_THF(Pad2dQSymm16CustomPadding, Pad2dTestCommon<DataType::QSymmS16>, 2.0f, 0, 1.0f)
798ARMNN_AUTO_TEST_CASE_WITH_THF(Pad3dQSymm16, Pad3dTestCommon<DataType::QSymmS16>, 2.0f, 0)
799ARMNN_AUTO_TEST_CASE_WITH_THF(Pad4dQSymm16, Pad4dTestCommon<DataType::QSymmS16>, 2.0f, 0)
FinnWilliamsArm8b797a82019-07-09 16:01:41 +0100800
telsoa014fcda012018-03-09 14:13:49 +0000801// Permute
Finn Williams826a5432020-08-27 16:15:20 +0100802ARMNN_AUTO_TEST_CASE_WITH_THF(SimplePermuteFloat32, SimplePermuteTest<DataType::Float32>)
803ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteFloat32ValueSet1Test, PermuteValueSet1Test<DataType::Float32>)
804ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteFloat32ValueSet2Test, PermuteValueSet2Test<DataType::Float32>)
805ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteFloat32ValueSet3Test, PermuteValueSet3Test<DataType::Float32>)
806ARMNN_AUTO_TEST_CASE_WITH_THF(SimplePermuteQASymmS8, SimplePermuteTest<DataType::QAsymmS8>)
807ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteQASymmS8ValueSet1Test, PermuteValueSet1Test<DataType::QAsymmS8>)
808ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteQASymmS8ValueSet2Test, PermuteValueSet2Test<DataType::QAsymmS8>)
809ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteQASymmS8ValueSet3Test, PermuteValueSet3Test<DataType::QAsymmS8>)
810ARMNN_AUTO_TEST_CASE_WITH_THF(SimplePermuteQASymm8, SimplePermuteTest<DataType::QAsymmU8>)
811ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteQASymm8ValueSet1Test, PermuteValueSet1Test<DataType::QAsymmU8>)
812ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteQASymm8ValueSet2Test, PermuteValueSet2Test<DataType::QAsymmU8>)
813ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteQASymm8ValueSet3Test, PermuteValueSet3Test<DataType::QAsymmU8>)
surmeh01bceff2f2018-03-29 16:29:27 +0100814
Les Bellde9011b2018-10-03 10:37:52 +0100815// Lstm
Finn Williamsc43de6a2020-08-27 11:13:25 +0100816ARMNN_AUTO_TEST_CASE_WITH_THF(LstmLayerFloat32WithCifgWithPeepholeNoProjection,
817 LstmLayerFloat32WithCifgWithPeepholeNoProjectionTest)
818ARMNN_AUTO_TEST_CASE_WITH_THF(LstmLayerFloat32NoCifgNoPeepholeNoProjection,
819 LstmLayerFloat32NoCifgNoPeepholeNoProjectionTest)
820ARMNN_AUTO_TEST_CASE_WITH_THF(LstmLayerFloat32NoCifgWithPeepholeWithProjection,
821 LstmLayerFloat32NoCifgWithPeepholeWithProjectionTest)
822ARMNN_AUTO_TEST_CASE_WITH_THF(LstmLayerFloat32NoCifgWithPeepholeWithProjectionWithLayerNorm,
823 LstmLayerFloat32NoCifgWithPeepholeWithProjectionWithLayerNormTest)
Les Bellde9011b2018-10-03 10:37:52 +0100824
James Conroycc340932020-05-12 18:08:52 +0100825// QLstm
Finn Williamsc43de6a2020-08-27 11:13:25 +0100826ARMNN_AUTO_TEST_CASE_WITH_THF(QLstm, QLstmTest)
827ARMNN_AUTO_TEST_CASE_WITH_THF(QLstm1, QLstmTest1)
828ARMNN_AUTO_TEST_CASE_WITH_THF(QLstm2, QLstmTest2)
James Conroycc340932020-05-12 18:08:52 +0100829
830// QuantizedLstm
Finn Williamsc43de6a2020-08-27 11:13:25 +0100831ARMNN_AUTO_TEST_CASE_WITH_THF(QuantizedLstm, QuantizedLstmTest)
Francis Murtagh4fc3c482019-08-02 13:20:54 +0100832
Matthew Benthamfd899962018-12-31 15:49:42 +0000833// Mean
Finn Williams826a5432020-08-27 16:15:20 +0100834ARMNN_AUTO_TEST_CASE_WITH_THF(MeanSimpleFloat32, MeanSimpleTest<DataType::Float32>)
835ARMNN_AUTO_TEST_CASE_WITH_THF(MeanSimpleAxisFloat32, MeanSimpleAxisTest<DataType::Float32>)
836ARMNN_AUTO_TEST_CASE_WITH_THF(MeanKeepDimsFloat32, MeanKeepDimsTest<DataType::Float32>)
837ARMNN_AUTO_TEST_CASE_WITH_THF(MeanMultipleDimsFloat32, MeanMultipleDimsTest<DataType::Float32>)
838ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts1Float32, MeanVts1Test<DataType::Float32>)
839ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts2Float32, MeanVts2Test<DataType::Float32>)
840ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts3Float32, MeanVts3Test<DataType::Float32>)
Matthew Benthamfd899962018-12-31 15:49:42 +0000841
Finn Williams826a5432020-08-27 16:15:20 +0100842ARMNN_AUTO_TEST_CASE_WITH_THF(MeanSimpleQuantisedAsymmS8, MeanSimpleTest<DataType::QAsymmS8>)
843ARMNN_AUTO_TEST_CASE_WITH_THF(MeanSimpleAxisQuantisedAsymmS8, MeanSimpleAxisTest<DataType::QAsymmS8>)
844ARMNN_AUTO_TEST_CASE_WITH_THF(MeanKeepDimsQuantisedAsymmS8, MeanKeepDimsTest<DataType::QAsymmS8>)
845ARMNN_AUTO_TEST_CASE_WITH_THF(MeanMultipleDimsQuantisedAsymmS8, MeanMultipleDimsTest<DataType::QAsymmS8>)
846ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts1QuantisedAsymmS8, MeanVts1Test<DataType::QAsymmS8>)
847ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts2QuantisedAsymmS8, MeanVts2Test<DataType::QAsymmS8>)
848ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts3QuantisedAsymmS8, MeanVts3Test<DataType::QAsymmS8>)
Sadik Armagan303980c2020-04-17 12:45:14 +0100849
Finn Williams826a5432020-08-27 16:15:20 +0100850ARMNN_AUTO_TEST_CASE_WITH_THF(MeanSimpleQuantisedAsymm8, MeanSimpleTest<DataType::QAsymmU8>)
851ARMNN_AUTO_TEST_CASE_WITH_THF(MeanSimpleAxisQuantisedAsymm8, MeanSimpleAxisTest<DataType::QAsymmU8>)
852ARMNN_AUTO_TEST_CASE_WITH_THF(MeanKeepDimsQuantisedAsymm8, MeanKeepDimsTest<DataType::QAsymmU8>)
853ARMNN_AUTO_TEST_CASE_WITH_THF(MeanMultipleDimsQuantisedAsymm8, MeanMultipleDimsTest<DataType::QAsymmU8>)
854ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts1QuantisedAsymm8, MeanVts1Test<DataType::QAsymmU8>)
855ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts2QuantisedAsymm8, MeanVts2Test<DataType::QAsymmU8>)
856ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts3QuantisedAsymm8, MeanVts3Test<DataType::QAsymmU8>)
Matthew Benthamfd899962018-12-31 15:49:42 +0000857
Nattapat Chaimanowong4e6597a2018-12-20 14:14:06 +0000858// Max
Keith Davis33a626f2020-08-27 15:38:12 +0100859ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaximum, MaximumSimpleTest)
860ARMNN_AUTO_TEST_CASE_WITH_THF(MaximumBroadcast1Element, MaximumBroadcast1ElementTest)
861ARMNN_AUTO_TEST_CASE_WITH_THF(MaximumBroadcast1DVector, MaximumBroadcast1DVectorTest)
862ARMNN_AUTO_TEST_CASE_WITH_THF(MaximumUint8, MaximumUint8Test)
863ARMNN_AUTO_TEST_CASE_WITH_THF(MaximumBroadcast1ElementUint8, MaximumBroadcast1ElementUint8Test)
864ARMNN_AUTO_TEST_CASE_WITH_THF(MaximumBroadcast1DVectorUint8, MaximumBroadcast1DVectorUint8Test)
Nattapat Chaimanowong4e6597a2018-12-20 14:14:06 +0000865
Conor Kennedy54b21692019-01-09 07:57:38 +0000866// Min
Keith Davis33a626f2020-08-27 15:38:12 +0100867ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMinimum1, MinimumBroadcast1ElementTest1)
868ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMinimum2, MinimumBroadcast1ElementTest2)
869ARMNN_AUTO_TEST_CASE_WITH_THF(Minimum1DVectorUint8, MinimumBroadcast1DVectorUint8Test)
Conor Kennedy54b21692019-01-09 07:57:38 +0000870
narpra0155a97bc2018-10-02 14:35:53 +0100871// Normalization
Finn Williams826a5432020-08-27 16:15:20 +0100872ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleNormalizationAcross, SimpleNormalizationAcrossTest)
873ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleNormalizationWithin, SimpleNormalizationWithinTest)
874ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleNormalizationAcrossNhwc, SimpleNormalizationAcrossNhwcTest)
narpra0155a97bc2018-10-02 14:35:53 +0100875
Sadik Armaganc625f002018-12-17 11:32:16 +0000876// Resize Bilinear - NCHW data layout
Finn Williamsec36d3e2020-08-28 13:17:05 +0100877ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeBilinear, SimpleResizeBilinearTest<DataType::Float32>, DataLayout::NCHW)
878ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearNop, ResizeBilinearNopTest<DataType::Float32>, DataLayout::NCHW)
879ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearSqMin, ResizeBilinearSqMinTest<DataType::Float32>, DataLayout::NCHW)
880ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMin, ResizeBilinearMinTest<DataType::Float32>, DataLayout::NCHW)
881ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMag, ResizeBilinearMagTest<DataType::Float32>, DataLayout::NCHW)
Sadik Armaganc625f002018-12-17 11:32:16 +0000882
Finn Williamsec36d3e2020-08-28 13:17:05 +0100883ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeBilinearUint8,
884 SimpleResizeBilinearTest<DataType::QAsymmU8>,
885 DataLayout::NCHW)
886ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearNopUint8,
887 ResizeBilinearNopTest<DataType::QAsymmU8>,
888 DataLayout::NCHW)
889ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearSqMinUint8,
890 ResizeBilinearSqMinTest<DataType::QAsymmU8>,
891 DataLayout::NCHW)
892ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMinUint8,
893 ResizeBilinearMinTest<DataType::QAsymmU8>,
894 DataLayout::NCHW)
895ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMagUint8,
896 ResizeBilinearMagTest<DataType::QAsymmU8>,
897 DataLayout::NCHW)
Conor Kennedydc1c0822019-02-01 07:23:17 +0000898
Finn Williamsec36d3e2020-08-28 13:17:05 +0100899ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeBilinear,
900 HalfPixelCentersResizeBilinearTest<DataType::Float32>,
901 DataLayout::NCHW)
902ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeBilinear,
903 AlignCornersResizeBilinearTest<DataType::Float32>,
904 DataLayout::NCHW)
905ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeBilinearInt8,
906 HalfPixelCentersResizeBilinearTest<DataType::QAsymmS8>,
907 DataLayout::NCHW)
908ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeBilinearInt8,
909 AlignCornersResizeBilinearTest<DataType::QAsymmS8>,
910 DataLayout::NCHW)
911ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeBilinearUint8,
912 HalfPixelCentersResizeBilinearTest<DataType::QAsymmU8>,
913 DataLayout::NCHW)
914ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeBilinearUint8,
915 AlignCornersResizeBilinearTest<DataType::QAsymmU8>,
916 DataLayout::NCHW)
Teresa Charlin686f45c2020-07-23 17:59:00 +0100917
Sadik Armaganc625f002018-12-17 11:32:16 +0000918// Resize Bilinear - NHWC data layout
Finn Williamsec36d3e2020-08-28 13:17:05 +0100919ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearNopNhwc,
920 ResizeBilinearNopTest<DataType::Float32>,
921 DataLayout::NHWC)
922ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeBilinearNhwc,
923 SimpleResizeBilinearTest<DataType::Float32>,
924 DataLayout::NHWC)
925ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearSqMinNhwc,
926 ResizeBilinearSqMinTest<DataType::Float32>,
927 DataLayout::NHWC)
928ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMinNhwc,
929 ResizeBilinearMinTest<DataType::Float32>,
930 DataLayout::NHWC)
931ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMagNhwc,
932 ResizeBilinearMagTest<DataType::Float32>,
933 DataLayout::NHWC)
Ellen Norris-Thompsonc4223d12019-06-10 14:15:12 +0100934
Finn Williamsec36d3e2020-08-28 13:17:05 +0100935ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearNopUint8Nhwc,
936 ResizeBilinearNopTest<DataType::QAsymmU8>,
937 DataLayout::NHWC)
938ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeBilinearUint8Nhwc,
939 SimpleResizeBilinearTest<DataType::QAsymmU8>,
940 DataLayout::NHWC)
941ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearSqMinUint8Nhwc,
942 ResizeBilinearSqMinTest<DataType::QAsymmU8>,
943 DataLayout::NHWC)
944ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMinUint8Nhwc,
945 ResizeBilinearMinTest<DataType::QAsymmU8>,
946 DataLayout::NHWC)
947ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMagUint8Nhwc,
948 ResizeBilinearMagTest<DataType::QAsymmU8>,
949 DataLayout::NHWC)
Sadik Armaganc625f002018-12-17 11:32:16 +0000950
Finn Williamsec36d3e2020-08-28 13:17:05 +0100951ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeBilinearNhwc,
952 HalfPixelCentersResizeBilinearTest<DataType::Float32>,
953 DataLayout::NHWC)
954ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeBilinearNhwc,
955 AlignCornersResizeBilinearTest<DataType::Float32>,
956 DataLayout::NHWC)
957ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeBilinearInt8Nhwc,
958 HalfPixelCentersResizeBilinearTest<DataType::QAsymmS8>,
959 DataLayout::NHWC)
960ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeBilinearInt8Nhwc,
961 AlignCornersResizeBilinearTest<DataType::QAsymmS8>,
962 DataLayout::NHWC)
963ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeBilinearUint8Nhwc,
964 HalfPixelCentersResizeBilinearTest<DataType::QAsymmU8>,
965 DataLayout::NHWC)
966ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeBilinearUint8Nhwc,
967 AlignCornersResizeBilinearTest<DataType::QAsymmU8>,
968 DataLayout::NHWC)
Teresa Charlin686f45c2020-07-23 17:59:00 +0100969
Ellen Norris-Thompson37e68682019-07-15 14:23:30 +0100970// Resize NearestNeighbor - NCHW
Finn Williamsec36d3e2020-08-28 13:17:05 +0100971ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeNearestNeighbor,
972 SimpleResizeNearestNeighborTest<DataType::Float32>,
973 DataLayout::NCHW)
974ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborNop,
975 ResizeNearestNeighborNopTest<DataType::Float32>,
976 DataLayout::NCHW)
977ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborSqMin,
978 ResizeNearestNeighborSqMinTest<DataType::Float32>,
979 DataLayout::NCHW)
980ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMin,
981 ResizeNearestNeighborMinTest<DataType::Float32>,
982 DataLayout::NCHW)
983ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMag,
984 ResizeNearestNeighborMagTest<DataType::Float32>,
985 DataLayout::NCHW, 0.1f, 50, 0.1f, 50)
Ellen Norris-Thompson37e68682019-07-15 14:23:30 +0100986
Finn Williamsec36d3e2020-08-28 13:17:05 +0100987ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeNearestNeighborUint8,
988 SimpleResizeNearestNeighborTest<DataType::QAsymmU8>,
989 DataLayout::NCHW)
990ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborNopUint8,
991 ResizeNearestNeighborNopTest<DataType::QAsymmU8>,
992 DataLayout::NCHW)
993ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborSqMinUint8,
994 ResizeNearestNeighborSqMinTest<DataType::QAsymmU8>,
995 DataLayout::NCHW)
996ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMinUint8,
997 ResizeNearestNeighborMinTest<DataType::QAsymmU8>,
998 DataLayout::NCHW)
999ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMagUint8,
1000 ResizeNearestNeighborMagTest<DataType::QAsymmU8>,
1001 DataLayout::NCHW, 0.1f, 50, 0.1f, 50)
Ellen Norris-Thompson37e68682019-07-15 14:23:30 +01001002
Finn Williamsec36d3e2020-08-28 13:17:05 +01001003ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeNearestNeighbour,
1004 HalfPixelCentersResizeNearestNeighbourTest<DataType::Float32>,
1005 DataLayout::NCHW)
1006ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeNearestNeighbour,
1007 AlignCornersResizeNearestNeighbourTest<DataType::Float32>,
1008 DataLayout::NCHW)
1009ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeNearestNeighbourInt8,
1010 HalfPixelCentersResizeNearestNeighbourTest<DataType::QAsymmS8>,
1011 DataLayout::NCHW)
1012ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeNearestNeighbourInt8,
1013 AlignCornersResizeNearestNeighbourTest<DataType::QAsymmS8>,
1014 DataLayout::NCHW)
1015ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeNearestNeighbourUint8,
1016 HalfPixelCentersResizeNearestNeighbourTest<DataType::QAsymmU8>,
1017 DataLayout::NCHW)
1018ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeNearestNeighbourUint8,
1019 AlignCornersResizeNearestNeighbourTest<DataType::QAsymmU8>,
1020 DataLayout::NCHW)
Teresa Charlin686f45c2020-07-23 17:59:00 +01001021
Ellen Norris-Thompson37e68682019-07-15 14:23:30 +01001022// Resize NearestNeighbor - NHWC
Finn Williamsec36d3e2020-08-28 13:17:05 +01001023ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborNopNhwc,
1024 ResizeNearestNeighborNopTest<DataType::Float32>,
1025 DataLayout::NHWC)
1026ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeNearestNeighborNhwc,
1027 SimpleResizeNearestNeighborTest<DataType::Float32>,
1028 DataLayout::NHWC)
1029ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborSqMinNhwc,
1030 ResizeNearestNeighborSqMinTest<DataType::Float32>,
1031 DataLayout::NHWC)
1032ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMinNhwc,
1033 ResizeNearestNeighborMinTest<DataType::Float32>,
1034 DataLayout::NHWC)
1035ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMagNhwc,
1036 ResizeNearestNeighborMagTest<DataType::Float32>,
1037 DataLayout::NHWC, 0.1f, 50, 0.1f, 50)
Ellen Norris-Thompson37e68682019-07-15 14:23:30 +01001038
Finn Williamsec36d3e2020-08-28 13:17:05 +01001039ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborNopUint8Nhwc,
1040 ResizeNearestNeighborNopTest<DataType::QAsymmU8>,
1041 DataLayout::NHWC)
1042ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeNearestNeighborUint8Nhwc,
1043 SimpleResizeNearestNeighborTest<DataType::QAsymmU8>,
1044 DataLayout::NHWC)
1045ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborSqMinUint8Nhwc,
1046 ResizeNearestNeighborSqMinTest<DataType::QAsymmU8>,
1047 DataLayout::NHWC)
1048ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMinUint8Nhwc,
1049 ResizeNearestNeighborMinTest<DataType::QAsymmU8>,
1050 DataLayout::NHWC)
1051ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMagUint8Nhwc,
1052 ResizeNearestNeighborMagTest<DataType::QAsymmU8>,
1053 DataLayout::NHWC, 0.1f, 50, 0.1f, 50)
Ellen Norris-Thompson37e68682019-07-15 14:23:30 +01001054
Finn Williamsec36d3e2020-08-28 13:17:05 +01001055ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeNearestNeighbourNhwc,
1056 HalfPixelCentersResizeNearestNeighbourTest<DataType::Float32>,
1057 DataLayout::NHWC)
1058ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeNearestNeighbourNhwc,
1059 AlignCornersResizeNearestNeighbourTest<DataType::Float32>,
1060 DataLayout::NHWC)
1061ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeNearestNeighbourInt8Nhwc,
1062 HalfPixelCentersResizeNearestNeighbourTest<DataType::QAsymmS8>,
1063 DataLayout::NHWC)
1064ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeNearestNeighbourInt8Nhwc,
1065 AlignCornersResizeNearestNeighbourTest<DataType::QAsymmS8>,
1066 DataLayout::NHWC)
1067ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeNearestNeighbourUint8Nhwc,
1068 HalfPixelCentersResizeNearestNeighbourTest<DataType::QAsymmU8>,
1069 DataLayout::NHWC)
1070ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeNearestNeighbourUint8Nhwc,
1071 AlignCornersResizeNearestNeighbourTest<DataType::QAsymmU8>,
1072 DataLayout::NHWC)
Teresa Charlin686f45c2020-07-23 17:59:00 +01001073
josh minor036f02d2019-11-15 14:53:22 -06001074// Slice
1075ARMNN_AUTO_TEST_CASE(Slice4dFloat32, Slice4dFloat32Test)
1076ARMNN_AUTO_TEST_CASE(Slice3dFloat32, Slice3dFloat32Test)
1077ARMNN_AUTO_TEST_CASE(Slice2dFloat32, Slice2dFloat32Test)
1078ARMNN_AUTO_TEST_CASE(Slice1dFloat32, Slice1dFloat32Test)
1079ARMNN_AUTO_TEST_CASE(Slice4dUint8, Slice4dUint8Test)
1080ARMNN_AUTO_TEST_CASE(Slice3dUint8, Slice3dUint8Test)
1081ARMNN_AUTO_TEST_CASE(Slice2dUint8, Slice2dUint8Test)
1082ARMNN_AUTO_TEST_CASE(Slice1dUint8, Slice1dUint8Test)
1083ARMNN_AUTO_TEST_CASE(Slice4dInt16, Slice4dInt16Test)
1084ARMNN_AUTO_TEST_CASE(Slice3dInt16, Slice3dInt16Test)
1085ARMNN_AUTO_TEST_CASE(Slice2dInt16, Slice2dInt16Test)
1086ARMNN_AUTO_TEST_CASE(Slice1dInt16, Slice1dInt16Test)
1087
FinnWilliamsArm1fa19192019-08-02 17:26:31 +01001088// Strided Slice
Sadik Armagan5e719842020-08-28 12:17:33 +01001089ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice4dFloat32, StridedSlice4dFloat32Test)
1090ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice4dReverseFloat32, StridedSlice4dReverseFloat32Test)
1091ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSliceSimpleStrideFloat32, StridedSliceSimpleStrideFloat32Test)
1092ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSliceSimpleRangeMaskFloat32, StridedSliceSimpleRangeMaskFloat32Test)
1093ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSliceShrinkAxisMaskFloat32, StridedSliceShrinkAxisMaskFloat32Test)
Finn Williams826a5432020-08-27 16:15:20 +01001094ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSliceShrinkAxisMaskCTSFloat32,
1095 StridedSliceShrinkAxisMaskCTSFloat32Test)
Sadik Armagan5e719842020-08-28 12:17:33 +01001096ARMNN_AUTO_TEST_CASE_WITH_THF(
1097 StridedSliceShrinkAxisMaskBitPosition0Dim3Float32, StridedSliceShrinkAxisMaskBitPosition0Dim3Float32Test)
1098ARMNN_AUTO_TEST_CASE_WITH_THF(
1099 StridedSliceShrinkAxisMaskBitPosition0Float32, StridedSliceShrinkAxisMaskBitPosition0Float32Test)
1100ARMNN_AUTO_TEST_CASE_WITH_THF(
1101 StridedSliceShrinkAxisMaskBitPosition1Float32, StridedSliceShrinkAxisMaskBitPosition1Float32Test)
1102ARMNN_AUTO_TEST_CASE_WITH_THF(
1103 StridedSliceShrinkAxisMaskBitPosition2Float32, StridedSliceShrinkAxisMaskBitPosition2Float32Test)
1104ARMNN_AUTO_TEST_CASE_WITH_THF(
1105 StridedSliceShrinkAxisMaskBitPosition3Float32, StridedSliceShrinkAxisMaskBitPosition3Float32Test)
1106ARMNN_AUTO_TEST_CASE_WITH_THF(
1107 StridedSliceShrinkAxisMaskBitPosition0And1Float32, StridedSliceShrinkAxisMaskBitPosition0And1Float32Test)
1108ARMNN_AUTO_TEST_CASE_WITH_THF(
1109 StridedSliceShrinkAxisMaskBitPosition0And2Float32, StridedSliceShrinkAxisMaskBitPosition0And2Float32Test)
1110ARMNN_AUTO_TEST_CASE_WITH_THF(
1111 StridedSliceShrinkAxisMaskBitPosition0And3Float32, StridedSliceShrinkAxisMaskBitPosition0And3Float32Test)
1112ARMNN_AUTO_TEST_CASE_WITH_THF(
1113 StridedSliceShrinkAxisMaskBitPosition0And1And3Float32, StridedSliceShrinkAxisMaskBitPosition0And1And3Float32Test)
1114ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice3dFloat32, StridedSlice3dFloat32Test)
1115ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice3dReverseFloat32, StridedSlice3dReverseFloat32Test)
1116ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice2dFloat32, StridedSlice2dFloat32Test)
1117ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice2dReverseFloat32, StridedSlice2dReverseFloat32Test)
FinnWilliamsArm1fa19192019-08-02 17:26:31 +01001118
Sadik Armagan5e719842020-08-28 12:17:33 +01001119ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice4dUint8, StridedSlice4dUint8Test)
1120ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice4dReverseUint8, StridedSlice4dReverseUint8Test)
1121ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSliceSimpleStrideUint8, StridedSliceSimpleStrideUint8Test)
1122ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSliceSimpleRangeMaskUint8, StridedSliceSimpleRangeMaskUint8Test)
1123ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSliceShrinkAxisMaskUint8, StridedSliceShrinkAxisMaskUint8Test)
1124ARMNN_AUTO_TEST_CASE_WITH_THF(
1125 StridedSliceShrinkAxisMaskBitPosition0Dim3Uint8, StridedSliceShrinkAxisMaskBitPosition0Dim3Uint8Test)
1126ARMNN_AUTO_TEST_CASE_WITH_THF(
1127 StridedSliceShrinkAxisMaskBitPosition0Uint8, StridedSliceShrinkAxisMaskBitPosition0Uint8Test)
1128ARMNN_AUTO_TEST_CASE_WITH_THF(
1129 StridedSliceShrinkAxisMaskBitPosition1Uint8, StridedSliceShrinkAxisMaskBitPosition1Uint8Test)
1130ARMNN_AUTO_TEST_CASE_WITH_THF(
1131 StridedSliceShrinkAxisMaskBitPosition2Uint8, StridedSliceShrinkAxisMaskBitPosition2Uint8Test)
1132ARMNN_AUTO_TEST_CASE_WITH_THF(
1133 StridedSliceShrinkAxisMaskBitPosition3Uint8, StridedSliceShrinkAxisMaskBitPosition3Uint8Test)
1134ARMNN_AUTO_TEST_CASE_WITH_THF(
1135 StridedSliceShrinkAxisMaskBitPosition0And1Uint8, StridedSliceShrinkAxisMaskBitPosition0And1Uint8Test)
1136ARMNN_AUTO_TEST_CASE_WITH_THF(
1137 StridedSliceShrinkAxisMaskBitPosition0And2Uint8, StridedSliceShrinkAxisMaskBitPosition0And2Uint8Test)
1138ARMNN_AUTO_TEST_CASE_WITH_THF(
1139 StridedSliceShrinkAxisMaskBitPosition0And3Uint8, StridedSliceShrinkAxisMaskBitPosition0And3Uint8Test)
1140ARMNN_AUTO_TEST_CASE_WITH_THF(
1141 StridedSliceShrinkAxisMaskBitPosition0And1And3Uint8, StridedSliceShrinkAxisMaskBitPosition0And1And3Uint8Test)
1142ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice3dUint8, StridedSlice3dUint8Test)
1143ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice3dReverseUint8, StridedSlice3dReverseUint8Test)
1144ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice2dUint8, StridedSlice2dUint8Test)
1145ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice2dReverseUint8, StridedSlice2dReverseUint8Test)
FinnWilliamsArm1fa19192019-08-02 17:26:31 +01001146
Sadik Armaganfabc2892019-05-31 09:05:11 +01001147// Quantize
Finn Williamsec36d3e2020-08-28 13:17:05 +01001148ARMNN_AUTO_TEST_CASE_WITH_THF(QuantizeSimpleUint8, QuantizeSimpleUint8Test)
1149ARMNN_AUTO_TEST_CASE_WITH_THF(QuantizeClampUint8, QuantizeClampUint8Test)
Sadik Armaganfabc2892019-05-31 09:05:11 +01001150
Nikhil Raj9b461482019-07-03 15:58:31 +01001151// PReLU
Finn Williams826a5432020-08-27 16:15:20 +01001152ARMNN_AUTO_TEST_CASE_WITH_THF(PreluFloat32, PreluTest<DataType::Float32>)
1153ARMNN_AUTO_TEST_CASE_WITH_THF(PreluUint8, PreluTest<DataType::QAsymmU8>)
Nikhil Raj9b461482019-07-03 15:58:31 +01001154
Matthew Jackson87f65ea2019-08-01 10:01:34 +01001155// Stack
Sadik Armagan5e719842020-08-28 12:17:33 +01001156ARMNN_AUTO_TEST_CASE_WITH_THF(Stack0Axis, StackAxis0Float32Test)
1157ARMNN_AUTO_TEST_CASE_WITH_THF(StackOutput4DAxis1, StackOutput4DAxis1Float32Test)
1158ARMNN_AUTO_TEST_CASE_WITH_THF(StackOutput4DAxis2, StackOutput4DAxis2Float32Test)
1159ARMNN_AUTO_TEST_CASE_WITH_THF(StackOutput4DAxis3, StackOutput4DAxis3Float32Test)
1160ARMNN_AUTO_TEST_CASE_WITH_THF(StackOutput3DInputs3, StackOutput3DInputs3Float32Test)
1161ARMNN_AUTO_TEST_CASE_WITH_THF(StackOutput5D, StackOutput5DFloat32Test)
Matthew Jackson87f65ea2019-08-01 10:01:34 +01001162
Mike Kellyc9ea45a2020-02-28 18:11:58 +00001163// Transpose
Finn Williamsec36d3e2020-08-28 13:17:05 +01001164ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleTransposeFloat32, SimpleTransposeTest<DataType::Float32>)
1165ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeFloat32ValueSet1Test, TransposeValueSet1Test<DataType::Float32>)
1166ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeFloat32ValueSet2Test, TransposeValueSet2Test<DataType::Float32>)
1167ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeFloat32ValueSet3Test, TransposeValueSet3Test<DataType::Float32>)
1168ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleTransposeQASymms8, SimpleTransposeTest<DataType::QAsymmS8>)
1169ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeQASymmS8ValueSet1Test, TransposeValueSet1Test<DataType::QAsymmS8>)
1170ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeQASymmS8ValueSet2Test, TransposeValueSet2Test<DataType::QAsymmS8>)
1171ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeQASymmS8ValueSet3Test, TransposeValueSet3Test<DataType::QAsymmS8>)
1172ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleTransposeQASymm8, SimpleTransposeTest<DataType::QAsymmU8>)
1173ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeQASymm8ValueSet1Test, TransposeValueSet1Test<DataType::QAsymmU8>)
1174ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeQASymm8ValueSet2Test, TransposeValueSet2Test<DataType::QAsymmU8>)
1175ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeQASymm8ValueSet3Test, TransposeValueSet3Test<DataType::QAsymmU8>)
1176ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleTransposeQSymm16, SimpleTransposeTest<DataType::QSymmS16>)
1177ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeQSymm16ValueSet1Test, TransposeValueSet1Test<DataType::QSymmS16>)
1178ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeQSymm16ValueSet2Test, TransposeValueSet2Test<DataType::QSymmS16>)
1179ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeQSymm16ValueSet3Test, TransposeValueSet3Test<DataType::QSymmS16>)
Mike Kellyc9ea45a2020-02-28 18:11:58 +00001180
Sadik Armagan581742d2019-08-12 14:11:37 +01001181// TransposeConvolution2d
Finn Williamsec36d3e2020-08-28 13:17:05 +01001182ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleTransposeConvolution2dFloatNchw,
1183 SimpleTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1184 true,
1185 DataLayout::NCHW)
1186ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleTransposeConvolution2dFloatNhwc,
1187 SimpleTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1188 true,
1189 DataLayout::NHWC)
1190ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleTransposeConvolution2dUint8Nchw,
1191 SimpleTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1192 true,
1193 DataLayout::NCHW)
1194ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleTransposeConvolution2dUint8Nhwc,
1195 SimpleTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1196 true,
1197 DataLayout::NHWC)
Sadik Armagan581742d2019-08-12 14:11:37 +01001198
Finn Williamsec36d3e2020-08-28 13:17:05 +01001199ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedSimpleTransposeConvolution2dFloatNchw,
1200 SimpleTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1201 false,
1202 DataLayout::NCHW)
1203ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedSimpleTransposeConvolution2dFloatNhwc,
1204 SimpleTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1205 true,
1206 DataLayout::NHWC)
1207ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedSimpleTransposeConvolution2dUint8Nchw,
1208 SimpleTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1209 true,
1210 DataLayout::NCHW)
1211ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedSimpleTransposeConvolution2dUint8Nhwc,
1212 SimpleTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1213 true,
1214 DataLayout::NHWC)
Sadik Armagan581742d2019-08-12 14:11:37 +01001215
Finn Williamsec36d3e2020-08-28 13:17:05 +01001216ARMNN_AUTO_TEST_CASE_WITH_THF(PaddedTransposeConvolution2dFloatNchw,
1217 PaddedTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1218 true,
1219 DataLayout::NCHW)
1220ARMNN_AUTO_TEST_CASE_WITH_THF(PaddedTransposeConvolution2dFloatNhwc,
1221 PaddedTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1222 true,
1223 DataLayout::NHWC)
1224ARMNN_AUTO_TEST_CASE_WITH_THF(PaddedTransposeConvolution2dUint8Nchw,
1225 PaddedTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1226 true,
1227 DataLayout::NCHW)
1228ARMNN_AUTO_TEST_CASE_WITH_THF(PaddedTransposeConvolution2dUint8Nhwc,
1229 PaddedTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1230 true,
1231 DataLayout::NHWC)
Sadik Armagan581742d2019-08-12 14:11:37 +01001232
Finn Williamsec36d3e2020-08-28 13:17:05 +01001233ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedPaddedTransposeConvolution2dFloatNchw,
1234 PaddedTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1235 false,
1236 DataLayout::NCHW)
1237ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedPaddedTransposeConvolution2dFloatNhwc,
1238 PaddedTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1239 true,
1240 DataLayout::NHWC)
1241ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedPaddedTransposeConvolution2dInt8Nchw,
1242 PaddedTransposeConvolution2dTest<DataType::QAsymmS8, DataType::Signed32>,
1243 true,
1244 DataLayout::NCHW)
1245ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedPaddedTransposeConvolution2dInt8Nhwc,
1246 PaddedTransposeConvolution2dTest<DataType::QAsymmS8, DataType::Signed32>,
1247 true,
1248 DataLayout::NHWC)
1249ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedPaddedTransposeConvolution2dUint8Nchw,
1250 PaddedTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1251 true,
1252 DataLayout::NCHW)
1253ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedPaddedTransposeConvolution2dUint8Nhwc,
1254 PaddedTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1255 true,
1256 DataLayout::NHWC)
Sadik Armagan581742d2019-08-12 14:11:37 +01001257
Finn Williamsec36d3e2020-08-28 13:17:05 +01001258ARMNN_AUTO_TEST_CASE_WITH_THF(StridedTransposeConvolution2dFloatNchw,
1259 StridedTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1260 true,
1261 DataLayout::NCHW)
1262ARMNN_AUTO_TEST_CASE_WITH_THF(StridedTransposeConvolution2dFloatNhwc,
1263 StridedTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1264 true,
1265 DataLayout::NHWC)
1266ARMNN_AUTO_TEST_CASE_WITH_THF(StridedTransposeConvolution2dUint8Nchw,
1267 StridedTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1268 true,
1269 DataLayout::NCHW)
1270ARMNN_AUTO_TEST_CASE_WITH_THF(StridedTransposeConvolution2dUint8Nhwc,
1271 StridedTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1272 true,
1273 DataLayout::NHWC)
Sadik Armagan581742d2019-08-12 14:11:37 +01001274
Finn Williamsec36d3e2020-08-28 13:17:05 +01001275ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedStridedTransposeConvolution2dFloatNchw,
1276 StridedTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1277 false,
1278 DataLayout::NCHW)
1279ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedStridedTransposeConvolution2dFloatNhwc,
1280 StridedTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1281 true,
1282 DataLayout::NHWC)
1283ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedStridedTransposeConvolution2dUint8Nchw,
1284 StridedTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1285 true,
1286 DataLayout::NCHW)
1287ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedStridedTransposeConvolution2dUint8Nhwc,
1288 StridedTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1289 true,
1290 DataLayout::NHWC)
Sadik Armagan581742d2019-08-12 14:11:37 +01001291
Finn Williamsec36d3e2020-08-28 13:17:05 +01001292ARMNN_AUTO_TEST_CASE_WITH_THF(MultiChannelTransposeConvolution2dFloatNchw,
1293 MultiChannelTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1294 DataLayout::NCHW)
1295ARMNN_AUTO_TEST_CASE_WITH_THF(MultiChannelTransposeConvolution2dFloatNhwc,
1296 MultiChannelTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1297 DataLayout::NHWC)
1298ARMNN_AUTO_TEST_CASE_WITH_THF(MultiChannelTransposeConvolution2dUint8Nchw,
1299 MultiChannelTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1300 DataLayout::NCHW)
1301ARMNN_AUTO_TEST_CASE_WITH_THF(MultiChannelTransposeConvolution2dUint8Nhwc,
1302 MultiChannelTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1303 DataLayout::NHWC)
Aron Virginas-Tar7599aa42019-08-20 15:30:04 +01001304
Aron Virginas-Tar914e4db2019-09-09 13:36:45 +01001305// Abs
Keith Davis33a626f2020-08-27 15:38:12 +01001306ARMNN_AUTO_TEST_CASE_WITH_THF(Abs2d, Abs2dTest<DataType::Float32>)
1307ARMNN_AUTO_TEST_CASE_WITH_THF(Abs3d, Abs3dTest<DataType::Float32>)
1308ARMNN_AUTO_TEST_CASE_WITH_THF(AbsZero, AbsZeroTest<DataType::Float32>)
Aron Virginas-Tar914e4db2019-09-09 13:36:45 +01001309
Keith Davis33a626f2020-08-27 15:38:12 +01001310ARMNN_AUTO_TEST_CASE_WITH_THF(Abs2dSigned32, Abs2dTest<DataType::Signed32>)
1311ARMNN_AUTO_TEST_CASE_WITH_THF(Abs3dSigned32, Abs3dTest<DataType::Signed32>)
1312ARMNN_AUTO_TEST_CASE_WITH_THF(AbsZeroSigned32, AbsZeroTest<DataType::Signed32>)
Kevin Mayec52c3a2020-04-24 09:42:31 +01001313
Aron Virginas-Tar0dd3b432019-09-10 13:55:09 +01001314// Rsqrt
Keith Davis33a626f2020-08-27 15:38:12 +01001315ARMNN_AUTO_TEST_CASE_WITH_THF(Rsqrt2d, Rsqrt2dTest<DataType::Float32>)
1316ARMNN_AUTO_TEST_CASE_WITH_THF(Rsqrt3d, Rsqrt3dTest<DataType::Float32>)
1317ARMNN_AUTO_TEST_CASE_WITH_THF(RsqrtZero, RsqrtZeroTest<DataType::Float32>)
1318ARMNN_AUTO_TEST_CASE_WITH_THF(RsqrtNegative, RsqrtNegativeTest<DataType::Float32>)
Aron Virginas-Tar0dd3b432019-09-10 13:55:09 +01001319
James Conroyd47a0642019-09-17 14:22:06 +01001320// ArgMinMax
Sadik Armagan43528ac2020-08-27 09:06:40 +01001321ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinFloat32, ArgMinSimpleTest<DataType::Float32>)
1322ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxFloat32, ArgMaxSimpleTest<DataType::Float32>)
1323ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinChannel, ArgMinChannelTest<DataType::Float32>)
1324ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxChannel, ArgMaxChannelTest<DataType::Float32>)
1325ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxHeight, ArgMaxHeightTest<DataType::Float32>)
1326ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinWidth, ArgMinWidthTest<DataType::Float32>)
James Conroyd47a0642019-09-17 14:22:06 +01001327
Sadik Armagan43528ac2020-08-27 09:06:40 +01001328ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinQAsymmS8, ArgMinSimpleTest<DataType::QAsymmS8>)
1329ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxQAsymmS8, ArgMaxSimpleTest<DataType::QAsymmS8>)
1330ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinChannelQAsymmS8, ArgMinChannelTest<DataType::QAsymmS8>)
1331ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxChannelQAsymmS8, ArgMaxChannelTest<DataType::QAsymmS8>)
1332ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxHeightQAsymmS8, ArgMaxHeightTest<DataType::QAsymmS8>)
1333ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinWidthQAsymmS8, ArgMinWidthTest<DataType::QAsymmS8>)
Sadik Armagan303980c2020-04-17 12:45:14 +01001334
Sadik Armagan43528ac2020-08-27 09:06:40 +01001335ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinQAsymm8, ArgMinSimpleTest<DataType::QAsymmU8>)
1336ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxQAsymm8, ArgMaxSimpleTest<DataType::QAsymmU8>)
1337ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinChannelQAsymm8, ArgMinChannelTest<DataType::QAsymmU8>)
1338ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxChannelQAsymm8, ArgMaxChannelTest<DataType::QAsymmU8>)
1339ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxHeightQAsymm8, ArgMaxHeightTest<DataType::QAsymmU8>)
1340ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinWidthQAsymm8, ArgMinWidthTest<DataType::QAsymmU8>)
Francis Murtagh52ec3462019-11-19 12:24:19 +00001341
Sadik Armaganac472102020-03-24 09:54:36 +00001342// Neg
Keith Davis33a626f2020-08-27 15:38:12 +01001343ARMNN_AUTO_TEST_CASE_WITH_THF(Neg2d, Neg2dTest<DataType::Float32>)
1344ARMNN_AUTO_TEST_CASE_WITH_THF(Neg3d, Neg3dTest<DataType::Float32>)
1345ARMNN_AUTO_TEST_CASE_WITH_THF(NegZero, NegZeroTest<DataType::Float32>)
1346ARMNN_AUTO_TEST_CASE_WITH_THF(NegNegative, NegNegativeTest<DataType::Float32>)
Sadik Armaganac472102020-03-24 09:54:36 +00001347
Sadik Armagan9fabf432020-05-27 13:40:58 +01001348// Exp
Keith Davis33a626f2020-08-27 15:38:12 +01001349ARMNN_AUTO_TEST_CASE_WITH_THF(Exp2d, Exp2dTest<DataType::Float32>)
1350ARMNN_AUTO_TEST_CASE_WITH_THF(Exo3d, Exp3dTest<DataType::Float32>)
1351ARMNN_AUTO_TEST_CASE_WITH_THF(ExpZero, ExpZeroTest<DataType::Float32>)
1352ARMNN_AUTO_TEST_CASE_WITH_THF(ExpNegative, ExpNegativeTest<DataType::Float32>)
Sadik Armagan9fabf432020-05-27 13:40:58 +01001353
Sadik Armagana792a052020-06-23 16:22:23 +01001354// Fill
Finn Williams87020072020-08-26 16:19:15 +01001355ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleFill, SimpleFillTest<DataType::Float32>)
1356ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleFillF16, SimpleFillTest<DataType::Float16>)
1357ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleFillS32, SimpleFillTest<DataType::Signed32>)
Sadik Armagana792a052020-06-23 16:22:23 +01001358
James Conroy177df1e2020-11-13 10:18:51 +00001359// Logical
1360ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalNot, LogicalNotTest)
1361ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalNotInt, LogicalNotIntTest)
1362
1363ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalAnd, LogicalAndTest)
1364ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalAndInt, LogicalAndIntTest)
1365ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalAndBroadcast1, LogicalAndBroadcast1Test)
1366ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalAndBroadcast2, LogicalAndBroadcast2Test)
1367ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalAndBroadcast3, LogicalAndBroadcast3Test)
1368
1369ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalOr, LogicalOrTest)
1370ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalOrInt, LogicalOrIntTest)
1371ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalOrBroadcast1, LogicalOrBroadcast1Test)
1372ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalOrBroadcast2, LogicalOrBroadcast2Test)
1373ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalOrBroadcast3, LogicalOrBroadcast3Test)
1374
Matteo Martincighdb16dd32019-08-27 16:41:11 +01001375#if defined(ARMNNREF_ENABLED)
Matteo Martincighe67edb22019-08-14 14:05:46 +01001376
1377// The ARMNN_COMPARE_REF_AUTO_TEST_CASE and the ARMNN_COMPARE_REF_FIXTURE_TEST_CASE test units are not available
1378// if the reference backend is not built
1379
telsoa014fcda012018-03-09 14:13:49 +00001380// ============================================================================
1381// COMPARE tests
1382
Keith Davisf500d6c2020-08-31 08:32:55 +01001383ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareConv2dWithReference, CompareConvolution2dTest)
telsoa014fcda012018-03-09 14:13:49 +00001384
Keith Davisf500d6c2020-08-31 08:32:55 +01001385ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareDepthwiseConv2dWithReferenceFloat32,
Nattapat Chaimanowong649dd952019-01-22 16:10:44 +00001386 CompareDepthwiseConvolution2dFloatTest,
Sadik Armagan581742d2019-08-12 14:11:37 +01001387 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +01001388ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareDepthwiseConv2dWithReferenceUint8,
Nattapat Chaimanowong649dd952019-01-22 16:10:44 +00001389 CompareDepthwiseConvolution2dUint8Test,
Sadik Armagan581742d2019-08-12 14:11:37 +01001390 DataLayout::NCHW)
jimfly017af00da2018-10-31 14:43:53 +00001391
Keith Davisf500d6c2020-08-31 08:32:55 +01001392ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareDepthwiseConv2dWithReferenceFloat32Nhwc,
Nattapat Chaimanowong649dd952019-01-22 16:10:44 +00001393 CompareDepthwiseConvolution2dFloatTest,
Sadik Armagan581742d2019-08-12 14:11:37 +01001394 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +01001395ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareDepthwiseConv2dWithReferenceUint8Nhwc,
Nattapat Chaimanowong649dd952019-01-22 16:10:44 +00001396 CompareDepthwiseConvolution2dUint8Test,
Sadik Armagan581742d2019-08-12 14:11:37 +01001397 DataLayout::NHWC)
telsoa014fcda012018-03-09 14:13:49 +00001398
Finn Williams826a5432020-08-27 16:15:20 +01001399ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareNormalizationWithinWithReference, CompareNormalizationTest,
1400 NormalizationAlgorithmChannel::Within,
1401 NormalizationAlgorithmMethod::LocalBrightness)
1402ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareNormalizationAcrossWithReference, CompareNormalizationTest,
1403 NormalizationAlgorithmChannel::Across,
1404 NormalizationAlgorithmMethod::LocalBrightness)
telsoa014fcda012018-03-09 14:13:49 +00001405
Finn Williams826a5432020-08-27 16:15:20 +01001406ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareMaxPooling2dWithReference, ComparePooling2dTest, PoolingAlgorithm::Max)
1407ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareMaxPooling2dWithReferenceUint8, ComparePooling2dUint8Test,
1408 PoolingAlgorithm::Max)
1409ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareAveragePooling2dWithReference, ComparePooling2dTest,
1410 PoolingAlgorithm::Average)
1411ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareAveragePooling2dWithReferenceUint8, ComparePooling2dUint8Test,
1412 PoolingAlgorithm::Average)
1413ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareL2Pooling2dWithReference, ComparePooling2dTest, PoolingAlgorithm::L2)
1414ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(UNSUPPORTED_CompareL2Pooling2dWithReferenceUint8, ComparePooling2dUint8Test,
1415 PoolingAlgorithm::L2)
telsoa014fcda012018-03-09 14:13:49 +00001416
Sadik Armagan56785c72020-08-27 12:57:20 +01001417ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareSoftmaxBeta1WithReference, CompareSoftmaxTest, 1.0f)
1418ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareSoftmaxBeta2WithReference, CompareSoftmaxTest, 2.0f)
telsoa014fcda012018-03-09 14:13:49 +00001419
Sadik Armagan56785c72020-08-27 12:57:20 +01001420ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareSoftmaxUint8Beta1WithReference, CompareSoftmaxUint8Test, 1.0f)
1421ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareSoftmaxUint8Beta2WithReference, CompareSoftmaxUint8Test, 2.0f)
telsoa014fcda012018-03-09 14:13:49 +00001422
Keith Davis33a626f2020-08-27 15:38:12 +01001423ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareAddition, CompareAdditionTest)
telsoa014fcda012018-03-09 14:13:49 +00001424
Keith Davis33a626f2020-08-27 15:38:12 +01001425ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareMultiplicationWithReference, CompareMultiplicationTest)
telsoa014fcda012018-03-09 14:13:49 +00001426
Keith Davis33a626f2020-08-27 15:38:12 +01001427ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareBatchNorm, CompareBatchNormTest)
telsoa014fcda012018-03-09 14:13:49 +00001428
Keith Davis33a626f2020-08-27 15:38:12 +01001429ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareReLu1, CompareBoundedReLuTest, 1.0f, -1.0f)
1430ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareReLu6, CompareBoundedReLuTest, 6.0f, 0.0f)
telsoa014fcda012018-03-09 14:13:49 +00001431
1432// ============================================================================
1433// FIXTURE tests
1434
Keith Davis33a626f2020-08-27 15:38:12 +01001435ARMNN_COMPARE_REF_FIXTURE_TEST_CASE_WITH_THF(CompareSigmoidActivationWithReference, ActivationFixture,
Sadik Armagan581742d2019-08-12 14:11:37 +01001436 CompareActivationTest, ActivationFunction::Sigmoid, 5u)
telsoa014fcda012018-03-09 14:13:49 +00001437
Keith Davis33a626f2020-08-27 15:38:12 +01001438ARMNN_COMPARE_REF_FIXTURE_TEST_CASE_WITH_THF(CompareTanhActivationWithReference, ActivationFixture,
Sadik Armagan581742d2019-08-12 14:11:37 +01001439 CompareActivationTest, ActivationFunction::TanH, 5u)
telsoa014fcda012018-03-09 14:13:49 +00001440
Keith Davis33a626f2020-08-27 15:38:12 +01001441ARMNN_COMPARE_REF_FIXTURE_TEST_CASE_WITH_THF(CompareLinearActivationWithReference, ActivationFixture,
Sadik Armagan581742d2019-08-12 14:11:37 +01001442 CompareActivationTest, ActivationFunction::Linear, 5u)
telsoa014fcda012018-03-09 14:13:49 +00001443
Keith Davis33a626f2020-08-27 15:38:12 +01001444ARMNN_COMPARE_REF_FIXTURE_TEST_CASE_WITH_THF(CompareReLuActivationWithReference, ActivationFixture,
Sadik Armagan581742d2019-08-12 14:11:37 +01001445 CompareActivationTest, ActivationFunction::ReLu, 5u)
telsoa014fcda012018-03-09 14:13:49 +00001446
Keith Davis33a626f2020-08-27 15:38:12 +01001447ARMNN_COMPARE_REF_FIXTURE_TEST_CASE_WITH_THF(CompareBoundedReLuActivationWithReference, ActivationFixture,
Sadik Armagan581742d2019-08-12 14:11:37 +01001448 CompareActivationTest, ActivationFunction::BoundedReLu, 5u)
Keith Davis33a626f2020-08-27 15:38:12 +01001449ARMNN_COMPARE_REF_FIXTURE_TEST_CASE_WITH_THF(CompareBoundedReLuActivationWithReferenceUint8, ActivationFixture,
Sadik Armagan581742d2019-08-12 14:11:37 +01001450 CompareActivationUint8Test, ActivationFunction::BoundedReLu)
telsoa014fcda012018-03-09 14:13:49 +00001451
Keith Davis33a626f2020-08-27 15:38:12 +01001452ARMNN_COMPARE_REF_FIXTURE_TEST_CASE_WITH_THF(CompareSoftReLuActivationWithReference, ActivationFixture,
Sadik Armagan581742d2019-08-12 14:11:37 +01001453 CompareActivationTest, ActivationFunction::SoftReLu, 1u)
telsoa014fcda012018-03-09 14:13:49 +00001454
Keith Davis33a626f2020-08-27 15:38:12 +01001455ARMNN_COMPARE_REF_FIXTURE_TEST_CASE_WITH_THF(CompareLeakyReLuActivationWithReference, ActivationFixture,
Sadik Armagan581742d2019-08-12 14:11:37 +01001456 CompareActivationTest, ActivationFunction::LeakyReLu, 5u)
telsoa014fcda012018-03-09 14:13:49 +00001457
Keith Davis33a626f2020-08-27 15:38:12 +01001458ARMNN_COMPARE_REF_FIXTURE_TEST_CASE_WITH_THF(CompareAbsActivationWithReference, ActivationFixture,
Sadik Armagan581742d2019-08-12 14:11:37 +01001459 CompareActivationTest, ActivationFunction::Abs, 5u)
telsoa014fcda012018-03-09 14:13:49 +00001460
Keith Davis33a626f2020-08-27 15:38:12 +01001461ARMNN_COMPARE_REF_FIXTURE_TEST_CASE_WITH_THF(CompareSqrtActivationWithReference, PositiveActivationFixture,
Sadik Armagan581742d2019-08-12 14:11:37 +01001462 CompareActivationTest, ActivationFunction::Sqrt, 5u)
telsoa014fcda012018-03-09 14:13:49 +00001463
Keith Davis33a626f2020-08-27 15:38:12 +01001464ARMNN_COMPARE_REF_FIXTURE_TEST_CASE_WITH_THF(CompareSquareActivationWithReference, ActivationFixture,
Sadik Armagan581742d2019-08-12 14:11:37 +01001465 CompareActivationTest, ActivationFunction::Square, 5u)
Matteo Martincighe67edb22019-08-14 14:05:46 +01001466
Keith Davis33a626f2020-08-27 15:38:12 +01001467ARMNN_COMPARE_REF_FIXTURE_TEST_CASE_WITH_THF(CompareEluActivationWithReference, ActivationFixture,
Teresa Charline32c8442020-03-05 14:41:50 +00001468 CompareActivationTest, ActivationFunction::Elu, 5u)
1469
Matteo Martincighe67edb22019-08-14 14:05:46 +01001470#endif
1471
telsoa014fcda012018-03-09 14:13:49 +00001472BOOST_AUTO_TEST_SUITE_END()