Gitiles
Code Review Sign In
review.mlplatform.org / ml / armnn / 34c1c38944b47b881febdfb9f98103dbdc949ed0 / . / src / backends / neon / test / NeonLayerTests.cpp
blob: c0b995baaaf3fdc29e5810b398ce958fb965fd9f [file] [log] [blame]
Laurent Carlier749294b2020-06-01 09:03:17 +0100[diff] [blame]1//
Teresa Charlinca5c82a2023-03-28 11:00:36 +0100[diff] [blame]2// Copyright © 2017-2023 Arm Ltd and Contributors. All rights reserved.
David Beckecb56cd2018-09-05 12:52:57 +0100[diff] [blame]3// SPDX-License-Identifier: MIT
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]4//
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]5
Aron Virginas-Tar56055192018-11-12 18:10:43 +0000[diff] [blame]6#include "NeonWorkloadFactoryHelper.hpp"
7
Colm Donelanc42a9872022-02-02 16:35:09 +0000[diff] [blame]8#include <armnnTestUtils/TensorHelpers.hpp>
Sadik Armagana097d2a2021-11-24 15:47:28 +0000[diff] [blame]9#include <UnitTests.hpp>
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]10
Aron Virginas-Tarc9cc8042018-11-01 16:15:57 +0000[diff] [blame]11#include <neon/NeonLayerSupport.hpp>
12#include <neon/NeonWorkloadFactory.hpp>
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]13
Aron Virginas-Tarc9cc8042018-11-01 16:15:57 +0000[diff] [blame]14#include <backendsCommon/test/ActivationFixture.hpp>
15#include <backendsCommon/test/LayerTests.hpp>
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]16
Sadik Armagan1625efc2021-06-10 18:24:34 +0100[diff] [blame]17#include <doctest/doctest.h>
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]18
Sadik Armagan1625efc2021-06-10 18:24:34 +0100[diff] [blame]19TEST_SUITE("Compute_ArmComputeNeon")
20{
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]21using namespace armnn;
22
23using FactoryType = NeonWorkloadFactory;
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]24
25// ============================================================================
26// UNIT tests
27
Mike Kelly56858022020-01-27 12:14:47 +0000[diff] [blame]28// BatchToSpace
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]29ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcFloat321, BatchToSpaceNdNhwcTest1<DataType::Float32>)
30ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcFloat322, BatchToSpaceNdNhwcTest2<DataType::Float32>)
31ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcFloat323, BatchToSpaceNdNhwcTest3<DataType::Float32>)
Teresa Charlinca5c82a2023-03-28 11:00:36 +0100[diff] [blame]32ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcFloat324, BatchToSpaceNdNhwcTest4<DataType::Float32>)
33ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcFloat325, BatchToSpaceNdNhwcTest5<DataType::Float32>)
34ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcFloat326, BatchToSpaceNdNhwcTest6<DataType::Float32>)
35ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcFloat327, BatchToSpaceNdNhwcTest7<DataType::Float32>)
Mike Kelly56858022020-01-27 12:14:47 +0000[diff] [blame]36
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]37ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwFloat321, BatchToSpaceNdNchwTest1<DataType::Float32>)
38ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwFloat322, BatchToSpaceNdNchwTest2<DataType::Float32>)
39ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwFloat323, BatchToSpaceNdNchwTest3<DataType::Float32>)
Teresa Charlinca5c82a2023-03-28 11:00:36 +0100[diff] [blame]40ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwFloat324, BatchToSpaceNdNchwTest4<DataType::Float32>)
41ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwFloat325, BatchToSpaceNdNchwTest5<DataType::Float32>)
42ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwFloat326, BatchToSpaceNdNchwTest6<DataType::Float32>)
43ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwFloat327, BatchToSpaceNdNchwTest7<DataType::Float32>)
Mike Kelly56858022020-01-27 12:14:47 +0000[diff] [blame]44
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]45ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcInt1, BatchToSpaceNdNhwcTest1<DataType::QAsymmS8>)
46ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcInt2, BatchToSpaceNdNhwcTest2<DataType::QAsymmS8>)
47ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcInt3, BatchToSpaceNdNhwcTest3<DataType::QAsymmS8>)
Teresa Charlinca5c82a2023-03-28 11:00:36 +0100[diff] [blame]48ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcInt4, BatchToSpaceNdNhwcTest4<DataType::QAsymmS8>)
49ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcInt5, BatchToSpaceNdNhwcTest5<DataType::QAsymmS8>)
50ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcInt6, BatchToSpaceNdNhwcTest6<DataType::QAsymmS8>)
51ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcInt7, BatchToSpaceNdNhwcTest7<DataType::QAsymmS8>)
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]52
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]53ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwInt1, BatchToSpaceNdNchwTest1<DataType::QAsymmS8>)
54ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwInt2, BatchToSpaceNdNchwTest2<DataType::QAsymmS8>)
55ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwInt3, BatchToSpaceNdNchwTest3<DataType::QAsymmS8>)
Teresa Charlinca5c82a2023-03-28 11:00:36 +0100[diff] [blame]56ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwInt4, BatchToSpaceNdNchwTest4<DataType::QAsymmS8>)
57ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwInt5, BatchToSpaceNdNchwTest5<DataType::QAsymmS8>)
58ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwInt6, BatchToSpaceNdNchwTest6<DataType::QAsymmS8>)
59ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwInt7, BatchToSpaceNdNchwTest7<DataType::QAsymmS8>)
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]60
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]61ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcUint1, BatchToSpaceNdNhwcTest1<DataType::QAsymmU8>)
62ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcUint2, BatchToSpaceNdNhwcTest2<DataType::QAsymmU8>)
63ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcUint3, BatchToSpaceNdNhwcTest3<DataType::QAsymmU8>)
Teresa Charlinca5c82a2023-03-28 11:00:36 +0100[diff] [blame]64ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcUint4, BatchToSpaceNdNhwcTest4<DataType::QAsymmU8>)
65ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcUint5, BatchToSpaceNdNhwcTest5<DataType::QAsymmU8>)
66ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcUint6, BatchToSpaceNdNhwcTest6<DataType::QAsymmU8>)
67ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcUint7, BatchToSpaceNdNhwcTest7<DataType::QAsymmU8>)
Mike Kelly56858022020-01-27 12:14:47 +0000[diff] [blame]68
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]69ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwUint1, BatchToSpaceNdNchwTest1<DataType::QAsymmU8>)
70ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwUint2, BatchToSpaceNdNchwTest2<DataType::QAsymmU8>)
71ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwUint3, BatchToSpaceNdNchwTest3<DataType::QAsymmU8>)
Teresa Charlinca5c82a2023-03-28 11:00:36 +0100[diff] [blame]72ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwUint4, BatchToSpaceNdNchwTest4<DataType::QAsymmU8>)
73ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwUint5, BatchToSpaceNdNchwTest5<DataType::QAsymmU8>)
74ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwUint6, BatchToSpaceNdNchwTest6<DataType::QAsymmU8>)
75ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwUint7, BatchToSpaceNdNchwTest7<DataType::QAsymmU8>)
Mike Kelly56858022020-01-27 12:14:47 +0000[diff] [blame]76
Teresa Charlin0f86ecf2022-10-13 15:47:08 +0100[diff] [blame]77// Batch Mat Mul
78ARMNN_AUTO_TEST_CASE_WITH_THF(BatchMatMul2DSimpleFloat32, BatchMatMul2DSimpleTest<DataType::Float32>);
Teresa Charlin1fe6c812022-11-01 15:59:50 +0000[diff] [blame]79ARMNN_AUTO_TEST_CASE_WITH_THF(BatchMatMul2DSimpleInt8, BatchMatMul2DSimpleTest<DataType::QAsymmS8>);
Teresa Charlin0f86ecf2022-10-13 15:47:08 +0100[diff] [blame]80
81ARMNN_AUTO_TEST_CASE_WITH_THF(BatchMatMul3DSimpleFloat32, BatchMatMul3DSimpleTest<DataType::Float32>);
Teresa Charlin1fe6c812022-11-01 15:59:50 +0000[diff] [blame]82ARMNN_AUTO_TEST_CASE_WITH_THF(BatchMatMul3DSimpleInt8, BatchMatMul3DSimpleTest<DataType::QAsymmS8>);
Teresa Charlin0f86ecf2022-10-13 15:47:08 +0100[diff] [blame]83
84ARMNN_AUTO_TEST_CASE_WITH_THF(BatchMatMulNCHWSimpleFloat32, BatchMatMulNCHWSimpleTest<DataType::Float32>);
Teresa Charlin1fe6c812022-11-01 15:59:50 +0000[diff] [blame]85ARMNN_AUTO_TEST_CASE_WITH_THF(BatchMatMulNCHWSimpleInt8, BatchMatMulNCHWSimpleTest<DataType::QAsymmS8>);
Teresa Charlin0f86ecf2022-10-13 15:47:08 +0100[diff] [blame]86
87ARMNN_AUTO_TEST_CASE_WITH_THF(BatchMatMul3DBatchFloat32, BatchMatMul3DBatchTest<DataType::Float32>);
Teresa Charlin1fe6c812022-11-01 15:59:50 +0000[diff] [blame]88ARMNN_AUTO_TEST_CASE_WITH_THF(BatchMatMul3DBatchInt8, BatchMatMul3DBatchTest<DataType::QAsymmS8>);
Teresa Charlin0f86ecf2022-10-13 15:47:08 +0100[diff] [blame]89
Teresa Charlin1fe6c812022-11-01 15:59:50 +0000[diff] [blame]90ARMNN_AUTO_TEST_CASE_WITH_THF(UNSUPPORTED_BatchMatMul3DBroadcastFloat32,
91 BatchMatMul3DBroadcastTest<DataType::Float32>);
92ARMNN_AUTO_TEST_CASE_WITH_THF(UNSUPPORTED_BatchMatMul3DBroadcastInt8,
93 BatchMatMul3DBroadcastTest<DataType::QAsymmS8>);
Teresa Charlin0f86ecf2022-10-13 15:47:08 +0100[diff] [blame]94
Teresa Charlin1fe6c812022-11-01 15:59:50 +0000[diff] [blame]95ARMNN_AUTO_TEST_CASE_WITH_THF(UNSUPPORTED_BatchMatMul3D2DBroadcastFloat32,
96 BatchMatMul3D2DBroadcastTest<DataType::Float32>);
97ARMNN_AUTO_TEST_CASE_WITH_THF(UNSUPPORTED_BatchMatMul3D2DBroadcastInt8,
98 BatchMatMul3D2DBroadcastTest<DataType::QAsymmS8>);
Teresa Charlin0f86ecf2022-10-13 15:47:08 +0100[diff] [blame]99
100ARMNN_AUTO_TEST_CASE_WITH_THF(BatchMatMul2DTinyFloat32, BatchMatMul2DTinyTest<DataType::Float32>);
Teresa Charlin1fe6c812022-11-01 15:59:50 +0000[diff] [blame]101ARMNN_AUTO_TEST_CASE_WITH_THF(BatchMatMul2DTinyInt8, BatchMatMul2DTinyTest<DataType::QAsymmS8>);
Teresa Charlin0f86ecf2022-10-13 15:47:08 +0100[diff] [blame]102
103ARMNN_AUTO_TEST_CASE_WITH_THF(BatchMatMul2DTranspSimpleFloat32, BatchMatMul2DTranspSimpleTest<DataType::Float32>);
Teresa Charlin1fe6c812022-11-01 15:59:50 +0000[diff] [blame]104ARMNN_AUTO_TEST_CASE_WITH_THF(BatchMatMul2DTranspSimpleInt8, BatchMatMul2DTranspSimpleTest<DataType::QAsymmS8>);
Teresa Charlin0f86ecf2022-10-13 15:47:08 +0100[diff] [blame]105
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]106// Convolution
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]107ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution1d, Convolution1dTest, true)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]108
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]109ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution2d, SimpleConvolution2d3x5Test, true, DataLayout::NCHW)
110ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution2dNhwc, SimpleConvolution2d3x5Test, true, DataLayout::NHWC)
111ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution2d3x3Uint8, SimpleConvolution2d3x3Uint8Test, true, DataLayout::NCHW)
112ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution2d3x3Uint8Nhwc, SimpleConvolution2d3x3Uint8Test, true, DataLayout::NHWC)
113ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedConvolution2d, SimpleConvolution2d3x5Test, false, DataLayout::NCHW)
114ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedConvolution2dNhwc, SimpleConvolution2d3x5Test, false, DataLayout::NHWC)
115ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedConvolution2dStride2x2Nhwc,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]116 SimpleConvolution2d3x3Stride2x2Test, false, DataLayout::NHWC)
jimfly010a088a62018-10-25 17:05:05 +0100[diff] [blame]117
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]118ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedConvolution2dSquare, SimpleConvolution2d3x3Test, false, DataLayout::NCHW)
119ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution2dAsymmetricPadding,
120 Convolution2dAsymmetricPaddingTest,
121 DataLayout::NCHW)
narpra015f703182018-10-26 16:24:58 +0100[diff] [blame]122
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]123ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedConvolution2dSquareNhwc, SimpleConvolution2d3x3Test, false, DataLayout::NHWC)
124ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution2dAsymmetricPaddingNhwc,
narpra015f703182018-10-26 16:24:58 +0100[diff] [blame]125 Convolution2dAsymmetricPaddingTest,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]126 DataLayout::NHWC)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]127
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]128ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution2dSquareNhwc, SimpleConvolution2d3x3NhwcTest, false)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]129
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]130ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d3x3Dilation3x3,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]131 Convolution2d3x3Dilation3x3Test<DataType::Float32, DataType::Float32>,
Jan Eilers59c66702019-07-11 10:22:36 +0100[diff] [blame]132 false,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]133 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]134ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d3x3Dilation3x3Nhwc,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]135 Convolution2d3x3Dilation3x3Test<DataType::Float32, DataType::Float32>,
Jan Eilers59c66702019-07-11 10:22:36 +0100[diff] [blame]136 false,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]137 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]138ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d3x3Dilation3x3Int8,
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]139 Convolution2d3x3Dilation3x3Test<DataType::QAsymmS8, DataType::Signed32>,
140 false,
141 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]142ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d3x3Dilation3x3NhwcInt8,
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]143 Convolution2d3x3Dilation3x3Test<DataType::QAsymmS8, DataType::Signed32>,
144 false,
145 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]146ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d3x3Dilation3x3Uint8,
Derek Lambertif90c56d2020-01-10 17:14:08 +0000[diff] [blame]147 Convolution2d3x3Dilation3x3Test<DataType::QAsymmU8, DataType::Signed32>,
Jan Eilers59c66702019-07-11 10:22:36 +0100[diff] [blame]148 false,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]149 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]150ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d3x3Dilation3x3NhwcUint8,
Derek Lambertif90c56d2020-01-10 17:14:08 +0000[diff] [blame]151 Convolution2d3x3Dilation3x3Test<DataType::QAsymmU8, DataType::Signed32>,
Jan Eilers59c66702019-07-11 10:22:36 +0100[diff] [blame]152 false,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]153 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]154ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x3x3Dilation3x3,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]155 Convolution2d2x3x3Dilation3x3Test<DataType::Float32, DataType::Float32>,
Jan Eilers59c66702019-07-11 10:22:36 +0100[diff] [blame]156 false,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]157 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]158ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x3x3Dilation3x3Nhwc,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]159 Convolution2d2x3x3Dilation3x3Test<DataType::Float32, DataType::Float32>,
Jan Eilers59c66702019-07-11 10:22:36 +0100[diff] [blame]160 false,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]161 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]162ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x3x3Dilation3x3Int8,
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]163 Convolution2d2x3x3Dilation3x3Test<DataType::QAsymmS8, DataType::Signed32>,
164 false,
165 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]166ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x3x3Dilation3x3NhwcInt8,
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]167 Convolution2d2x3x3Dilation3x3Test<DataType::QAsymmS8, DataType::Signed32>,
168 false,
169 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]170ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x3x3Dilation3x3Uint8,
Derek Lambertif90c56d2020-01-10 17:14:08 +0000[diff] [blame]171 Convolution2d2x3x3Dilation3x3Test<DataType::QAsymmU8, DataType::Signed32>,
Jan Eilers59c66702019-07-11 10:22:36 +0100[diff] [blame]172 false,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]173 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]174ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x3x3Dilation3x3NhwcUint8,
Derek Lambertif90c56d2020-01-10 17:14:08 +0000[diff] [blame]175 Convolution2d2x3x3Dilation3x3Test<DataType::QAsymmU8, DataType::Signed32>,
Jan Eilers59c66702019-07-11 10:22:36 +0100[diff] [blame]176 false,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]177 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]178ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x2Dilation2x2Padding2x2Stride3x3,
Jan Eilers59c66702019-07-11 10:22:36 +0100[diff] [blame]179 Convolution2d2x2Dilation2x2Padding2x2Stride3x3Test
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]180 <DataType::Float32, DataType::Float32>,
Jan Eilers59c66702019-07-11 10:22:36 +0100[diff] [blame]181 false,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]182 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]183ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x2Dilation2x2Padding2x2Stride3x3Nhwc,
Jan Eilers59c66702019-07-11 10:22:36 +0100[diff] [blame]184 Convolution2d2x2Dilation2x2Padding2x2Stride3x3Test
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]185 <DataType::Float32, DataType::Float32>,
Jan Eilers59c66702019-07-11 10:22:36 +0100[diff] [blame]186 false,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]187 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]188ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x2Dilation2x2Padding2x2Stride3x3Int8,
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]189 Convolution2d2x2Dilation2x2Padding2x2Stride3x3Test
190 <DataType::QAsymmS8, DataType::Signed32>,
191 false,
192 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]193ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x2Dilation2x2Padding2x2Stride3x3NhwcInt8,
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]194 Convolution2d2x2Dilation2x2Padding2x2Stride3x3Test
195 <DataType::QAsymmS8, DataType::Signed32>,
196 false,
197 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]198ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x2Dilation2x2Padding2x2Stride3x3Uint8,
Jan Eilers59c66702019-07-11 10:22:36 +0100[diff] [blame]199 Convolution2d2x2Dilation2x2Padding2x2Stride3x3Test
Derek Lambertif90c56d2020-01-10 17:14:08 +0000[diff] [blame]200 <DataType::QAsymmU8, DataType::Signed32>,
Jan Eilers59c66702019-07-11 10:22:36 +0100[diff] [blame]201 false,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]202 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]203ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x2Dilation2x2Padding2x2Stride3x3NhwcUint8,
Jan Eilers59c66702019-07-11 10:22:36 +0100[diff] [blame]204 Convolution2d2x2Dilation2x2Padding2x2Stride3x3Test
Derek Lambertif90c56d2020-01-10 17:14:08 +0000[diff] [blame]205 <DataType::QAsymmU8, DataType::Signed32>,
Jan Eilers59c66702019-07-11 10:22:36 +0100[diff] [blame]206 false,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]207 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]208ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dMult4,
Kevin May665a964a2019-08-21 16:53:50 +0100[diff] [blame]209 DepthwiseConvolution2dMult4Test<armnn::DataType::Float32, armnn::DataType::Float32>,
210 false,
211 armnn::DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]212ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dMult2,
Kevin May665a964a2019-08-21 16:53:50 +0100[diff] [blame]213 DepthwiseConvolution2dMult2Test<armnn::DataType::Float32, armnn::DataType::Float32>,
214 false,
215 armnn::DataLayout::NCHW)
Jan Eilers59c66702019-07-11 10:22:36 +0100[diff] [blame]216
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]217ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2dPerAxisQuantTestNchw, Convolution2dPerAxisQuantTest, DataLayout::NCHW);
218ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2dPerAxisQuantTestNhwc, Convolution2dPerAxisQuantTest, DataLayout::NHWC);
Aron Virginas-Tar21fc28b2019-11-26 14:04:54 +0000[diff] [blame]219
Teresa Charlinec5f7d12021-10-22 17:15:00 +0100[diff] [blame]220// Convolution 3d - NDHWC
221ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution3d3x3x3Float32,
222 SimpleConvolution3d3x3x3Float32Test,
223 false,
224 DataLayout::NDHWC)
225ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution3d3x3x3Int8,
226 SimpleConvolution3d3x3x3Int8Test,
227 false,
228 DataLayout::NDHWC)
229ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution3d3x3x3Uint8,
230 SimpleConvolution3d3x3x3Uint8Test,
231 false,
232 DataLayout::NDHWC)
233
234ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution3d2x2x2Strides3x5x5Float32,
235 Convolution3d2x2x2Strides3x5x5Float32Test,
236 false,
237 DataLayout::NDHWC)
238ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution3d2x2x2Strides3x5x5TestInt8,
239 Convolution3d2x2x2Strides3x5x5Int8Test,
240 true,
241 DataLayout::NDHWC)
242ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution3d2x2x2Strides3x5x5TestUint8,
243 Convolution3d2x2x2Strides3x5x5Uint8Test,
244 false,
245 DataLayout::NDHWC)
246
247ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution3dPaddingSame3x3x3Float32,
248 Convolution3dPaddingSame3x3x3Float32Test,
249 false,
250 DataLayout::NDHWC)
251ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution3dPaddingSame3x3x3TestInt8,
252 Convolution3dPaddingSame3x3x3Int8Test,
253 false,
254 DataLayout::NDHWC)
255ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution3dPaddingSame3x3x3TestUint8,
256 Convolution3dPaddingSame3x3x3Uint8Test,
257 false,
258 DataLayout::NDHWC)
259
260ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution3d2x2x2Stride3x3x3SmallTestFloat32,
261 Convolution3d2x2x2Stride3x3x3SmallFloat32Test,
262 false,
263 DataLayout::NDHWC)
264#ifdef __ARM_FEATURE_FP16_VECTOR_ARITHMETIC
265ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution3d2x3x3TestFloat16,
266 Convolution3d2x3x3Float16Test,
267 true,
268 DataLayout::NDHWC)
269ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution3d2x2x2SmallTestFloat16,
270 Convolution3d2x2x2SmallFloat16Test,
271 false,
272 DataLayout::NDHWC)
273#endif
274
Aron Virginas-Tar2f00b742019-09-30 13:28:08 +0100[diff] [blame]275// DepthToSpace
276ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwFloat32_1, DepthToSpaceTest1<DataType::Float32>, DataLayout::NCHW);
277ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwFloat32_2, DepthToSpaceTest2<DataType::Float32>, DataLayout::NCHW);
278ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwFloat32_3, DepthToSpaceTest3<DataType::Float32>, DataLayout::NCHW);
279ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwFloat32_4, DepthToSpaceTest4<DataType::Float32>, DataLayout::NCHW);
280
Aron Virginas-Tar99836d32019-09-30 16:34:31 +0100[diff] [blame]281ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwFloat16_1, DepthToSpaceTest1<DataType::Float16>, DataLayout::NCHW);
282ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwFloat16_2, DepthToSpaceTest2<DataType::Float16>, DataLayout::NCHW);
283ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwFloat16_3, DepthToSpaceTest3<DataType::Float16>, DataLayout::NCHW);
284ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwFloat16_4, DepthToSpaceTest4<DataType::Float16>, DataLayout::NCHW);
285
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]286ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwInt8_1, DepthToSpaceTest1<DataType::QAsymmS8>, DataLayout::NCHW);
287ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwInt8_2, DepthToSpaceTest2<DataType::QAsymmS8>, DataLayout::NCHW);
288ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwInt8_3, DepthToSpaceTest3<DataType::QAsymmS8>, DataLayout::NCHW);
289ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwInt8_4, DepthToSpaceTest4<DataType::QAsymmS8>, DataLayout::NCHW);
290
Derek Lambertif90c56d2020-01-10 17:14:08 +0000[diff] [blame]291ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwUint8_1, DepthToSpaceTest1<DataType::QAsymmU8>, DataLayout::NCHW);
292ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwUint8_2, DepthToSpaceTest2<DataType::QAsymmU8>, DataLayout::NCHW);
293ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwUint8_3, DepthToSpaceTest3<DataType::QAsymmU8>, DataLayout::NCHW);
294ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwUint8_4, DepthToSpaceTest4<DataType::QAsymmU8>, DataLayout::NCHW);
Aron Virginas-Tar2f00b742019-09-30 13:28:08 +0100[diff] [blame]295
Derek Lambertif90c56d2020-01-10 17:14:08 +0000[diff] [blame]296ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwInt16_1, DepthToSpaceTest1<DataType::QSymmS16>, DataLayout::NCHW);
297ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwInt16_2, DepthToSpaceTest2<DataType::QSymmS16>, DataLayout::NCHW);
298ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwInt16_3, DepthToSpaceTest3<DataType::QSymmS16>, DataLayout::NCHW);
299ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwInt16_4, DepthToSpaceTest4<DataType::QSymmS16>, DataLayout::NCHW);
Aron Virginas-Tar2f00b742019-09-30 13:28:08 +0100[diff] [blame]300
301ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcFloat32_1, DepthToSpaceTest1<DataType::Float32>, DataLayout::NHWC);
302ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcFloat32_2, DepthToSpaceTest2<DataType::Float32>, DataLayout::NHWC);
303ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcFloat32_3, DepthToSpaceTest3<DataType::Float32>, DataLayout::NHWC);
304ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcFloat32_4, DepthToSpaceTest4<DataType::Float32>, DataLayout::NHWC);
305
Aron Virginas-Tar99836d32019-09-30 16:34:31 +0100[diff] [blame]306ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcFloat16_1, DepthToSpaceTest1<DataType::Float16>, DataLayout::NHWC);
307ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcFloat16_2, DepthToSpaceTest2<DataType::Float16>, DataLayout::NHWC);
308ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcFloat16_3, DepthToSpaceTest3<DataType::Float16>, DataLayout::NHWC);
309ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcFloat16_4, DepthToSpaceTest4<DataType::Float16>, DataLayout::NHWC);
310
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]311ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcInt8_1, DepthToSpaceTest1<DataType::QAsymmS8>, DataLayout::NHWC);
312ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcInt8_2, DepthToSpaceTest2<DataType::QAsymmS8>, DataLayout::NHWC);
313ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcInt8_3, DepthToSpaceTest3<DataType::QAsymmS8>, DataLayout::NHWC);
314ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcInt8_4, DepthToSpaceTest4<DataType::QAsymmS8>, DataLayout::NHWC);
315
Derek Lambertif90c56d2020-01-10 17:14:08 +0000[diff] [blame]316ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcUint8_1, DepthToSpaceTest1<DataType::QAsymmU8>, DataLayout::NHWC);
317ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcUint8_2, DepthToSpaceTest2<DataType::QAsymmU8>, DataLayout::NHWC);
318ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcUint8_3, DepthToSpaceTest3<DataType::QAsymmU8>, DataLayout::NHWC);
319ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcUint8_4, DepthToSpaceTest4<DataType::QAsymmU8>, DataLayout::NHWC);
Aron Virginas-Tar2f00b742019-09-30 13:28:08 +0100[diff] [blame]320
Derek Lambertif90c56d2020-01-10 17:14:08 +0000[diff] [blame]321ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcInt16_1, DepthToSpaceTest1<DataType::QSymmS16>, DataLayout::NHWC);
322ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcInt16_2, DepthToSpaceTest2<DataType::QSymmS16>, DataLayout::NHWC);
323ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcInt16_3, DepthToSpaceTest3<DataType::QSymmS16>, DataLayout::NHWC);
324ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcInt16_4, DepthToSpaceTest4<DataType::QSymmS16>, DataLayout::NHWC);
Aron Virginas-Tar2f00b742019-09-30 13:28:08 +0100[diff] [blame]325
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]326// Depthwise Convolution
Jan Eilers53ef7952021-06-02 12:01:25 +0100[diff] [blame]327ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2d, DepthwiseConvolution2dTest, true, DataLayout::NCHW)
328ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dUint8, DepthwiseConvolution2dUint8Test, true, DataLayout::NCHW)
329
330ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedDepthwiseConvolution2d, DepthwiseConvolution2dTest, false, DataLayout::NCHW)
331
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]332ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dDepthMul1,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]333 DepthwiseConvolution2dDepthMul1Test, true, DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]334ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedDepthwiseConvolution2dDepthMul1,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]335 DepthwiseConvolution2dDepthMul1Test, false, DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]336ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dDepthMul1Uint8,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]337 DepthwiseConvolution2dDepthMul1Uint8Test, true, DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]338ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedDepthwiseConvolution2dDepthMul1Uint8,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]339 DepthwiseConvolution2dDepthMul1Uint8Test, false, DataLayout::NCHW)
jimfly01b9c89632018-10-26 16:50:13 +0100[diff] [blame]340
341// NHWC Depthwise Convolution
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]342ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dDepthMul1NHhwc,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]343 DepthwiseConvolution2dDepthMul1Test, true, DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]344ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedDepthwiseConvolution2dDepthMul1Nhwc,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]345 DepthwiseConvolution2dDepthMul1Test, false, DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]346ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dDepthMul1Uint8Nhwc,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]347 DepthwiseConvolution2dDepthMul1Uint8Test, true, DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]348ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedDepthwiseConvolution2dDepthMul1Uint8Nhwc,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]349 DepthwiseConvolution2dDepthMul1Uint8Test, false, DataLayout::NHWC)
jimfly01b9c89632018-10-26 16:50:13 +0100[diff] [blame]350
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]351ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dDepthNhwc, DepthwiseConvolution2dDepthNhwcTest, false)
352ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleDepthwiseConvolution2d3x3Dilation3x3Nhwc,
Bruno Goncalves22972f02019-04-26 21:03:24 -0300[diff] [blame]353 SimpleDepthwiseConvolution2d3x3Dilation3x3NhwcTest)
jimfly01b9c89632018-10-26 16:50:13 +0100[diff] [blame]354
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]355
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]356ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dAsymmetric,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]357 DepthwiseConvolution2dAsymmetricTest, true, DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]358ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedDepthwiseConvolution2dAsymmetric,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]359 DepthwiseConvolution2dAsymmetricTest, false, DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]360ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dAsymmetricNhwc,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]361 DepthwiseConvolution2dAsymmetricTest, true, DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]362ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedDepthwiseConvolution2dAsymmetricNhwc,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]363 DepthwiseConvolution2dAsymmetricTest, false, DataLayout::NHWC)
surmeh013537c2c2018-05-18 16:31:43 +0100[diff] [blame]364
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]365ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dDepthMul64, DepthwiseConvolution2dDepthMul64Test);
Matthew Jackson99107922019-07-25 12:58:56 +0100[diff] [blame]366
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]367ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dPerAxisQuantTestNchw, DepthwiseConvolution2dPerAxisQuantTest,
Aron Virginas-Tar24457442019-11-27 13:18:25 +0000[diff] [blame]368 DataLayout::NCHW);
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]369ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dPerAxisQuantTestNhwc, DepthwiseConvolution2dPerAxisQuantTest,
Aron Virginas-Tar24457442019-11-27 13:18:25 +0000[diff] [blame]370 DataLayout::NHWC);
371
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]372namespace
373{
374
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]375DepthwiseConvolution2dDescriptor MakeDepthwiseConv2dDesc(uint32_t strideX, uint32_t strideY,
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]376 uint32_t depthMultiplier = 1, uint32_t padLeft = 0, uint32_t padRight = 0,
377 uint32_t padTop = 0, uint32_t padBottom = 0)
378{
Jan Eilers8eb25602020-03-09 12:13:48 +0000[diff] [blame]379 IgnoreUnused(depthMultiplier);
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]380
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]381 DepthwiseConvolution2dDescriptor desc;
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]382
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]383 desc.m_PadLeft = padLeft;
384 desc.m_PadRight = padRight;
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]385
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]386 desc.m_PadTop = padTop;
387 desc.m_PadBottom = padBottom;
388 desc.m_StrideX = strideX;
389 desc.m_StrideY = strideY;
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]390 desc.m_BiasEnabled = false;
391
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]392 return desc;
393}
394
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]395TensorInfo CreateOutputTensorInfo(const TensorInfo& inputInfo,
396 const TensorInfo& weightsInfo,
397 const DepthwiseConvolution2dDescriptor& descriptor,
398 DataType dataType)
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]399{
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]400 const TensorShape& inputShape = inputInfo.GetShape();
401 const TensorShape& filterShape = weightsInfo.GetShape();
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]402
403 unsigned int inWidth = inputShape[3];
404 unsigned int inHeight = inputShape[2];
405 unsigned int inBatchSize = inputShape[0];
406
Jan Eilers53ef7952021-06-02 12:01:25 +0100[diff] [blame]407 unsigned int filterWidth = filterShape[2];
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]408 unsigned int readWidth = (inWidth + descriptor.m_PadLeft + descriptor.m_PadRight) - (filterWidth);
409 unsigned int outWidth = 1u + (readWidth / descriptor.m_StrideX);
410
Jan Eilers53ef7952021-06-02 12:01:25 +0100[diff] [blame]411 unsigned int filterHeight = filterShape[1];
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]412 unsigned int readHeight = (inHeight + descriptor.m_PadTop + descriptor.m_PadBottom) - (filterHeight);
413 unsigned int outHeight = 1u + (readHeight / descriptor.m_StrideY);
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]414
Jan Eilers53ef7952021-06-02 12:01:25 +0100[diff] [blame]415 unsigned int outChannels = filterShape[3];
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]416 unsigned int outBatchSize = inBatchSize;
417
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]418 TensorShape outputShape({outBatchSize, outChannels, outHeight, outWidth});
419 return TensorInfo(outputShape, dataType);
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]420}
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]421}
422
Sadik Armagan1625efc2021-06-10 18:24:34 +0100[diff] [blame]423TEST_CASE("DepthwiseConv2dUtils")
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]424{
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]425 const DataType dataType = DataType::Float32;
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]426
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]427 TensorInfo inputInfo({1, 1, 10, 10 }, dataType);
428 TensorInfo outputInfo;
Cathal Corbett4452baf2022-05-13 09:55:59 +0100[diff] [blame]429 TensorInfo weightsInfo3x3({ 1, 3, 3, 1 }, dataType, 0, 0, true); // [1,H,W,I*M]
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]430 TensorInfo biasesInfo;
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]431
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]432 DepthwiseConvolution2dDescriptor descriptor;
433 NeonLayerSupport layerSupport;
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]434
435 // Strides supported: 1,2,3
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]436 descriptor = MakeDepthwiseConv2dDesc(1, 1);
437 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo3x3, descriptor, dataType);
Sadik Armagan1625efc2021-06-10 18:24:34 +0100[diff] [blame]438 CHECK(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100[diff] [blame]439 weightsInfo3x3, biasesInfo));
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]440
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]441 descriptor = MakeDepthwiseConv2dDesc(1, 2);
442 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo3x3, descriptor, dataType);
Sadik Armagan1625efc2021-06-10 18:24:34 +0100[diff] [blame]443 CHECK(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100[diff] [blame]444 weightsInfo3x3, biasesInfo));
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]445
446 descriptor = MakeDepthwiseConv2dDesc(1, 3);
447 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo3x3, descriptor, dataType);
Sadik Armagan1625efc2021-06-10 18:24:34 +0100[diff] [blame]448 CHECK(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100[diff] [blame]449 weightsInfo3x3, biasesInfo));
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]450
451 descriptor = MakeDepthwiseConv2dDesc(2, 1);
452 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo3x3, descriptor, dataType);
Sadik Armagan1625efc2021-06-10 18:24:34 +0100[diff] [blame]453 CHECK(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100[diff] [blame]454 weightsInfo3x3, biasesInfo));
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]455
456 descriptor = MakeDepthwiseConv2dDesc(2, 2);
457 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo3x3, descriptor, dataType);
Sadik Armagan1625efc2021-06-10 18:24:34 +0100[diff] [blame]458 CHECK(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100[diff] [blame]459 weightsInfo3x3, biasesInfo));
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]460
461 descriptor = MakeDepthwiseConv2dDesc(2, 3);
462 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo3x3, descriptor, dataType);
Sadik Armagan1625efc2021-06-10 18:24:34 +0100[diff] [blame]463 CHECK(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100[diff] [blame]464 weightsInfo3x3, biasesInfo));
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]465
466 descriptor = MakeDepthwiseConv2dDesc(3, 1);
467 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo3x3, descriptor, dataType);
Sadik Armagan1625efc2021-06-10 18:24:34 +0100[diff] [blame]468 CHECK(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100[diff] [blame]469 weightsInfo3x3, biasesInfo));
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]470
471 descriptor = MakeDepthwiseConv2dDesc(3, 2);
472 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo3x3, descriptor, dataType);
Sadik Armagan1625efc2021-06-10 18:24:34 +0100[diff] [blame]473 CHECK(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100[diff] [blame]474 weightsInfo3x3, biasesInfo));
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]475
476 descriptor = MakeDepthwiseConv2dDesc(3, 3);
477 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo3x3, descriptor, dataType);
Sadik Armagan1625efc2021-06-10 18:24:34 +0100[diff] [blame]478 CHECK(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100[diff] [blame]479 weightsInfo3x3, biasesInfo));
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]480
481 // Supported stride 4
482 descriptor = MakeDepthwiseConv2dDesc(4, 1);
483 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo3x3, descriptor, dataType);
Sadik Armagan1625efc2021-06-10 18:24:34 +0100[diff] [blame]484 CHECK(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100[diff] [blame]485 weightsInfo3x3, biasesInfo));
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]486
487 // Supported weights shape 1x1
Cathal Corbett4452baf2022-05-13 09:55:59 +0100[diff] [blame]488 TensorInfo weightsInfo1x1({ 1, 1, 1, 1 }, DataType::Float32, 0, 0, true);
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]489 descriptor = MakeDepthwiseConv2dDesc(1, 1);
490 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo1x1, descriptor, dataType);
Sadik Armagan1625efc2021-06-10 18:24:34 +0100[diff] [blame]491 CHECK(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100[diff] [blame]492 weightsInfo1x1, biasesInfo));
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]493
494 // Supported shape 2x2
Cathal Corbett4452baf2022-05-13 09:55:59 +0100[diff] [blame]495 TensorInfo weightsInfo2x2({ 1, 2, 2, 1 }, DataType::Float32, 0, 0, true);
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]496 descriptor = MakeDepthwiseConv2dDesc(1, 1);
497 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo2x2, descriptor, dataType);
Sadik Armagan1625efc2021-06-10 18:24:34 +0100[diff] [blame]498 CHECK(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100[diff] [blame]499 weightsInfo2x2, biasesInfo));
surmeh013537c2c2018-05-18 16:31:43 +0100[diff] [blame]500
501 // Asymmetric padding
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]502 descriptor = MakeDepthwiseConv2dDesc(1, 1, 1, 1, 2, 1, 2);
503 outputInfo = CreateOutputTensorInfo(inputInfo, weightsInfo3x3, descriptor, dataType);
Sadik Armagan1625efc2021-06-10 18:24:34 +0100[diff] [blame]504 CHECK(layerSupport.IsDepthwiseConvolutionSupported(inputInfo, outputInfo, descriptor,
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100[diff] [blame]505 weightsInfo3x3, biasesInfo));
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]506}
507
Narumol Prangnawarat01961a72019-05-30 16:47:12 +0100[diff] [blame]508// Dequantize
Jan Eilersf7107932019-11-01 11:09:36 +0000[diff] [blame]509// Fp16 is only supported if __ARM_FEATURE_FP16_VECTOR_ARITHMETIC is enabled
Narumol Prangnawarat01961a72019-05-30 16:47:12 +0100[diff] [blame]510ARMNN_AUTO_TEST_CASE(DequantizeSimpleUint8, DequantizeSimpleUint8Test)
511ARMNN_AUTO_TEST_CASE(DequantizeOffsetUint8, DequantizeOffsetUint8Test)
Jan Eilersf7107932019-11-01 11:09:36 +0000[diff] [blame]512ARMNN_AUTO_TEST_CASE(DequantizeSimpleInt16, DequantizeSimpleInt16Test)
Narumol Prangnawarat01961a72019-05-30 16:47:12 +0100[diff] [blame]513
Ryan OShea19e79422022-05-04 00:38:03 +0100[diff] [blame]514// Pooling2d
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]515ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling2dSize3x3Stride2x4,
516 SimpleMaxPooling2dSize3x3Stride2x4Test, true)
517ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling2dSize3x3Stride2x4Uint8,
518 SimpleMaxPooling2dSize3x3Stride2x4Uint8Test, true)
James Conroy45a9b772018-10-31 11:47:53 +0000[diff] [blame]519
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]520ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling2d, SimpleMaxPooling2dTest, DataLayout::NCHW)
521ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling2dNhwc, SimpleMaxPooling2dTest, DataLayout::NHWC)
522ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling2dUint8, SimpleMaxPooling2dUint8Test, DataLayout::NCHW)
523ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling2dUint8Nhwc, SimpleMaxPooling2dUint8Test, DataLayout::NHWC)
James Conroy45a9b772018-10-31 11:47:53 +0000[diff] [blame]524
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]525ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleAveragePooling2d, SimpleAveragePooling2dTest, DataLayout::NCHW)
526ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleAveragePooling2dNhwc, SimpleAveragePooling2dTest, DataLayout::NCHW)
527ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleAveragePooling2dUint8,
528 SimpleAveragePooling2dUint8Test, DataLayout::NCHW)
529ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleAveragePooling2dUint8Nhwc,
530 SimpleAveragePooling2dUint8Test, DataLayout::NHWC)
surmeh01bceff2f2018-03-29 16:29:27 +0100[diff] [blame]531
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]532ARMNN_AUTO_TEST_CASE_WITH_THF(LargeTensorsAveragePooling2d, LargeTensorsAveragePooling2dTest)
533ARMNN_AUTO_TEST_CASE_WITH_THF(LargeTensorsAveragePooling2dUint8, LargeTensorsAveragePooling2dUint8Test)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]534
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]535ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleL2Pooling2d, SimpleL2Pooling2dTest, DataLayout::NCHW)
536ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleL2Pooling2dNeon, SimpleL2Pooling2dTest, DataLayout::NHWC)
537ARMNN_AUTO_TEST_CASE_WITH_THF(UNSUPPORTED_SimpleL2Pooling2dUint8, SimpleL2Pooling2dUint8Test, DataLayout::NCHW)
James Conroy45a9b772018-10-31 11:47:53 +0000[diff] [blame]538
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]539ARMNN_AUTO_TEST_CASE_WITH_THF(L2Pooling2dSize3Stride1, L2Pooling2dSize3Stride1Test)
540ARMNN_AUTO_TEST_CASE_WITH_THF(UNSUPPORTED_L2Pooling2dSize3Stride1Uint8, L2Pooling2dSize3Stride1Uint8Test)
541ARMNN_AUTO_TEST_CASE_WITH_THF(L2Pooling2dSize3Stride3, L2Pooling2dSize3Stride3Test)
542ARMNN_AUTO_TEST_CASE_WITH_THF(UNSUPPORTED_L2Pooling2dSize3Stride3Uint8, L2Pooling2dSize3Stride3Uint8Test)
543ARMNN_AUTO_TEST_CASE_WITH_THF(L2Pooling2dSize3Stride4, L2Pooling2dSize3Stride4Test)
544ARMNN_AUTO_TEST_CASE_WITH_THF(UNSUPPORTED_L2Pooling2dSize3Stride4Uint8, L2Pooling2dSize3Stride4Uint8Test)
545ARMNN_AUTO_TEST_CASE_WITH_THF(L2Pooling2dSize7, L2Pooling2dSize7Test)
546ARMNN_AUTO_TEST_CASE_WITH_THF(UNSUPPORTED_L2Pooling2dSize7Uint8, L2Pooling2dSize7Uint8Test)
547ARMNN_AUTO_TEST_CASE_WITH_THF(L2Pooling2dSize9, L2Pooling2dSize9Test)
548ARMNN_AUTO_TEST_CASE_WITH_THF(UNSUPPORTED_L2Pooling2dSize9Uint8, L2Pooling2dSize9Uint8Test)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]549
550// Ignore padding values for pooling but count padding fields into the divisor
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]551ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleMaxPooling2d, IgnorePaddingSimpleMaxPooling2dTest)
552ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleMaxPooling2dUint8, IgnorePaddingSimpleMaxPooling2dUint8Test)
553ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingMaxPooling2dSize3, IgnorePaddingMaxPooling2dSize3Test)
554ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingMaxPooling2dSize3Uint8, IgnorePaddingMaxPooling2dSize3Uint8Test)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]555
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]556ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleAveragePooling2d, IgnorePaddingSimpleAveragePooling2dTest)
557ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleAveragePooling2dUint8, IgnorePaddingSimpleAveragePooling2dUint8Test)
558ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleAveragePooling2dNoPadding,
559 IgnorePaddingSimpleAveragePooling2dNoPaddingTest)
560ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleAveragePooling2dNoPaddingUint8,
561 IgnorePaddingSimpleAveragePooling2dNoPaddingUint8Test)
562ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingAveragePooling2dSize3, IgnorePaddingAveragePooling2dSize3Test)
563ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingAveragePooling2dSize3Uint8,
564 IgnorePaddingAveragePooling2dSize3Uint8Test)
565ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingAveragePooling2dSize3x2Stride2x2,
566 IgnorePaddingAveragePooling2dSize3x2Stride2x2Test, false)
567ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingAveragePooling2dSize3x2Stride2x2NoPadding,
568 IgnorePaddingAveragePooling2dSize3x2Stride2x2Test,
569 true)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]570
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]571ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleL2Pooling2d, IgnorePaddingSimpleL2Pooling2dTest)
572ARMNN_AUTO_TEST_CASE_WITH_THF(UNSUPPORTED_IgnorePaddingSimpleL2Pooling2dUint8,
573 IgnorePaddingSimpleL2Pooling2dUint8Test)
574ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingL2Pooling2dSize3, IgnorePaddingL2Pooling2dSize3Test)
575ARMNN_AUTO_TEST_CASE_WITH_THF(UNSUPPORTED_IgnorePaddingL2Pooling2dSize3Uint8,
576 IgnorePaddingL2Pooling2dSize3Uint8Test)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]577
Ryan OShea19e79422022-05-04 00:38:03 +0100[diff] [blame]578//Pooling 3d
579ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling3dSize2x2x2Stride1x1x1,
580 SimpleMaxPooling3dSize2x2x2Stride1x1x1Test,
581 DataLayout::NDHWC)
582
583ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling3dSize2x2x2Stride1x1x1Uint8,
584 SimpleMaxPooling3dSize2x2x2Stride1x1x1Uint8Test,
585 DataLayout::NDHWC)
586
587ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling3d,
588 SimpleMaxPooling3dTest,
589 DataLayout::NDHWC)
590
591ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling3dUint8,
592 SimpleMaxPooling3dUint8Test,
593 DataLayout::NDHWC)
594
595ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleMaxPooling3d,
596 IgnorePaddingSimpleMaxPooling3dTest,
597 DataLayout::NDHWC)
598
599ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleMaxPooling3dUint8,
600 IgnorePaddingSimpleMaxPooling3dUint8Test,
601 DataLayout::NDHWC)
602
603ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleAveragePooling3d,
604 SimpleAveragePooling3dTest,
605 DataLayout::NDHWC)
606
607ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleAveragePooling3dUint8,
608 SimpleAveragePooling3dUint8Test,
609 DataLayout::NDHWC)
610
611ARMNN_AUTO_TEST_CASE_WITH_THF(LargeTensorsAveragePooling3d,
612 LargeTensorsAveragePooling3dTest,
613 DataLayout::NDHWC)
614
615ARMNN_AUTO_TEST_CASE_WITH_THF(LargeTensorsAveragePooling3dUint8,
616 LargeTensorsAveragePooling3dUint8Test,
617 DataLayout::NDHWC)
618
619ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleAveragePooling3d,
620 IgnorePaddingSimpleAveragePooling3dTest,
621 DataLayout::NDHWC)
622
623ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleL2Pooling3d,
624 SimpleL2Pooling3dTest,
625 DataLayout::NDHWC)
626
627ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleL2Pooling3d,
628 IgnorePaddingSimpleL2Pooling3dTest,
629 DataLayout::NDHWC)
630
631ARMNN_AUTO_TEST_CASE_WITH_THF(AsymmetricNonSquareMaxPooling3d,
632 AsymmetricNonSquareMaxPooling3dTest,
633 DataLayout::NDHWC)
634
635ARMNN_AUTO_TEST_CASE_WITH_THF(AsymmetricNonSquareMaxPooling3dUint8,
636 AsymmetricNonSquareMaxPooling3dUint8Test,
637 DataLayout::NDHWC)
638
639ARMNN_AUTO_TEST_CASE_WITH_THF(AsymmetricNonSquareAveragePooling3d,
640 AsymmetricNonSquareAveragePooling3dTest,
641 DataLayout::NDHWC)
642
643ARMNN_AUTO_TEST_CASE_WITH_THF(AsymmetricNonSquareAveragePooling3dUint8,
644 AsymmetricNonSquareAveragePooling3dUint8Test,
645 DataLayout::NDHWC)
646
647ARMNN_AUTO_TEST_CASE_WITH_THF(AsymmetricNonSquareL2Pooling3d,
648 AsymmetricNonSquareL2Pooling3dTest,
649 DataLayout::NDHWC)
650
651ARMNN_AUTO_TEST_CASE_WITH_THF(UNSUPPORTED_AsymmetricNonSquareMaxPooling3dWithPaddingOnlyPool,
652 AsymmetricNonSquareMaxPooling3dWithPaddingOnlyPoolTest,
653 DataLayout::NDHWC)
654
655ARMNN_AUTO_TEST_CASE_WITH_THF(UNSUPPORTED_AsymmetricNonSquareMaxPooling3dWithPaddingOnlyPoolUint8,
656 AsymmetricNonSquareMaxPooling3dWithPaddingOnlyPoolUint8Test,
657 DataLayout::NDHWC)
658
659ARMNN_AUTO_TEST_CASE_WITH_THF(UNSUPPORTED_AsymmetricNonSquareAveragePooling3dWithPaddingOnlyPool,
660 AsymmetricNonSquareAveragePooling3dWithPaddingOnlyPoolTest,
661 DataLayout::NDHWC)
662
663ARMNN_AUTO_TEST_CASE_WITH_THF(UNSUPPORTED_AsymmetricNonSquareAveragePooling3dWithPaddingOnlyPoolUint8,
664 AsymmetricNonSquareAveragePooling3dWithPaddingOnlyPoolUint8Test,
665 DataLayout::NDHWC)
666
667ARMNN_AUTO_TEST_CASE_WITH_THF(UNSUPPORTEDAsymmetricNonSquareL2Pooling3dWithPaddingOnlyPool,
668 AsymmetricNonSquareL2Pooling3dWithPaddingOnlyPoolTest,
669 DataLayout::NDHWC)
670
671ARMNN_AUTO_TEST_CASE_WITH_THF(UNSUPPORTEDAsymmetricNonSquareL2Pooling3dWithPaddingOnlyPoolUint8,
672 AsymmetricNonSquareL2Pooling3dWithPaddingOnlyPoolUint8Test,
673 DataLayout::NDHWC)
674
675ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleAveragePooling3d,
676 SimpleAveragePooling3dTest,
677 DataLayout::NDHWC)
678
679ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleAveragePooling3dUint8,
680 SimpleAveragePooling3dUint8Test,
681 DataLayout::NDHWC)
682
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]683// Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]684ARMNN_AUTO_TEST_CASE_WITH_THF(ConstantLinearActivation, ConstantLinearActivationTest)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]685
Teresa Charline32c8442020-03-05 14:41:50 +0000[diff] [blame]686// Sigmoid Activation / Logistic
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]687ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSigmoid, SimpleSigmoidTest)
688ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSigmoidUint8, SimpleSigmoidUint8Test)
Teresa Charline32c8442020-03-05 14:41:50 +0000[diff] [blame]689
690// BoundedReLU Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]691ARMNN_AUTO_TEST_CASE_WITH_THF(ReLu1, BoundedReLuUpperAndLowerBoundTest)
692ARMNN_AUTO_TEST_CASE_WITH_THF(ReLu6, BoundedReLuUpperBoundOnlyTest)
693ARMNN_AUTO_TEST_CASE_WITH_THF(ReLu1Uint8, BoundedReLuUint8UpperAndLowerBoundTest)
694ARMNN_AUTO_TEST_CASE_WITH_THF(ReLu6Uint8, BoundedReLuUint8UpperBoundOnlyTest)
Narumol Prangnawarat65d30962019-03-14 11:55:03 +0000[diff] [blame]695
Teresa Charline32c8442020-03-05 14:41:50 +0000[diff] [blame]696// ReLU Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]697ARMNN_AUTO_TEST_CASE_WITH_THF(ReLu, ReLuTest)
698ARMNN_AUTO_TEST_CASE_WITH_THF(ReLuUint8, ReLuUint8Test)
David Monahan3b3c3812020-02-25 09:03:29 +0000[diff] [blame]699
Teresa Charline32c8442020-03-05 14:41:50 +0000[diff] [blame]700// SoftReLU Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]701ARMNN_AUTO_TEST_CASE_WITH_THF(SoftReLu, SoftReLuTest)
Teresa Charline32c8442020-03-05 14:41:50 +0000[diff] [blame]702
703// LeakyReLU Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]704ARMNN_AUTO_TEST_CASE_WITH_THF(LeakyReLu, LeakyReLuTest)
Teresa Charline32c8442020-03-05 14:41:50 +0000[diff] [blame]705
706// Abs Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]707ARMNN_AUTO_TEST_CASE_WITH_THF(Abs, AbsTest)
Narumol Prangnawaratb98bbcf2019-04-15 14:14:27 +0100[diff] [blame]708
Sadik Armagan6095ba52019-09-13 17:07:19 +0100[diff] [blame]709// Sqrt Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]710ARMNN_AUTO_TEST_CASE_WITH_THF(Sqrt, SqrtTest)
711ARMNN_AUTO_TEST_CASE_WITH_THF(SqrtNN, SqrtNNTest)
Sadik Armagan6095ba52019-09-13 17:07:19 +0100[diff] [blame]712
Teresa Charline32c8442020-03-05 14:41:50 +0000[diff] [blame]713// Square Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]714ARMNN_AUTO_TEST_CASE_WITH_THF(Square, SquareTest)
Teresa Charline32c8442020-03-05 14:41:50 +0000[diff] [blame]715
716// Tanh Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]717ARMNN_AUTO_TEST_CASE_WITH_THF(Tanh, TanhTest)
Teresa Charline32c8442020-03-05 14:41:50 +0000[diff] [blame]718
719// Elu Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]720ARMNN_AUTO_TEST_CASE_WITH_THF(Elu, EluTest)
Teresa Charline32c8442020-03-05 14:41:50 +0000[diff] [blame]721
Narumol Prangnawarat65d30962019-03-14 11:55:03 +0000[diff] [blame]722// Softmax
Colm Donelanade8c1d2021-03-24 14:51:31 +0000[diff] [blame]723// Moved to NeonLayerTests_NDK_Bug.cpp
724//ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSoftmaxBeta1, SimpleSoftmaxTest, 1.0f)
725//ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSoftmaxBeta2, SimpleSoftmaxTest, 2.0f)
Sadik Armagan56785c72020-08-27 12:57:20 +0100[diff] [blame]726ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSoftmaxBeta1Uint8, SimpleSoftmaxUint8Test, 1.0f)
727ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSoftmaxBeta2Uint8, SimpleSoftmaxUint8Test, 2.0f)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]728
Keith Davis69e653f2020-07-02 11:49:26 +0100[diff] [blame]729// LogSoftmax
Colm Donelanade8c1d2021-03-24 14:51:31 +0000[diff] [blame]730// Moved to NeonLayerTests_NDK_Bug.cpp
731//ARMNN_AUTO_TEST_CASE_WITH_THF(LogSoftmaxFloat32_1, LogSoftmaxTest1<DataType::Float32>)
Keith Davis69e653f2020-07-02 11:49:26 +0100[diff] [blame]732
Mike Kelly0be3a882020-01-24 11:27:50 +0000[diff] [blame]733// Space To Batch Nd
Sadik Armagan714fe5b2020-08-27 12:46:36 +0100[diff] [blame]734ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdSimpleFloat32, SpaceToBatchNdSimpleFloat32Test)
735ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiChannelsFloat32, SpaceToBatchNdMultiChannelsFloat32Test)
736ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiBlockFloat32, SpaceToBatchNdMultiBlockFloat32Test)
737ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdPaddingFloat32, SpaceToBatchNdPaddingFloat32Test)
Mike Kelly0be3a882020-01-24 11:27:50 +0000[diff] [blame]738
Sadik Armagan714fe5b2020-08-27 12:46:36 +0100[diff] [blame]739ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdSimpleUint8, SpaceToBatchNdSimpleUint8Test)
740ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiChannelsUint8, SpaceToBatchNdMultiChannelsUint8Test)
741ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiBlockUint8, SpaceToBatchNdMultiBlockUint8Test)
742ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdPaddingUint8, SpaceToBatchNdPaddingUint8Test)
Mike Kelly0be3a882020-01-24 11:27:50 +0000[diff] [blame]743
Sadik Armagan714fe5b2020-08-27 12:46:36 +0100[diff] [blame]744ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdSimpleNhwcFloat32, SpaceToBatchNdSimpleNhwcFloat32Test)
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]745ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiChannelsNhwcFloat32,
746 SpaceToBatchNdMultiChannelsNhwcFloat32Test)
Sadik Armagan714fe5b2020-08-27 12:46:36 +0100[diff] [blame]747ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiBlockNhwcFloat32, SpaceToBatchNdMultiBlockNhwcFloat32Test)
748ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdPaddingNhwcFloat32, SpaceToBatchNdPaddingNhwcFloat32Test)
Mike Kelly0be3a882020-01-24 11:27:50 +0000[diff] [blame]749
Sadik Armagan714fe5b2020-08-27 12:46:36 +0100[diff] [blame]750ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdSimpleNhwcUint8, SpaceToBatchNdSimpleNhwcUint8Test)
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]751ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiChannelsNhwcUint8,
752 SpaceToBatchNdMultiChannelsNhwcUint8Test)
Sadik Armagan714fe5b2020-08-27 12:46:36 +0100[diff] [blame]753ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiBlockNhwcUint8, SpaceToBatchNdMultiBlockNhwcUint8Test)
754ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdPaddingNhwcUint8, SpaceToBatchNdPaddingNhwcUint8Test)
Mike Kelly0be3a882020-01-24 11:27:50 +0000[diff] [blame]755
Ellen Norris-Thompson29794572019-06-26 16:40:36 +0100[diff] [blame]756// SpaceToDepth
Sadik Armagan5e719842020-08-28 12:17:33 +0100[diff] [blame]757ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToDepthNchwAsymmQ8, SpaceToDepthNchwAsymmQ8Test)
758ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToDepthNhwcsymmQ8, SpaceToDepthNhwcAsymmQ8Test)
Ellen Norris-Thompson29794572019-06-26 16:40:36 +0100[diff] [blame]759
Sadik Armagan5e719842020-08-28 12:17:33 +0100[diff] [blame]760ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToDepthNhwc1Float32, SpaceToDepthNhwcFloat32Test1)
761ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToDepthNchw1Float32, SpaceToDepthNchwFloat32Test1)
Ellen Norris-Thompson29794572019-06-26 16:40:36 +0100[diff] [blame]762
Sadik Armagan5e719842020-08-28 12:17:33 +0100[diff] [blame]763ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToDepthNhwc2Float32, SpaceToDepthNhwcFloat32Test2)
764ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToDepthNchw2Float32, SpaceToDepthNchwFloat32Test2)
Ellen Norris-Thompson29794572019-06-26 16:40:36 +0100[diff] [blame]765
Sadik Armagan5e719842020-08-28 12:17:33 +0100[diff] [blame]766ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToDepthNhwcQSymm16, SpaceToDepthNhwcQSymm16Test)
767ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToDepthNchwQSymm16, SpaceToDepthNchwQSymm16Test)
Ellen Norris-Thompson29794572019-06-26 16:40:36 +0100[diff] [blame]768
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]769// Splitter
Sadik Armagan5e719842020-08-28 12:17:33 +0100[diff] [blame]770ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSplitterFloat32, SplitterFloat32Test)
771ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSplitterUint8, SplitterUint8Test)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]772
Sadik Armagan5e719842020-08-28 12:17:33 +0100[diff] [blame]773ARMNN_AUTO_TEST_CASE_WITH_THF(CopyViaSplitterFloat32, CopyViaSplitterFloat32Test)
774ARMNN_AUTO_TEST_CASE_WITH_THF(CopyViaSplitterUint8, CopyViaSplitterUint8Test)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]775
Jim Flynn4ed6c832019-05-20 11:02:46 +0100[diff] [blame]776// Concat
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]777ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConcat, ConcatTest)
778ARMNN_AUTO_TEST_CASE_WITH_THF(ConcatUint8, ConcatUint8Test)
779ARMNN_AUTO_TEST_CASE_WITH_THF(ConcatUint8DifferentInputOutputQParam,
Derek Lambertif90c56d2020-01-10 17:14:08 +0000[diff] [blame]780 ConcatDifferentInputOutputQParamTest<DataType::QAsymmU8>, false)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]781
782// Fully Connected
Finn Williams7faf9a82020-08-27 10:37:36 +0100[diff] [blame]783ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleFullyConnected, FullyConnectedFloat32Test, false, false)
784ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleFullyConnectedWithBias, FullyConnectedFloat32Test, true, false)
785ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleFullyConnectedWithTranspose, FullyConnectedFloat32Test, false, true)
786ARMNN_AUTO_TEST_CASE_WITH_THF(FullyConnectedLarge, FullyConnectedLargeTest, false)
787ARMNN_AUTO_TEST_CASE_WITH_THF(FullyConnectedLargeTransposed, FullyConnectedLargeTest, true)
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000[diff] [blame]788ARMNN_AUTO_TEST_CASE_WITH_THF(FullyConnectedUint8, FullyConnectedTest<DataType::QAsymmU8>, false, true)
789ARMNN_AUTO_TEST_CASE_WITH_THF(FullyConnectedBiasedUint8, FullyConnectedTest<DataType::QAsymmU8>, true, true)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]790
791// Add
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]792ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleAdd, AdditionTest)
793ARMNN_AUTO_TEST_CASE_WITH_THF(Add5d, Addition5dTest)
794ARMNN_AUTO_TEST_CASE_WITH_THF(AddBroadcast, AdditionBroadcastTest)
795ARMNN_AUTO_TEST_CASE_WITH_THF(AddBroadcast1Element, AdditionBroadcast1ElementTest)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]796
David Beckbc392452018-09-10 14:47:28 +0100[diff] [blame]797// Sub
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]798ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSub, SubtractionTest)
799ARMNN_AUTO_TEST_CASE_WITH_THF(SubBroadcast1Element, SubtractionBroadcast1ElementTest)
800ARMNN_AUTO_TEST_CASE_WITH_THF(SubBroadcast, SubtractionBroadcastTest)
801ARMNN_AUTO_TEST_CASE_WITH_THF(SubtractionUint8, SubtractionUint8Test)
802ARMNN_AUTO_TEST_CASE_WITH_THF(SubBroadcastUint8, SubtractionBroadcastUint8Test)
803ARMNN_AUTO_TEST_CASE_WITH_THF(SubBroadcast1ElementUint8, SubtractionBroadcast1ElementUint8Test)
David Beckbc392452018-09-10 14:47:28 +0100[diff] [blame]804
Pablo Telloe61f0712020-01-23 10:37:17 +0000[diff] [blame]805// Div
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]806ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleDivision, DivisionTest)
807ARMNN_AUTO_TEST_CASE_WITH_THF(DivisionByZero, DivisionByZeroTest)
808ARMNN_AUTO_TEST_CASE_WITH_THF(DivisionBroadcast1Element, DivisionBroadcast1ElementTest)
809ARMNN_AUTO_TEST_CASE_WITH_THF(DivisionBroadcast1DVector, DivisionBroadcast1DVectorTest)
Pablo Telloe61f0712020-01-23 10:37:17 +0000[diff] [blame]810
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]811// Mul
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]812ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMultiplication, MultiplicationTest)
813ARMNN_AUTO_TEST_CASE_WITH_THF(MultiplicationBroadcast1Element, MultiplicationBroadcast1ElementTest)
814ARMNN_AUTO_TEST_CASE_WITH_THF(MultiplicationBroadcast1DVector, MultiplicationBroadcast1DVectorTest)
815ARMNN_AUTO_TEST_CASE_WITH_THF(MultiplicationUint8, MultiplicationUint8Test)
816ARMNN_AUTO_TEST_CASE_WITH_THF(MultiplicationBroadcast1ElementUint8, MultiplicationBroadcast1ElementUint8Test)
817ARMNN_AUTO_TEST_CASE_WITH_THF(MultiplicationBroadcast1DVectorUint8, MultiplicationBroadcast1DVectorUint8Test)
818ARMNN_AUTO_TEST_CASE_WITH_THF(Multiplication5d, Multiplication5dTest)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]819
John Mcloughlin34c1c382023-05-17 15:08:36 +0100[diff] [blame^]820// SquaredDifference
821ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSquaredDifference, SquaredDifferenceTest)
822ARMNN_AUTO_TEST_CASE_WITH_THF(SquaredDiffBroadcast1Element, SquaredDiffBroadcast1ElementTest)
823ARMNN_AUTO_TEST_CASE_WITH_THF(SquaredDiffBroadcast, SquaredDiffBroadcastTest)
824
825ARMNN_AUTO_TEST_CASE_WITH_THF(SquaredDifferenceUint8, SquaredDifferenceUint8Test)
826ARMNN_AUTO_TEST_CASE_WITH_THF(SquaredDiffBroadcastUint8, SquaredDiffBroadcastUint8Test)
827ARMNN_AUTO_TEST_CASE_WITH_THF(SquaredDiffBroadcast1ElementUint8, SquaredDiffBroadcast1ElementUint8Test)
828
829ARMNN_AUTO_TEST_CASE_WITH_THF(SquaredDifferenceInt32, SquaredDifferenceInt32Test)
830ARMNN_AUTO_TEST_CASE_WITH_THF(SquaredDiffBroadcastInt32, SquaredDiffBroadcastInt32Test)
831ARMNN_AUTO_TEST_CASE_WITH_THF(SquaredDiffBroadcast1ElementInt32, SquaredDiffBroadcast1ElementInt32Test)
832
833// Power
834ARMNN_AUTO_TEST_CASE_WITH_THF(SimplePower, PowerTest)
835ARMNN_AUTO_TEST_CASE_WITH_THF(PowerBroadcast1Element, PowerBroadcast1ElementTest)
836ARMNN_AUTO_TEST_CASE_WITH_THF(PowerBroadcast, PowerBroadcastTest)
837
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]838// Batch Norm
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]839ARMNN_AUTO_TEST_CASE_WITH_THF(BatchNormFloat32, BatchNormFloat32Test)
840ARMNN_AUTO_TEST_CASE_WITH_THF(BatchNormFloat32Nhwc, BatchNormFloat32NhwcTest)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]841
David Monahan97451b42020-12-03 09:48:06 +0000[diff] [blame]842// Rank
843ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize1Float16, RankDimSize1Test<DataType::Float16>)
844ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize1Float32, RankDimSize1Test<DataType::Float32>)
845ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize1QAsymmU8, RankDimSize1Test<DataType::QAsymmU8>)
846ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize1Signed32, RankDimSize1Test<DataType::Signed32>)
847ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize1QSymmS16, RankDimSize1Test<DataType::QSymmS16>)
848ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize1QAsymmS8, RankDimSize1Test<DataType::QAsymmS8>)
David Monahan97451b42020-12-03 09:48:06 +0000[diff] [blame]849
850ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize2Float16, RankDimSize2Test<DataType::Float16>)
851ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize2Float32, RankDimSize2Test<DataType::Float32>)
852ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize2QAsymmU8, RankDimSize2Test<DataType::QAsymmU8>)
853ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize2Signed32, RankDimSize2Test<DataType::Signed32>)
854ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize2QSymmS16, RankDimSize2Test<DataType::QSymmS16>)
855ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize2QAsymmS8, RankDimSize2Test<DataType::QAsymmS8>)
David Monahan97451b42020-12-03 09:48:06 +0000[diff] [blame]856
857ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize3Float16, RankDimSize3Test<DataType::Float16>)
858ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize3Float32, RankDimSize3Test<DataType::Float32>)
859ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize3QAsymmU8, RankDimSize3Test<DataType::QAsymmU8>)
860ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize3Signed32, RankDimSize3Test<DataType::Signed32>)
861ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize3QSymmS16, RankDimSize3Test<DataType::QSymmS16>)
862ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize3QAsymmS8, RankDimSize3Test<DataType::QAsymmS8>)
David Monahan97451b42020-12-03 09:48:06 +0000[diff] [blame]863
864ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize4Float16, RankDimSize4Test<DataType::Float16>)
865ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize4Float32, RankDimSize4Test<DataType::Float32>)
866ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize4QAsymmU8, RankDimSize4Test<DataType::QAsymmU8>)
867ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize4Signed32, RankDimSize4Test<DataType::Signed32>)
868ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize4QSymmS16, RankDimSize4Test<DataType::QSymmS16>)
869ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize4QAsymmS8, RankDimSize4Test<DataType::QAsymmS8>)
David Monahan97451b42020-12-03 09:48:06 +0000[diff] [blame]870
Sadik Armagan0d4863d2019-10-09 14:26:32 +0100[diff] [blame]871// InstanceNormalization
Finn Williamsc43de6a2020-08-27 11:13:25 +0100[diff] [blame]872ARMNN_AUTO_TEST_CASE_WITH_THF(InstanceNormFloat32Nchw, InstanceNormFloat32Test, DataLayout::NCHW);
873ARMNN_AUTO_TEST_CASE_WITH_THF(InstanceNormFloat32Nhwc, InstanceNormFloat32Test, DataLayout::NHWC);
Sadik Armagan0d4863d2019-10-09 14:26:32 +0100[diff] [blame]874
Finn Williamsc43de6a2020-08-27 11:13:25 +0100[diff] [blame]875ARMNN_AUTO_TEST_CASE_WITH_THF(InstanceNormFloat32Nchw2, InstanceNormFloat32Test2, DataLayout::NCHW);
876ARMNN_AUTO_TEST_CASE_WITH_THF(InstanceNormFloat32Nhwc2, InstanceNormFloat32Test2, DataLayout::NHWC);
Sadik Armagan0d4863d2019-10-09 14:26:32 +0100[diff] [blame]877
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]878// Constant
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]879ARMNN_AUTO_TEST_CASE_WITH_THF(Constant, ConstantTest)
880ARMNN_AUTO_TEST_CASE_WITH_THF(ConstantUint8, ConstantUint8SimpleQuantizationScaleNoOffsetTest)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]881
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]882// Concat
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]883ARMNN_AUTO_TEST_CASE_WITH_THF(Concat1d, Concat1dTest)
884ARMNN_AUTO_TEST_CASE_WITH_THF(Concat1dUint8, Concat1dUint8Test)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]885
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]886ARMNN_AUTO_TEST_CASE_WITH_THF(Concat2dDim0, Concat2dDim0Test)
887ARMNN_AUTO_TEST_CASE_WITH_THF(Concat2dDim0Uint8, Concat2dDim0Uint8Test)
888ARMNN_AUTO_TEST_CASE_WITH_THF(Concat2dDim1, Concat2dDim1Test)
889ARMNN_AUTO_TEST_CASE_WITH_THF(Concat2dDim1Uint8, Concat2dDim1Uint8Test)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]890
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]891ARMNN_AUTO_TEST_CASE_WITH_THF(Concat2dDim0DiffInputDims, Concat2dDim0DiffInputDimsTest)
892ARMNN_AUTO_TEST_CASE_WITH_THF(Concat2dDim0DiffInputDimsUint8, Concat2dDim0DiffInputDimsUint8Test)
893ARMNN_AUTO_TEST_CASE_WITH_THF(Concat2dDim1DiffInputDims, Concat2dDim1DiffInputDimsTest)
894ARMNN_AUTO_TEST_CASE_WITH_THF(Concat2dDim1DiffInputDimsUint8, Concat2dDim1DiffInputDimsUint8Test)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]895
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]896ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim0, Concat3dDim0Test)
897ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim0Uint8, Concat3dDim0Uint8Test)
898ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim1, Concat3dDim1Test)
899ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim1Uint8, Concat3dDim1Uint8Test)
900ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim2, Concat3dDim2Test, false)
901ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim2Uint8, Concat3dDim2Uint8Test, false)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]902
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]903ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim0DiffInputDims, Concat3dDim0DiffInputDimsTest)
904ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim0DiffInputDimsUint8, Concat3dDim0DiffInputDimsUint8Test)
905ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim1DiffInputDims, Concat3dDim1DiffInputDimsTest)
906ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim1DiffInputDimsUint8, Concat3dDim1DiffInputDimsUint8Test)
907ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim2DiffInputDims, Concat3dDim2DiffInputDimsTest, false)
908ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim2DiffInputDimsUint8, Concat3dDim2DiffInputDimsUint8Test, false)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]909
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]910ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDim0, Concat4dDim0Test)
911ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDim1, Concat4dDim1Test)
912ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDim3, Concat4dDim3Test, false)
913ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDim0Uint8, Concat4dDim0Uint8Test)
914ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDim1Uint8, Concat4dDim1Uint8Test)
915ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDim3Uint8, Concat4dDim3Uint8Test, false)
narpra015cdda352018-11-19 15:30:27 +0000[diff] [blame]916
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]917ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDiffShapeDim0, Concat4dDiffShapeDim0Test)
918ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDiffShapeDim1, Concat4dDiffShapeDim1Test)
919ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDiffShapeDim3, Concat4dDiffShapeDim3Test, false)
920ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDiffShapeDim0Uint8, Concat4dDiffShapeDim0Uint8Test)
921ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDiffShapeDim1Uint8, Concat4dDiffShapeDim1Uint8Test)
922ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDiffShapeDim3Uint8, Concat4dDiffShapeDim3Uint8Test, false)
Sadik Armagan0d4863d2019-10-09 14:26:32 +0100[diff] [blame]923
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]924// L2 Normalization
Finn Williamsc43de6a2020-08-27 11:13:25 +0100[diff] [blame]925ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization1d, L2Normalization1dTest, DataLayout::NCHW)
926ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization2d, L2Normalization2dTest, DataLayout::NCHW)
927ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization3d, L2Normalization3dTest, DataLayout::NCHW)
928ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization4d, L2Normalization4dTest, DataLayout::NCHW)
Matteo Martincigh539b44d2018-10-01 09:26:39 +0100[diff] [blame]929
Colm Donelanade8c1d2021-03-24 14:51:31 +0000[diff] [blame]930// Moved to NeonLayerTests_NDK_Bug.cpp
931//ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization1dNhwc, L2Normalization1dTest, DataLayout::NHWC)
Finn Williamsc43de6a2020-08-27 11:13:25 +0100[diff] [blame]932ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization2dNhwc, L2Normalization2dTest, DataLayout::NHWC)
933ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization3dNhwc, L2Normalization3dTest, DataLayout::NHWC)
934ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization4dNhwc, L2Normalization4dTest, DataLayout::NHWC)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]935
Finn Williamsc43de6a2020-08-27 11:13:25 +0100[diff] [blame]936ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization2dShape, L2Normalization2dShapeTest);
Matthew Jackson82b15ed2019-07-25 16:14:30 +0100[diff] [blame]937
Finn Williamsc43de6a2020-08-27 11:13:25 +0100[diff] [blame]938ARMNN_AUTO_TEST_CASE_WITH_THF(L2NormalizationDefaultEpsilon, L2NormalizationDefaultEpsilonTest, DataLayout::NCHW)
939ARMNN_AUTO_TEST_CASE_WITH_THF(L2NormalizationNonDefaultEpsilon, L2NormalizationNonDefaultEpsilonTest, DataLayout::NCHW)
Ferran Balaguer0c64f462019-06-18 16:32:30 +0100[diff] [blame]940
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]941// Floor
Finn Williams7faf9a82020-08-27 10:37:36 +0100[diff] [blame]942ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleFloor, SimpleFloorTest<DataType::Float32>)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]943
Teresa Charlinf540eb82020-04-10 19:24:55 +0100[diff] [blame]944// Gather
Finn Williamsc43de6a2020-08-27 11:13:25 +0100[diff] [blame]945ARMNN_AUTO_TEST_CASE_WITH_THF(Gather1dParamsFloat32, Gather1dParamsFloat32Test)
946ARMNN_AUTO_TEST_CASE_WITH_THF(Gather1dParamsUint8, Gather1dParamsUint8Test)
947ARMNN_AUTO_TEST_CASE_WITH_THF(GatherMultiDimParamsFloat32, GatherMultiDimParamsFloat32Test)
948ARMNN_AUTO_TEST_CASE_WITH_THF(GatherMultiDimParamsUint8, GatherMultiDimParamsUint8Test)
Teresa Charlinf540eb82020-04-10 19:24:55 +0100[diff] [blame]949
Teresa Charlinbd22c7d2022-04-26 18:14:12 +0100[diff] [blame]950// GatherNd
951ARMNN_AUTO_TEST_CASE_WITH_THF(GatherNd2dFloat32, SimpleGatherNd2dTest<DataType::Float32>)
952ARMNN_AUTO_TEST_CASE_WITH_THF(GatherNd3dFloat32, SimpleGatherNd3dTest<DataType::Float32>)
953ARMNN_AUTO_TEST_CASE_WITH_THF(GatherNd4dFloat32, SimpleGatherNd4dTest<DataType::Float32>)
954ARMNN_AUTO_TEST_CASE_WITH_THF(GatherNd2dInt8, SimpleGatherNd2dTest<DataType::QAsymmS8>)
955ARMNN_AUTO_TEST_CASE_WITH_THF(GatherNd3dInt8, SimpleGatherNd3dTest<DataType::QAsymmS8>)
956ARMNN_AUTO_TEST_CASE_WITH_THF(GatherNd4dInt8, SimpleGatherNd4dTest<DataType::QAsymmS8>)
957ARMNN_AUTO_TEST_CASE_WITH_THF(GatherNd2dInt32, SimpleGatherNd2dTest<DataType::Signed32>)
958ARMNN_AUTO_TEST_CASE_WITH_THF(GatherNd3dInt32, SimpleGatherNd3dTest<DataType::Signed32>)
959ARMNN_AUTO_TEST_CASE_WITH_THF(GatherNd4dInt32, SimpleGatherNd4dTest<DataType::Signed32>)
960
Teresa Charlincedd34f2020-03-30 11:17:30 +0100[diff] [blame]961// Equal
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]962ARMNN_AUTO_TEST_CASE_WITH_THF(EqualSimple, EqualSimpleTest)
963ARMNN_AUTO_TEST_CASE_WITH_THF(EqualBroadcast1Element, EqualBroadcast1ElementTest)
964ARMNN_AUTO_TEST_CASE_WITH_THF(EqualBroadcast1dVector, EqualBroadcast1dVectorTest)
Teresa Charlincedd34f2020-03-30 11:17:30 +0100[diff] [blame]965
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]966ARMNN_AUTO_TEST_CASE_WITH_THF(EqualSimpleUint8, EqualSimpleUint8Test)
967ARMNN_AUTO_TEST_CASE_WITH_THF(EqualBroadcast1ElementUint8, EqualBroadcast1ElementUint8Test)
968ARMNN_AUTO_TEST_CASE_WITH_THF(EqualBroadcast1dVectorUint8, EqualBroadcast1dVectorUint8Test)
Teresa Charlincedd34f2020-03-30 11:17:30 +0100[diff] [blame]969
kevmay01eed85922019-01-28 08:37:25 +0000[diff] [blame]970// Greater
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]971ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterSimple, GreaterSimpleTest)
972ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterBroadcast1Element, GreaterBroadcast1ElementTest)
973ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterBroadcast1dVector, GreaterBroadcast1dVectorTest)
Aron Virginas-Tar3bc00ec2019-10-18 15:42:58 +0100[diff] [blame]974
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]975ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterSimpleUint8, GreaterSimpleUint8Test)
976ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterBroadcast1ElementUint8, GreaterBroadcast1ElementUint8Test)
977ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterBroadcast1dVectorUint8, GreaterBroadcast1dVectorUint8Test)
kevmay01eed85922019-01-28 08:37:25 +0000[diff] [blame]978
Teresa Charlincedd34f2020-03-30 11:17:30 +0100[diff] [blame]979// GreaterOrEqual
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]980ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterOrEqualSimple, GreaterOrEqualSimpleTest)
981ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterOrEqualBroadcast1Element, GreaterOrEqualBroadcast1ElementTest)
982ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterOrEqualBroadcast1dVector, GreaterOrEqualBroadcast1dVectorTest)
Teresa Charlincedd34f2020-03-30 11:17:30 +0100[diff] [blame]983
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]984ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterOrEqualSimpleUint8, GreaterOrEqualSimpleUint8Test)
985ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterOrEqualBroadcast1ElementUint8, GreaterOrEqualBroadcast1ElementUint8Test)
986ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterOrEqualBroadcast1dVectorUint8, GreaterOrEqualBroadcast1dVectorUint8Test)
Teresa Charlincedd34f2020-03-30 11:17:30 +0100[diff] [blame]987
988// Less
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]989ARMNN_AUTO_TEST_CASE_WITH_THF(LessSimple, LessSimpleTest)
990ARMNN_AUTO_TEST_CASE_WITH_THF(LessBroadcast1Element, LessBroadcast1ElementTest)
991ARMNN_AUTO_TEST_CASE_WITH_THF(LessBroadcast1dVector, LessBroadcast1dVectorTest)
Teresa Charlincedd34f2020-03-30 11:17:30 +0100[diff] [blame]992
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]993ARMNN_AUTO_TEST_CASE_WITH_THF(LessSimpleUint8, LessSimpleUint8Test)
994ARMNN_AUTO_TEST_CASE_WITH_THF(LessBroadcast1ElementUint8, LessBroadcast1ElementUint8Test)
995ARMNN_AUTO_TEST_CASE_WITH_THF(LessBroadcast1dVectorUint8, LessBroadcast1dVectorUint8Test)
Teresa Charlincedd34f2020-03-30 11:17:30 +0100[diff] [blame]996
997// LessOrEqual
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]998ARMNN_AUTO_TEST_CASE_WITH_THF(LessOrEqualSimple, LessOrEqualSimpleTest)
999ARMNN_AUTO_TEST_CASE_WITH_THF(LessOrEqualBroadcast1Element, LessOrEqualBroadcast1ElementTest)
1000ARMNN_AUTO_TEST_CASE_WITH_THF(LessOrEqualBroadcast1dVector, LessOrEqualBroadcast1dVectorTest)
Teresa Charlincedd34f2020-03-30 11:17:30 +0100[diff] [blame]1001
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]1002ARMNN_AUTO_TEST_CASE_WITH_THF(LessOrEqualSimpleUint8, LessOrEqualSimpleUint8Test)
1003ARMNN_AUTO_TEST_CASE_WITH_THF(LessOrEqualBroadcast1ElementUint8, LessOrEqualBroadcast1ElementUint8Test)
1004ARMNN_AUTO_TEST_CASE_WITH_THF(LessOrEqualBroadcast1dVectorUint8, LessOrEqualBroadcast1dVectorUint8Test)
Teresa Charlincedd34f2020-03-30 11:17:30 +0100[diff] [blame]1005
1006// NotEqual
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]1007ARMNN_AUTO_TEST_CASE_WITH_THF(NotEqualSimple, NotEqualSimpleTest)
1008ARMNN_AUTO_TEST_CASE_WITH_THF(NotEqualBroadcast1Element, NotEqualBroadcast1ElementTest)
1009ARMNN_AUTO_TEST_CASE_WITH_THF(NotEqualBroadcast1dVector, NotEqualBroadcast1dVectorTest)
Teresa Charlincedd34f2020-03-30 11:17:30 +0100[diff] [blame]1010
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]1011ARMNN_AUTO_TEST_CASE_WITH_THF(NotEqualSimpleUint8, NotEqualSimpleUint8Test)
1012ARMNN_AUTO_TEST_CASE_WITH_THF(NotEqualBroadcast1ElementUint8, NotEqualBroadcast1ElementUint8Test)
1013ARMNN_AUTO_TEST_CASE_WITH_THF(NotEqualBroadcast1dVectorUint8, NotEqualBroadcast1dVectorUint8Test)
Teresa Charlincedd34f2020-03-30 11:17:30 +0100[diff] [blame]1014
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1015// Reshape
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]1016ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleReshapeFloat32, SimpleReshapeTest<armnn::DataType::Float32>)
1017ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleReshapeInt8, SimpleReshapeTest<armnn::DataType::QAsymmS8>)
1018ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleReshapeUint8, SimpleReshapeTest<armnn::DataType::QAsymmU8>)
1019ARMNN_AUTO_TEST_CASE_WITH_THF(Reshape5d, Reshape5dTest<armnn::DataType::Float32>)
Narumol Prangnawarat0c95f4c2020-11-18 16:52:07 +0000[diff] [blame]1020ARMNN_AUTO_TEST_CASE_WITH_THF(ReshapeBoolean, ReshapeBooleanTest)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1021
Teresa Charlinf9de7712021-10-28 11:31:16 +0100[diff] [blame]1022// Pad - Constant
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]1023ARMNN_AUTO_TEST_CASE_WITH_THF(PadFloat322d, PadFloat322dTest)
1024ARMNN_AUTO_TEST_CASE_WITH_THF(PadFloat322dCustomPadding, PadFloat322dCustomPaddingTest)
1025ARMNN_AUTO_TEST_CASE_WITH_THF(PadFloat323d, PadFloat323dTest)
1026ARMNN_AUTO_TEST_CASE_WITH_THF(PadFloat324d, PadFloat324dTest)
Éanna Ó Catháin12055742019-01-25 10:01:40 +0000[diff] [blame]1027
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]1028ARMNN_AUTO_TEST_CASE_WITH_THF(PadUint82d, PadUint82dTest)
1029ARMNN_AUTO_TEST_CASE_WITH_THF(PadUint82dCustomPadding, PadUint82dCustomPaddingTest)
1030ARMNN_AUTO_TEST_CASE_WITH_THF(PadUint83d, PadUint83dTest)
1031ARMNN_AUTO_TEST_CASE_WITH_THF(PadUint84d, PadUint84dTest)
Teresa Charlinec8e1982019-07-02 16:24:09 +0100[diff] [blame]1032
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]1033ARMNN_AUTO_TEST_CASE_WITH_THF(Pad2dQSymm16, Pad2dTestCommon<DataType::QSymmS16>, 2.0f, 0, 0.0f)
1034ARMNN_AUTO_TEST_CASE_WITH_THF(Pad2dQSymm16CustomPadding, Pad2dTestCommon<DataType::QSymmS16>, 2.0f, 0, 1.0f)
1035ARMNN_AUTO_TEST_CASE_WITH_THF(Pad3dQSymm16, Pad3dTestCommon<DataType::QSymmS16>, 2.0f, 0)
1036ARMNN_AUTO_TEST_CASE_WITH_THF(Pad4dQSymm16, Pad4dTestCommon<DataType::QSymmS16>, 2.0f, 0)
FinnWilliamsArm8b797a82019-07-09 16:01:41 +0100[diff] [blame]1037
Teresa Charlinf9de7712021-10-28 11:31:16 +0100[diff] [blame]1038// Pad - Symmetric & Reflect
1039ARMNN_AUTO_TEST_CASE_WITH_THF(PadSymmetric2dFloat32, PadSymmetric2dFloat32Test)
1040ARMNN_AUTO_TEST_CASE_WITH_THF(PadReflect2dFloat32, PadReflect2dFloat32Test)
1041ARMNN_AUTO_TEST_CASE_WITH_THF(PadSymmetric3dFloat32, PadSymmetric3dFloat32Test)
1042ARMNN_AUTO_TEST_CASE_WITH_THF(PadReflect3dFloat32, PadReflect3dFloat32Test)
1043ARMNN_AUTO_TEST_CASE_WITH_THF(PadSymmetric3dUint8, PadSymmetric3dUint8Test)
1044ARMNN_AUTO_TEST_CASE_WITH_THF(PadReflect3dUint8, PadReflect3dUint8Test)
1045ARMNN_AUTO_TEST_CASE_WITH_THF(PadSymmetric3dInt8, PadSymmetric3dInt8Test)
1046ARMNN_AUTO_TEST_CASE_WITH_THF(PadReflect3dInt8, PadReflect3dInt8Test)
1047ARMNN_AUTO_TEST_CASE_WITH_THF(PadSymmetric4dFloat32, PadSymmetric4dFloat32Test)
1048ARMNN_AUTO_TEST_CASE_WITH_THF(PadReflect4dFloat32, PadReflect4dFloat32Test)
1049ARMNN_AUTO_TEST_CASE_WITH_THF(PadSymmetric4dUint8, PadSymmetric4dUint8Test)
1050ARMNN_AUTO_TEST_CASE_WITH_THF(PadReflect4dUint8, PadReflect4dUint8Test)
1051ARMNN_AUTO_TEST_CASE_WITH_THF(PadSymmetric4dInt8, PadSymmetric4dInt8Test)
1052ARMNN_AUTO_TEST_CASE_WITH_THF(PadReflect4dInt8, PadReflect4dInt8Test)
1053
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1054// Permute
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]1055ARMNN_AUTO_TEST_CASE_WITH_THF(SimplePermuteFloat32, SimplePermuteTest<DataType::Float32>)
1056ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteFloat32ValueSet1Test, PermuteValueSet1Test<DataType::Float32>)
1057ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteFloat32ValueSet2Test, PermuteValueSet2Test<DataType::Float32>)
1058ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteFloat32ValueSet3Test, PermuteValueSet3Test<DataType::Float32>)
1059ARMNN_AUTO_TEST_CASE_WITH_THF(SimplePermuteQASymmS8, SimplePermuteTest<DataType::QAsymmS8>)
1060ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteQASymmS8ValueSet1Test, PermuteValueSet1Test<DataType::QAsymmS8>)
1061ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteQASymmS8ValueSet2Test, PermuteValueSet2Test<DataType::QAsymmS8>)
1062ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteQASymmS8ValueSet3Test, PermuteValueSet3Test<DataType::QAsymmS8>)
1063ARMNN_AUTO_TEST_CASE_WITH_THF(SimplePermuteQASymm8, SimplePermuteTest<DataType::QAsymmU8>)
1064ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteQASymm8ValueSet1Test, PermuteValueSet1Test<DataType::QAsymmU8>)
1065ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteQASymm8ValueSet2Test, PermuteValueSet2Test<DataType::QAsymmU8>)
1066ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteQASymm8ValueSet3Test, PermuteValueSet3Test<DataType::QAsymmU8>)
surmeh01bceff2f2018-03-29 16:29:27 +0100[diff] [blame]1067
Les Bellde9011b2018-10-03 10:37:52 +0100[diff] [blame]1068// Lstm
Finn Williamsc43de6a2020-08-27 11:13:25 +0100[diff] [blame]1069ARMNN_AUTO_TEST_CASE_WITH_THF(LstmLayerFloat32WithCifgWithPeepholeNoProjection,
1070 LstmLayerFloat32WithCifgWithPeepholeNoProjectionTest)
1071ARMNN_AUTO_TEST_CASE_WITH_THF(LstmLayerFloat32NoCifgNoPeepholeNoProjection,
1072 LstmLayerFloat32NoCifgNoPeepholeNoProjectionTest)
1073ARMNN_AUTO_TEST_CASE_WITH_THF(LstmLayerFloat32NoCifgWithPeepholeWithProjection,
1074 LstmLayerFloat32NoCifgWithPeepholeWithProjectionTest)
Colm Donelanade8c1d2021-03-24 14:51:31 +0000[diff] [blame]1075// Moved to NeonLayerTests_NDK_Bug.cpp
1076//ARMNN_AUTO_TEST_CASE_WITH_THF(LstmLayerFloat32NoCifgWithPeepholeWithProjectionWithLayerNorm,
1077// LstmLayerFloat32NoCifgWithPeepholeWithProjectionWithLayerNormTest)
Les Bellde9011b2018-10-03 10:37:52 +0100[diff] [blame]1078
James Conroycc340932020-05-12 18:08:52 +0100[diff] [blame]1079// QLstm
Finn Williamsc43de6a2020-08-27 11:13:25 +0100[diff] [blame]1080ARMNN_AUTO_TEST_CASE_WITH_THF(QLstm, QLstmTest)
1081ARMNN_AUTO_TEST_CASE_WITH_THF(QLstm1, QLstmTest1)
1082ARMNN_AUTO_TEST_CASE_WITH_THF(QLstm2, QLstmTest2)
James Conroycc340932020-05-12 18:08:52 +0100[diff] [blame]1083
1084// QuantizedLstm
Finn Williamsc43de6a2020-08-27 11:13:25 +0100[diff] [blame]1085ARMNN_AUTO_TEST_CASE_WITH_THF(QuantizedLstm, QuantizedLstmTest)
Francis Murtagh4fc3c482019-08-02 13:20:54 +0100[diff] [blame]1086
Cathal Corbettfd5bec42022-03-03 15:13:23 +0000[diff] [blame]1087// Unidirectional Sequence Lstm
1088ARMNN_AUTO_TEST_CASE_WITH_THF(UnidirectionalSequenceLstmLayerFloat32TimeMajorSingleBatch,
1089 UnidirectionalSequenceLstmLayerFloat32TimeMajorSingleBatchTest)
1090ARMNN_AUTO_TEST_CASE_WITH_THF(UnidirectionalSequenceLstmLayerFloat32BatchMajorSingleBatch,
1091 UnidirectionalSequenceLstmLayerFloat32BatchMajorSingleBatchTest)
1092ARMNN_AUTO_TEST_CASE_WITH_THF(UnidirectionalSequenceLstmLayerFloat32,
1093 UnidirectionalSequenceLstmLayerFloat32Test)
1094ARMNN_AUTO_TEST_CASE_WITH_THF(UnidirectionalSequenceLstmLayerFloat32TimeMajor,
1095 UnidirectionalSequenceLstmLayerFloat32TimeMajorTest)
1096ARMNN_AUTO_TEST_CASE_WITH_THF(UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjection,
1097 UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest)
Cathal Corbettd83ffe02022-03-22 15:46:02 +0000[diff] [blame]1098// Moved to NeonLayerTests_NDK_Bug.cpp
1099//ARMNN_AUTO_TEST_CASE_WITH_THF(
1100// UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNorm,
1101// UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest)
Cathal Corbettfd5bec42022-03-03 15:13:23 +0000[diff] [blame]1102ARMNN_AUTO_TEST_CASE_WITH_THF(UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjection,
1103 UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest)
1104
Matthew Benthamfd899962018-12-31 15:49:42 +0000[diff] [blame]1105// Mean
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]1106ARMNN_AUTO_TEST_CASE_WITH_THF(MeanSimpleFloat32, MeanSimpleTest<DataType::Float32>)
1107ARMNN_AUTO_TEST_CASE_WITH_THF(MeanSimpleAxisFloat32, MeanSimpleAxisTest<DataType::Float32>)
1108ARMNN_AUTO_TEST_CASE_WITH_THF(MeanKeepDimsFloat32, MeanKeepDimsTest<DataType::Float32>)
1109ARMNN_AUTO_TEST_CASE_WITH_THF(MeanMultipleDimsFloat32, MeanMultipleDimsTest<DataType::Float32>)
1110ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts1Float32, MeanVts1Test<DataType::Float32>)
1111ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts2Float32, MeanVts2Test<DataType::Float32>)
1112ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts3Float32, MeanVts3Test<DataType::Float32>)
Matthew Benthamfd899962018-12-31 15:49:42 +0000[diff] [blame]1113
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]1114ARMNN_AUTO_TEST_CASE_WITH_THF(MeanSimpleQuantisedAsymmS8, MeanSimpleTest<DataType::QAsymmS8>)
1115ARMNN_AUTO_TEST_CASE_WITH_THF(MeanSimpleAxisQuantisedAsymmS8, MeanSimpleAxisTest<DataType::QAsymmS8>)
1116ARMNN_AUTO_TEST_CASE_WITH_THF(MeanKeepDimsQuantisedAsymmS8, MeanKeepDimsTest<DataType::QAsymmS8>)
1117ARMNN_AUTO_TEST_CASE_WITH_THF(MeanMultipleDimsQuantisedAsymmS8, MeanMultipleDimsTest<DataType::QAsymmS8>)
1118ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts1QuantisedAsymmS8, MeanVts1Test<DataType::QAsymmS8>)
1119ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts2QuantisedAsymmS8, MeanVts2Test<DataType::QAsymmS8>)
1120ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts3QuantisedAsymmS8, MeanVts3Test<DataType::QAsymmS8>)
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]1121
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]1122ARMNN_AUTO_TEST_CASE_WITH_THF(MeanSimpleQuantisedAsymm8, MeanSimpleTest<DataType::QAsymmU8>)
1123ARMNN_AUTO_TEST_CASE_WITH_THF(MeanSimpleAxisQuantisedAsymm8, MeanSimpleAxisTest<DataType::QAsymmU8>)
1124ARMNN_AUTO_TEST_CASE_WITH_THF(MeanKeepDimsQuantisedAsymm8, MeanKeepDimsTest<DataType::QAsymmU8>)
1125ARMNN_AUTO_TEST_CASE_WITH_THF(MeanMultipleDimsQuantisedAsymm8, MeanMultipleDimsTest<DataType::QAsymmU8>)
1126ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts1QuantisedAsymm8, MeanVts1Test<DataType::QAsymmU8>)
1127ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts2QuantisedAsymm8, MeanVts2Test<DataType::QAsymmU8>)
1128ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts3QuantisedAsymm8, MeanVts3Test<DataType::QAsymmU8>)
Matthew Benthamfd899962018-12-31 15:49:42 +0000[diff] [blame]1129
Nattapat Chaimanowong4e6597a2018-12-20 14:14:06 +0000[diff] [blame]1130// Max
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]1131ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaximum, MaximumSimpleTest)
1132ARMNN_AUTO_TEST_CASE_WITH_THF(MaximumBroadcast1Element, MaximumBroadcast1ElementTest)
1133ARMNN_AUTO_TEST_CASE_WITH_THF(MaximumBroadcast1DVector, MaximumBroadcast1DVectorTest)
1134ARMNN_AUTO_TEST_CASE_WITH_THF(MaximumUint8, MaximumUint8Test)
1135ARMNN_AUTO_TEST_CASE_WITH_THF(MaximumBroadcast1ElementUint8, MaximumBroadcast1ElementUint8Test)
1136ARMNN_AUTO_TEST_CASE_WITH_THF(MaximumBroadcast1DVectorUint8, MaximumBroadcast1DVectorUint8Test)
Nattapat Chaimanowong4e6597a2018-12-20 14:14:06 +0000[diff] [blame]1137
Conor Kennedy54b21692019-01-09 07:57:38 +0000[diff] [blame]1138// Min
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]1139ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMinimum1, MinimumBroadcast1ElementTest1)
1140ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMinimum2, MinimumBroadcast1ElementTest2)
1141ARMNN_AUTO_TEST_CASE_WITH_THF(Minimum1DVectorUint8, MinimumBroadcast1DVectorUint8Test)
Conor Kennedy54b21692019-01-09 07:57:38 +0000[diff] [blame]1142
narpra0155a97bc2018-10-02 14:35:53 +0100[diff] [blame]1143// Normalization
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]1144ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleNormalizationAcross, SimpleNormalizationAcrossTest)
1145ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleNormalizationWithin, SimpleNormalizationWithinTest)
1146ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleNormalizationAcrossNhwc, SimpleNormalizationAcrossNhwcTest)
Sadik Armagan483c8112021-06-01 09:24:52 +0100[diff] [blame]1147ARMNN_AUTO_TEST_CASE_WITH_THF(AcrossChannelNormalization, AcrossChannelNormalizationTest)
narpra0155a97bc2018-10-02 14:35:53 +0100[diff] [blame]1148
Sadik Armaganc625f002018-12-17 11:32:16 +0000[diff] [blame]1149// Resize Bilinear - NCHW data layout
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]1150ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeBilinear, SimpleResizeBilinearTest<DataType::Float32>, DataLayout::NCHW)
1151ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearNop, ResizeBilinearNopTest<DataType::Float32>, DataLayout::NCHW)
1152ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearSqMin, ResizeBilinearSqMinTest<DataType::Float32>, DataLayout::NCHW)
1153ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMin, ResizeBilinearMinTest<DataType::Float32>, DataLayout::NCHW)
1154ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMag, ResizeBilinearMagTest<DataType::Float32>, DataLayout::NCHW)
Sadik Armaganc625f002018-12-17 11:32:16 +0000[diff] [blame]1155
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]1156ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeBilinearUint8,
1157 SimpleResizeBilinearTest<DataType::QAsymmU8>,
1158 DataLayout::NCHW)
1159ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearNopUint8,
1160 ResizeBilinearNopTest<DataType::QAsymmU8>,
1161 DataLayout::NCHW)
1162ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearSqMinUint8,
1163 ResizeBilinearSqMinTest<DataType::QAsymmU8>,
1164 DataLayout::NCHW)
1165ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMinUint8,
1166 ResizeBilinearMinTest<DataType::QAsymmU8>,
1167 DataLayout::NCHW)
1168ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMagUint8,
1169 ResizeBilinearMagTest<DataType::QAsymmU8>,
1170 DataLayout::NCHW)
Conor Kennedydc1c0822019-02-01 07:23:17 +0000[diff] [blame]1171
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]1172ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeBilinear,
1173 HalfPixelCentersResizeBilinearTest<DataType::Float32>,
1174 DataLayout::NCHW)
1175ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeBilinear,
1176 AlignCornersResizeBilinearTest<DataType::Float32>,
1177 DataLayout::NCHW)
1178ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeBilinearInt8,
1179 HalfPixelCentersResizeBilinearTest<DataType::QAsymmS8>,
1180 DataLayout::NCHW)
1181ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeBilinearInt8,
1182 AlignCornersResizeBilinearTest<DataType::QAsymmS8>,
1183 DataLayout::NCHW)
1184ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeBilinearUint8,
1185 HalfPixelCentersResizeBilinearTest<DataType::QAsymmU8>,
1186 DataLayout::NCHW)
1187ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeBilinearUint8,
1188 AlignCornersResizeBilinearTest<DataType::QAsymmU8>,
1189 DataLayout::NCHW)
Teresa Charlin686f45c2020-07-23 17:59:00 +0100[diff] [blame]1190
Sadik Armaganc625f002018-12-17 11:32:16 +0000[diff] [blame]1191// Resize Bilinear - NHWC data layout
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]1192ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearNopNhwc,
1193 ResizeBilinearNopTest<DataType::Float32>,
1194 DataLayout::NHWC)
1195ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeBilinearNhwc,
1196 SimpleResizeBilinearTest<DataType::Float32>,
1197 DataLayout::NHWC)
1198ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearSqMinNhwc,
1199 ResizeBilinearSqMinTest<DataType::Float32>,
1200 DataLayout::NHWC)
1201ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMinNhwc,
1202 ResizeBilinearMinTest<DataType::Float32>,
1203 DataLayout::NHWC)
1204ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMagNhwc,
1205 ResizeBilinearMagTest<DataType::Float32>,
1206 DataLayout::NHWC)
Ellen Norris-Thompsonc4223d12019-06-10 14:15:12 +0100[diff] [blame]1207
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]1208ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearNopUint8Nhwc,
1209 ResizeBilinearNopTest<DataType::QAsymmU8>,
1210 DataLayout::NHWC)
1211ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeBilinearUint8Nhwc,
1212 SimpleResizeBilinearTest<DataType::QAsymmU8>,
1213 DataLayout::NHWC)
1214ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearSqMinUint8Nhwc,
1215 ResizeBilinearSqMinTest<DataType::QAsymmU8>,
1216 DataLayout::NHWC)
1217ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMinUint8Nhwc,
1218 ResizeBilinearMinTest<DataType::QAsymmU8>,
1219 DataLayout::NHWC)
1220ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMagUint8Nhwc,
1221 ResizeBilinearMagTest<DataType::QAsymmU8>,
1222 DataLayout::NHWC)
Sadik Armaganc625f002018-12-17 11:32:16 +0000[diff] [blame]1223
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]1224ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeBilinearNhwc,
1225 HalfPixelCentersResizeBilinearTest<DataType::Float32>,
1226 DataLayout::NHWC)
1227ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeBilinearNhwc,
1228 AlignCornersResizeBilinearTest<DataType::Float32>,
1229 DataLayout::NHWC)
1230ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeBilinearInt8Nhwc,
1231 HalfPixelCentersResizeBilinearTest<DataType::QAsymmS8>,
1232 DataLayout::NHWC)
1233ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeBilinearInt8Nhwc,
1234 AlignCornersResizeBilinearTest<DataType::QAsymmS8>,
1235 DataLayout::NHWC)
1236ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeBilinearUint8Nhwc,
1237 HalfPixelCentersResizeBilinearTest<DataType::QAsymmU8>,
1238 DataLayout::NHWC)
1239ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeBilinearUint8Nhwc,
1240 AlignCornersResizeBilinearTest<DataType::QAsymmU8>,
1241 DataLayout::NHWC)
Teresa Charlin686f45c2020-07-23 17:59:00 +0100[diff] [blame]1242
Ellen Norris-Thompson37e68682019-07-15 14:23:30 +0100[diff] [blame]1243// Resize NearestNeighbor - NCHW
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]1244ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeNearestNeighbor,
1245 SimpleResizeNearestNeighborTest<DataType::Float32>,
1246 DataLayout::NCHW)
1247ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborNop,
1248 ResizeNearestNeighborNopTest<DataType::Float32>,
1249 DataLayout::NCHW)
1250ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborSqMin,
1251 ResizeNearestNeighborSqMinTest<DataType::Float32>,
1252 DataLayout::NCHW)
1253ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMin,
1254 ResizeNearestNeighborMinTest<DataType::Float32>,
1255 DataLayout::NCHW)
1256ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMag,
1257 ResizeNearestNeighborMagTest<DataType::Float32>,
1258 DataLayout::NCHW, 0.1f, 50, 0.1f, 50)
Ellen Norris-Thompson37e68682019-07-15 14:23:30 +0100[diff] [blame]1259
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]1260ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeNearestNeighborUint8,
1261 SimpleResizeNearestNeighborTest<DataType::QAsymmU8>,
1262 DataLayout::NCHW)
1263ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborNopUint8,
1264 ResizeNearestNeighborNopTest<DataType::QAsymmU8>,
1265 DataLayout::NCHW)
1266ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborSqMinUint8,
1267 ResizeNearestNeighborSqMinTest<DataType::QAsymmU8>,
1268 DataLayout::NCHW)
1269ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMinUint8,
1270 ResizeNearestNeighborMinTest<DataType::QAsymmU8>,
1271 DataLayout::NCHW)
1272ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMagUint8,
1273 ResizeNearestNeighborMagTest<DataType::QAsymmU8>,
1274 DataLayout::NCHW, 0.1f, 50, 0.1f, 50)
Ellen Norris-Thompson37e68682019-07-15 14:23:30 +0100[diff] [blame]1275
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]1276ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeNearestNeighbour,
1277 HalfPixelCentersResizeNearestNeighbourTest<DataType::Float32>,
1278 DataLayout::NCHW)
1279ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeNearestNeighbour,
1280 AlignCornersResizeNearestNeighbourTest<DataType::Float32>,
1281 DataLayout::NCHW)
1282ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeNearestNeighbourInt8,
1283 HalfPixelCentersResizeNearestNeighbourTest<DataType::QAsymmS8>,
1284 DataLayout::NCHW)
1285ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeNearestNeighbourInt8,
1286 AlignCornersResizeNearestNeighbourTest<DataType::QAsymmS8>,
1287 DataLayout::NCHW)
1288ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeNearestNeighbourUint8,
1289 HalfPixelCentersResizeNearestNeighbourTest<DataType::QAsymmU8>,
1290 DataLayout::NCHW)
1291ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeNearestNeighbourUint8,
1292 AlignCornersResizeNearestNeighbourTest<DataType::QAsymmU8>,
1293 DataLayout::NCHW)
Teresa Charlin686f45c2020-07-23 17:59:00 +0100[diff] [blame]1294
Ellen Norris-Thompson37e68682019-07-15 14:23:30 +0100[diff] [blame]1295// Resize NearestNeighbor - NHWC
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]1296ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborNopNhwc,
1297 ResizeNearestNeighborNopTest<DataType::Float32>,
1298 DataLayout::NHWC)
1299ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeNearestNeighborNhwc,
1300 SimpleResizeNearestNeighborTest<DataType::Float32>,
1301 DataLayout::NHWC)
1302ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborSqMinNhwc,
1303 ResizeNearestNeighborSqMinTest<DataType::Float32>,
1304 DataLayout::NHWC)
1305ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMinNhwc,
1306 ResizeNearestNeighborMinTest<DataType::Float32>,
1307 DataLayout::NHWC)
1308ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMagNhwc,
1309 ResizeNearestNeighborMagTest<DataType::Float32>,
1310 DataLayout::NHWC, 0.1f, 50, 0.1f, 50)
Ellen Norris-Thompson37e68682019-07-15 14:23:30 +0100[diff] [blame]1311
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]1312ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborNopUint8Nhwc,
1313 ResizeNearestNeighborNopTest<DataType::QAsymmU8>,
1314 DataLayout::NHWC)
1315ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeNearestNeighborUint8Nhwc,
1316 SimpleResizeNearestNeighborTest<DataType::QAsymmU8>,
1317 DataLayout::NHWC)
1318ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborSqMinUint8Nhwc,
1319 ResizeNearestNeighborSqMinTest<DataType::QAsymmU8>,
1320 DataLayout::NHWC)
1321ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMinUint8Nhwc,
1322 ResizeNearestNeighborMinTest<DataType::QAsymmU8>,
1323 DataLayout::NHWC)
1324ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMagUint8Nhwc,
1325 ResizeNearestNeighborMagTest<DataType::QAsymmU8>,
1326 DataLayout::NHWC, 0.1f, 50, 0.1f, 50)
Ellen Norris-Thompson37e68682019-07-15 14:23:30 +0100[diff] [blame]1327
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]1328ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeNearestNeighbourNhwc,
1329 HalfPixelCentersResizeNearestNeighbourTest<DataType::Float32>,
1330 DataLayout::NHWC)
1331ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeNearestNeighbourNhwc,
1332 AlignCornersResizeNearestNeighbourTest<DataType::Float32>,
1333 DataLayout::NHWC)
1334ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeNearestNeighbourInt8Nhwc,
1335 HalfPixelCentersResizeNearestNeighbourTest<DataType::QAsymmS8>,
1336 DataLayout::NHWC)
1337ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeNearestNeighbourInt8Nhwc,
1338 AlignCornersResizeNearestNeighbourTest<DataType::QAsymmS8>,
1339 DataLayout::NHWC)
1340ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeNearestNeighbourUint8Nhwc,
1341 HalfPixelCentersResizeNearestNeighbourTest<DataType::QAsymmU8>,
1342 DataLayout::NHWC)
1343ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeNearestNeighbourUint8Nhwc,
1344 AlignCornersResizeNearestNeighbourTest<DataType::QAsymmU8>,
1345 DataLayout::NHWC)
Teresa Charlin686f45c2020-07-23 17:59:00 +0100[diff] [blame]1346
josh minor036f02d2019-11-15 14:53:22 -0600[diff] [blame]1347// Slice
1348ARMNN_AUTO_TEST_CASE(Slice4dFloat32, Slice4dFloat32Test)
1349ARMNN_AUTO_TEST_CASE(Slice3dFloat32, Slice3dFloat32Test)
1350ARMNN_AUTO_TEST_CASE(Slice2dFloat32, Slice2dFloat32Test)
1351ARMNN_AUTO_TEST_CASE(Slice1dFloat32, Slice1dFloat32Test)
1352ARMNN_AUTO_TEST_CASE(Slice4dUint8, Slice4dUint8Test)
1353ARMNN_AUTO_TEST_CASE(Slice3dUint8, Slice3dUint8Test)
1354ARMNN_AUTO_TEST_CASE(Slice2dUint8, Slice2dUint8Test)
1355ARMNN_AUTO_TEST_CASE(Slice1dUint8, Slice1dUint8Test)
1356ARMNN_AUTO_TEST_CASE(Slice4dInt16, Slice4dInt16Test)
1357ARMNN_AUTO_TEST_CASE(Slice3dInt16, Slice3dInt16Test)
1358ARMNN_AUTO_TEST_CASE(Slice2dInt16, Slice2dInt16Test)
1359ARMNN_AUTO_TEST_CASE(Slice1dInt16, Slice1dInt16Test)
1360
FinnWilliamsArm1fa19192019-08-02 17:26:31 +0100[diff] [blame]1361// Strided Slice
Sadik Armagan5e719842020-08-28 12:17:33 +0100[diff] [blame]1362ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice4dFloat32, StridedSlice4dFloat32Test)
1363ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice4dReverseFloat32, StridedSlice4dReverseFloat32Test)
1364ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSliceSimpleStrideFloat32, StridedSliceSimpleStrideFloat32Test)
1365ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSliceSimpleRangeMaskFloat32, StridedSliceSimpleRangeMaskFloat32Test)
1366ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSliceShrinkAxisMaskFloat32, StridedSliceShrinkAxisMaskFloat32Test)
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]1367ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSliceShrinkAxisMaskCTSFloat32,
1368 StridedSliceShrinkAxisMaskCTSFloat32Test)
Sadik Armagan5e719842020-08-28 12:17:33 +0100[diff] [blame]1369ARMNN_AUTO_TEST_CASE_WITH_THF(
1370 StridedSliceShrinkAxisMaskBitPosition0Dim3Float32, StridedSliceShrinkAxisMaskBitPosition0Dim3Float32Test)
1371ARMNN_AUTO_TEST_CASE_WITH_THF(
1372 StridedSliceShrinkAxisMaskBitPosition0Float32, StridedSliceShrinkAxisMaskBitPosition0Float32Test)
1373ARMNN_AUTO_TEST_CASE_WITH_THF(
1374 StridedSliceShrinkAxisMaskBitPosition1Float32, StridedSliceShrinkAxisMaskBitPosition1Float32Test)
1375ARMNN_AUTO_TEST_CASE_WITH_THF(
1376 StridedSliceShrinkAxisMaskBitPosition2Float32, StridedSliceShrinkAxisMaskBitPosition2Float32Test)
1377ARMNN_AUTO_TEST_CASE_WITH_THF(
1378 StridedSliceShrinkAxisMaskBitPosition3Float32, StridedSliceShrinkAxisMaskBitPosition3Float32Test)
1379ARMNN_AUTO_TEST_CASE_WITH_THF(
1380 StridedSliceShrinkAxisMaskBitPosition0And1Float32, StridedSliceShrinkAxisMaskBitPosition0And1Float32Test)
1381ARMNN_AUTO_TEST_CASE_WITH_THF(
1382 StridedSliceShrinkAxisMaskBitPosition0And2Float32, StridedSliceShrinkAxisMaskBitPosition0And2Float32Test)
1383ARMNN_AUTO_TEST_CASE_WITH_THF(
1384 StridedSliceShrinkAxisMaskBitPosition0And3Float32, StridedSliceShrinkAxisMaskBitPosition0And3Float32Test)
1385ARMNN_AUTO_TEST_CASE_WITH_THF(
1386 StridedSliceShrinkAxisMaskBitPosition0And1And3Float32, StridedSliceShrinkAxisMaskBitPosition0And1And3Float32Test)
1387ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice3dFloat32, StridedSlice3dFloat32Test)
1388ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice3dReverseFloat32, StridedSlice3dReverseFloat32Test)
1389ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice2dFloat32, StridedSlice2dFloat32Test)
1390ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice2dReverseFloat32, StridedSlice2dReverseFloat32Test)
FinnWilliamsArm1fa19192019-08-02 17:26:31 +0100[diff] [blame]1391
Sadik Armagan5e719842020-08-28 12:17:33 +0100[diff] [blame]1392ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice4dUint8, StridedSlice4dUint8Test)
1393ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice4dReverseUint8, StridedSlice4dReverseUint8Test)
1394ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSliceSimpleStrideUint8, StridedSliceSimpleStrideUint8Test)
1395ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSliceSimpleRangeMaskUint8, StridedSliceSimpleRangeMaskUint8Test)
1396ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSliceShrinkAxisMaskUint8, StridedSliceShrinkAxisMaskUint8Test)
1397ARMNN_AUTO_TEST_CASE_WITH_THF(
1398 StridedSliceShrinkAxisMaskBitPosition0Dim3Uint8, StridedSliceShrinkAxisMaskBitPosition0Dim3Uint8Test)
1399ARMNN_AUTO_TEST_CASE_WITH_THF(
1400 StridedSliceShrinkAxisMaskBitPosition0Uint8, StridedSliceShrinkAxisMaskBitPosition0Uint8Test)
1401ARMNN_AUTO_TEST_CASE_WITH_THF(
1402 StridedSliceShrinkAxisMaskBitPosition1Uint8, StridedSliceShrinkAxisMaskBitPosition1Uint8Test)
1403ARMNN_AUTO_TEST_CASE_WITH_THF(
1404 StridedSliceShrinkAxisMaskBitPosition2Uint8, StridedSliceShrinkAxisMaskBitPosition2Uint8Test)
1405ARMNN_AUTO_TEST_CASE_WITH_THF(
1406 StridedSliceShrinkAxisMaskBitPosition3Uint8, StridedSliceShrinkAxisMaskBitPosition3Uint8Test)
1407ARMNN_AUTO_TEST_CASE_WITH_THF(
1408 StridedSliceShrinkAxisMaskBitPosition0And1Uint8, StridedSliceShrinkAxisMaskBitPosition0And1Uint8Test)
1409ARMNN_AUTO_TEST_CASE_WITH_THF(
1410 StridedSliceShrinkAxisMaskBitPosition0And2Uint8, StridedSliceShrinkAxisMaskBitPosition0And2Uint8Test)
1411ARMNN_AUTO_TEST_CASE_WITH_THF(
1412 StridedSliceShrinkAxisMaskBitPosition0And3Uint8, StridedSliceShrinkAxisMaskBitPosition0And3Uint8Test)
1413ARMNN_AUTO_TEST_CASE_WITH_THF(
1414 StridedSliceShrinkAxisMaskBitPosition0And1And3Uint8, StridedSliceShrinkAxisMaskBitPosition0And1And3Uint8Test)
1415ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice3dUint8, StridedSlice3dUint8Test)
1416ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice3dReverseUint8, StridedSlice3dReverseUint8Test)
1417ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice2dUint8, StridedSlice2dUint8Test)
1418ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice2dReverseUint8, StridedSlice2dReverseUint8Test)
FinnWilliamsArm1fa19192019-08-02 17:26:31 +0100[diff] [blame]1419
Sadik Armaganfabc2892019-05-31 09:05:11 +0100[diff] [blame]1420// Quantize
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]1421ARMNN_AUTO_TEST_CASE_WITH_THF(QuantizeSimpleUint8, QuantizeSimpleUint8Test)
1422ARMNN_AUTO_TEST_CASE_WITH_THF(QuantizeClampUint8, QuantizeClampUint8Test)
Sadik Armaganfabc2892019-05-31 09:05:11 +0100[diff] [blame]1423
Nikhil Raj9b461482019-07-03 15:58:31 +0100[diff] [blame]1424// PReLU
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]1425ARMNN_AUTO_TEST_CASE_WITH_THF(PreluFloat32, PreluTest<DataType::Float32>)
1426ARMNN_AUTO_TEST_CASE_WITH_THF(PreluUint8, PreluTest<DataType::QAsymmU8>)
Nikhil Raj9b461482019-07-03 15:58:31 +0100[diff] [blame]1427
Matthew Jackson87f65ea2019-08-01 10:01:34 +0100[diff] [blame]1428// Stack
Sadik Armagan5e719842020-08-28 12:17:33 +0100[diff] [blame]1429ARMNN_AUTO_TEST_CASE_WITH_THF(Stack0Axis, StackAxis0Float32Test)
1430ARMNN_AUTO_TEST_CASE_WITH_THF(StackOutput4DAxis1, StackOutput4DAxis1Float32Test)
1431ARMNN_AUTO_TEST_CASE_WITH_THF(StackOutput4DAxis2, StackOutput4DAxis2Float32Test)
1432ARMNN_AUTO_TEST_CASE_WITH_THF(StackOutput4DAxis3, StackOutput4DAxis3Float32Test)
1433ARMNN_AUTO_TEST_CASE_WITH_THF(StackOutput3DInputs3, StackOutput3DInputs3Float32Test)
1434ARMNN_AUTO_TEST_CASE_WITH_THF(StackOutput5D, StackOutput5DFloat32Test)
Matthew Jackson87f65ea2019-08-01 10:01:34 +0100[diff] [blame]1435
Mike Kellyc9ea45a2020-02-28 18:11:58 +0000[diff] [blame]1436// Transpose
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]1437ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleTransposeFloat32, SimpleTransposeTest<DataType::Float32>)
1438ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeFloat32ValueSet1Test, TransposeValueSet1Test<DataType::Float32>)
1439ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeFloat32ValueSet2Test, TransposeValueSet2Test<DataType::Float32>)
1440ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeFloat32ValueSet3Test, TransposeValueSet3Test<DataType::Float32>)
1441ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleTransposeQASymms8, SimpleTransposeTest<DataType::QAsymmS8>)
1442ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeQASymmS8ValueSet1Test, TransposeValueSet1Test<DataType::QAsymmS8>)
1443ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeQASymmS8ValueSet2Test, TransposeValueSet2Test<DataType::QAsymmS8>)
1444ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeQASymmS8ValueSet3Test, TransposeValueSet3Test<DataType::QAsymmS8>)
1445ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleTransposeQASymm8, SimpleTransposeTest<DataType::QAsymmU8>)
1446ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeQASymm8ValueSet1Test, TransposeValueSet1Test<DataType::QAsymmU8>)
1447ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeQASymm8ValueSet2Test, TransposeValueSet2Test<DataType::QAsymmU8>)
1448ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeQASymm8ValueSet3Test, TransposeValueSet3Test<DataType::QAsymmU8>)
1449ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleTransposeQSymm16, SimpleTransposeTest<DataType::QSymmS16>)
1450ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeQSymm16ValueSet1Test, TransposeValueSet1Test<DataType::QSymmS16>)
1451ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeQSymm16ValueSet2Test, TransposeValueSet2Test<DataType::QSymmS16>)
1452ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeQSymm16ValueSet3Test, TransposeValueSet3Test<DataType::QSymmS16>)
Mike Kellyc9ea45a2020-02-28 18:11:58 +0000[diff] [blame]1453
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]1454// TransposeConvolution2d
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]1455ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleTransposeConvolution2dFloatNchw,
1456 SimpleTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1457 true,
1458 DataLayout::NCHW)
1459ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleTransposeConvolution2dFloatNhwc,
1460 SimpleTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1461 true,
1462 DataLayout::NHWC)
1463ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleTransposeConvolution2dUint8Nchw,
1464 SimpleTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1465 true,
1466 DataLayout::NCHW)
1467ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleTransposeConvolution2dUint8Nhwc,
1468 SimpleTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1469 true,
1470 DataLayout::NHWC)
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]1471
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]1472ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedSimpleTransposeConvolution2dFloatNchw,
1473 SimpleTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1474 false,
1475 DataLayout::NCHW)
1476ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedSimpleTransposeConvolution2dFloatNhwc,
1477 SimpleTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1478 true,
1479 DataLayout::NHWC)
1480ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedSimpleTransposeConvolution2dUint8Nchw,
1481 SimpleTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1482 true,
1483 DataLayout::NCHW)
1484ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedSimpleTransposeConvolution2dUint8Nhwc,
1485 SimpleTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1486 true,
1487 DataLayout::NHWC)
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]1488
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]1489ARMNN_AUTO_TEST_CASE_WITH_THF(PaddedTransposeConvolution2dFloatNchw,
1490 PaddedTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1491 true,
1492 DataLayout::NCHW)
1493ARMNN_AUTO_TEST_CASE_WITH_THF(PaddedTransposeConvolution2dFloatNhwc,
1494 PaddedTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1495 true,
1496 DataLayout::NHWC)
1497ARMNN_AUTO_TEST_CASE_WITH_THF(PaddedTransposeConvolution2dUint8Nchw,
1498 PaddedTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1499 true,
1500 DataLayout::NCHW)
1501ARMNN_AUTO_TEST_CASE_WITH_THF(PaddedTransposeConvolution2dUint8Nhwc,
1502 PaddedTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1503 true,
1504 DataLayout::NHWC)
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]1505
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]1506ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedPaddedTransposeConvolution2dFloatNchw,
1507 PaddedTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1508 false,
1509 DataLayout::NCHW)
1510ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedPaddedTransposeConvolution2dFloatNhwc,
1511 PaddedTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1512 true,
1513 DataLayout::NHWC)
1514ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedPaddedTransposeConvolution2dInt8Nchw,
1515 PaddedTransposeConvolution2dTest<DataType::QAsymmS8, DataType::Signed32>,
1516 true,
1517 DataLayout::NCHW)
1518ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedPaddedTransposeConvolution2dInt8Nhwc,
1519 PaddedTransposeConvolution2dTest<DataType::QAsymmS8, DataType::Signed32>,
1520 true,
1521 DataLayout::NHWC)
1522ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedPaddedTransposeConvolution2dUint8Nchw,
1523 PaddedTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1524 true,
1525 DataLayout::NCHW)
1526ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedPaddedTransposeConvolution2dUint8Nhwc,
1527 PaddedTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1528 true,
1529 DataLayout::NHWC)
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]1530
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]1531ARMNN_AUTO_TEST_CASE_WITH_THF(StridedTransposeConvolution2dFloatNchw,
1532 StridedTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1533 true,
1534 DataLayout::NCHW)
1535ARMNN_AUTO_TEST_CASE_WITH_THF(StridedTransposeConvolution2dFloatNhwc,
1536 StridedTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1537 true,
1538 DataLayout::NHWC)
1539ARMNN_AUTO_TEST_CASE_WITH_THF(StridedTransposeConvolution2dUint8Nchw,
1540 StridedTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1541 true,
1542 DataLayout::NCHW)
1543ARMNN_AUTO_TEST_CASE_WITH_THF(StridedTransposeConvolution2dUint8Nhwc,
1544 StridedTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1545 true,
1546 DataLayout::NHWC)
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]1547
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]1548ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedStridedTransposeConvolution2dFloatNchw,
1549 StridedTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1550 false,
1551 DataLayout::NCHW)
1552ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedStridedTransposeConvolution2dFloatNhwc,
1553 StridedTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1554 true,
1555 DataLayout::NHWC)
1556ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedStridedTransposeConvolution2dUint8Nchw,
1557 StridedTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1558 true,
1559 DataLayout::NCHW)
1560ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedStridedTransposeConvolution2dUint8Nhwc,
1561 StridedTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1562 true,
1563 DataLayout::NHWC)
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]1564
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]1565ARMNN_AUTO_TEST_CASE_WITH_THF(MultiChannelTransposeConvolution2dFloatNchw,
1566 MultiChannelTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1567 DataLayout::NCHW)
1568ARMNN_AUTO_TEST_CASE_WITH_THF(MultiChannelTransposeConvolution2dFloatNhwc,
1569 MultiChannelTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
1570 DataLayout::NHWC)
1571ARMNN_AUTO_TEST_CASE_WITH_THF(MultiChannelTransposeConvolution2dUint8Nchw,
1572 MultiChannelTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1573 DataLayout::NCHW)
1574ARMNN_AUTO_TEST_CASE_WITH_THF(MultiChannelTransposeConvolution2dUint8Nhwc,
1575 MultiChannelTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
1576 DataLayout::NHWC)
Aron Virginas-Tar7599aa42019-08-20 15:30:04 +0100[diff] [blame]1577
Aron Virginas-Tar914e4db2019-09-09 13:36:45 +0100[diff] [blame]1578// Abs
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]1579ARMNN_AUTO_TEST_CASE_WITH_THF(Abs2d, Abs2dTest<DataType::Float32>)
1580ARMNN_AUTO_TEST_CASE_WITH_THF(Abs3d, Abs3dTest<DataType::Float32>)
1581ARMNN_AUTO_TEST_CASE_WITH_THF(AbsZero, AbsZeroTest<DataType::Float32>)
Aron Virginas-Tar914e4db2019-09-09 13:36:45 +0100[diff] [blame]1582
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]1583ARMNN_AUTO_TEST_CASE_WITH_THF(Abs2dSigned32, Abs2dTest<DataType::Signed32>)
1584ARMNN_AUTO_TEST_CASE_WITH_THF(Abs3dSigned32, Abs3dTest<DataType::Signed32>)
1585ARMNN_AUTO_TEST_CASE_WITH_THF(AbsZeroSigned32, AbsZeroTest<DataType::Signed32>)
Kevin Mayec52c3a2020-04-24 09:42:31 +0100[diff] [blame]1586
Aron Virginas-Tar0dd3b432019-09-10 13:55:09 +0100[diff] [blame]1587// Rsqrt
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]1588ARMNN_AUTO_TEST_CASE_WITH_THF(Rsqrt2d, Rsqrt2dTest<DataType::Float32>)
1589ARMNN_AUTO_TEST_CASE_WITH_THF(Rsqrt3d, Rsqrt3dTest<DataType::Float32>)
1590ARMNN_AUTO_TEST_CASE_WITH_THF(RsqrtZero, RsqrtZeroTest<DataType::Float32>)
1591ARMNN_AUTO_TEST_CASE_WITH_THF(RsqrtNegative, RsqrtNegativeTest<DataType::Float32>)
Aron Virginas-Tar0dd3b432019-09-10 13:55:09 +0100[diff] [blame]1592
Teresa Charlin03027232022-05-09 17:27:08 +0100[diff] [blame]1593// Sqrt
1594ARMNN_AUTO_TEST_CASE_WITH_THF(Sqrt2d, Sqrt2dTest<DataType::Float32>)
1595ARMNN_AUTO_TEST_CASE_WITH_THF(Sqrt3d, Sqrt3dTest<DataType::Float32>)
1596ARMNN_AUTO_TEST_CASE_WITH_THF(SqrtZero, SqrtZeroTest<DataType::Float32>)
1597ARMNN_AUTO_TEST_CASE_WITH_THF(SqrtNegative, SqrtNegativeTest<DataType::Float32>)
1598
James Conroyd47a0642019-09-17 14:22:06 +0100[diff] [blame]1599// ArgMinMax
Sadik Armagan43528ac2020-08-27 09:06:40 +0100[diff] [blame]1600ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinFloat32, ArgMinSimpleTest<DataType::Float32>)
1601ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxFloat32, ArgMaxSimpleTest<DataType::Float32>)
1602ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinChannel, ArgMinChannelTest<DataType::Float32>)
1603ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxChannel, ArgMaxChannelTest<DataType::Float32>)
1604ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxHeight, ArgMaxHeightTest<DataType::Float32>)
1605ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinWidth, ArgMinWidthTest<DataType::Float32>)
James Conroyd47a0642019-09-17 14:22:06 +0100[diff] [blame]1606
Sadik Armagan43528ac2020-08-27 09:06:40 +0100[diff] [blame]1607ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinQAsymmS8, ArgMinSimpleTest<DataType::QAsymmS8>)
1608ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxQAsymmS8, ArgMaxSimpleTest<DataType::QAsymmS8>)
1609ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinChannelQAsymmS8, ArgMinChannelTest<DataType::QAsymmS8>)
1610ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxChannelQAsymmS8, ArgMaxChannelTest<DataType::QAsymmS8>)
1611ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxHeightQAsymmS8, ArgMaxHeightTest<DataType::QAsymmS8>)
1612ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinWidthQAsymmS8, ArgMinWidthTest<DataType::QAsymmS8>)
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]1613
Sadik Armagan43528ac2020-08-27 09:06:40 +0100[diff] [blame]1614ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinQAsymm8, ArgMinSimpleTest<DataType::QAsymmU8>)
1615ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxQAsymm8, ArgMaxSimpleTest<DataType::QAsymmU8>)
1616ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinChannelQAsymm8, ArgMinChannelTest<DataType::QAsymmU8>)
1617ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxChannelQAsymm8, ArgMaxChannelTest<DataType::QAsymmU8>)
1618ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxHeightQAsymm8, ArgMaxHeightTest<DataType::QAsymmU8>)
1619ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinWidthQAsymm8, ArgMinWidthTest<DataType::QAsymmU8>)
Francis Murtagh52ec3462019-11-19 12:24:19 +0000[diff] [blame]1620
Sadik Armaganac472102020-03-24 09:54:36 +0000[diff] [blame]1621// Neg
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]1622ARMNN_AUTO_TEST_CASE_WITH_THF(Neg2d, Neg2dTest<DataType::Float32>)
1623ARMNN_AUTO_TEST_CASE_WITH_THF(Neg3d, Neg3dTest<DataType::Float32>)
1624ARMNN_AUTO_TEST_CASE_WITH_THF(NegZero, NegZeroTest<DataType::Float32>)
1625ARMNN_AUTO_TEST_CASE_WITH_THF(NegNegative, NegNegativeTest<DataType::Float32>)
Sadik Armaganac472102020-03-24 09:54:36 +0000[diff] [blame]1626
Sadik Armagan9fabf432020-05-27 13:40:58 +0100[diff] [blame]1627// Exp
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]1628ARMNN_AUTO_TEST_CASE_WITH_THF(Exp2d, Exp2dTest<DataType::Float32>)
Teresa Charlin50de4fa2021-05-31 18:47:33 +0100[diff] [blame]1629ARMNN_AUTO_TEST_CASE_WITH_THF(Exp3d, Exp3dTest<DataType::Float32>)
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]1630ARMNN_AUTO_TEST_CASE_WITH_THF(ExpZero, ExpZeroTest<DataType::Float32>)
1631ARMNN_AUTO_TEST_CASE_WITH_THF(ExpNegative, ExpNegativeTest<DataType::Float32>)
Sadik Armagan9fabf432020-05-27 13:40:58 +0100[diff] [blame]1632
Teresa Charlin50de4fa2021-05-31 18:47:33 +0100[diff] [blame]1633// Log
1634ARMNN_AUTO_TEST_CASE_WITH_THF(Log2d, Log2dTest<DataType::Float32>)
1635ARMNN_AUTO_TEST_CASE_WITH_THF(Log3d, Log3dTest<DataType::Float32>)
1636ARMNN_AUTO_TEST_CASE_WITH_THF(LogZero, LogZeroTest<DataType::Float32>)
1637ARMNN_AUTO_TEST_CASE_WITH_THF(LogNegative, LogNegativeTest<DataType::Float32>)
1638
1639// Sin
1640ARMNN_AUTO_TEST_CASE_WITH_THF(Sin2d, Sin2dTest<DataType::Float32>)
1641ARMNN_AUTO_TEST_CASE_WITH_THF(Sin3d, Sin3dTest<DataType::Float32>)
1642ARMNN_AUTO_TEST_CASE_WITH_THF(SinZero, SinZeroTest<DataType::Float32>)
1643ARMNN_AUTO_TEST_CASE_WITH_THF(SinNegative, SinNegativeTest<DataType::Float32>)
1644
Sadik Armagana792a052020-06-23 16:22:23 +0100[diff] [blame]1645// Fill
Finn Williams87020072020-08-26 16:19:15 +0100[diff] [blame]1646ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleFill, SimpleFillTest<DataType::Float32>)
1647ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleFillF16, SimpleFillTest<DataType::Float16>)
1648ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleFillS32, SimpleFillTest<DataType::Signed32>)
Sadik Armagana792a052020-06-23 16:22:23 +0100[diff] [blame]1649
James Conroy177df1e2020-11-13 10:18:51 +0000[diff] [blame]1650// Logical
1651ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalNot, LogicalNotTest)
1652ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalNotInt, LogicalNotIntTest)
1653
1654ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalAnd, LogicalAndTest)
1655ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalAndInt, LogicalAndIntTest)
1656ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalAndBroadcast1, LogicalAndBroadcast1Test)
1657ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalAndBroadcast2, LogicalAndBroadcast2Test)
1658ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalAndBroadcast3, LogicalAndBroadcast3Test)
1659
1660ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalOr, LogicalOrTest)
1661ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalOrInt, LogicalOrIntTest)
1662ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalOrBroadcast1, LogicalOrBroadcast1Test)
1663ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalOrBroadcast2, LogicalOrBroadcast2Test)
1664ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalOrBroadcast3, LogicalOrBroadcast3Test)
1665
Sadik Armagana2747482021-02-09 10:28:54 +0000[diff] [blame]1666// ReduceSum
Colm Donelanade8c1d2021-03-24 14:51:31 +0000[diff] [blame]1667// Moved to NeonLayerTests_NDK_Bug.cpp
1668//ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceSumFloat32, ReduceSumSimpleTest<DataType::Float32>)
Sadik Armagana2747482021-02-09 10:28:54 +0000[diff] [blame]1669ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceSumSingleAxisFloat32_1, ReduceSumSingleAxisTest1<DataType::Float32>)
1670ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceSumSingleAxisFloat32_2, ReduceSumSingleAxisTest2<DataType::Float32>)
Colm Donelanade8c1d2021-03-24 14:51:31 +0000[diff] [blame]1671// Moved to NeonLayerTests_NDK_Bug.cpp
1672//ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceSumSingleAxisFloat32_3, ReduceSumSingleAxisTest3<DataType::Float32>)
Sadik Armagana2747482021-02-09 10:28:54 +0000[diff] [blame]1673
Teresa Charlin4e3e8312021-08-05 12:34:37 +0100[diff] [blame]1674// ReduceProd
1675ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceProdSingleAxisFloat32_1, ReduceProdSingleAxisTest1<DataType::Float32>)
1676ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceProdSingleAxisFloat32_2, ReduceProdSingleAxisTest2<DataType::Float32>)
1677
Sadik Armagana2747482021-02-09 10:28:54 +0000[diff] [blame]1678// ReduceMax
1679ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceMaxFloat32, ReduceMaxSimpleTest<DataType::Float32>)
1680ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceMaxNegativeAxisFloat32, ReduceMaxNegativeAxisTest<DataType::Float32>)
1681ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceMax2Float32, ReduceMaxSimpleTest2<DataType::Float32>)
1682
1683// ReduceMin
1684ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceMinFloat32, ReduceMinSimpleTest<DataType::Float32>)
1685ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceMinNegativeAxisFloat32, ReduceMinNegativeAxisTest<DataType::Float32>)
1686
Sadik Armagan48f011e2021-04-21 10:50:34 +0100[diff] [blame]1687// Cast
1688ARMNN_AUTO_TEST_CASE_WITH_THF(CastInt32ToFloat, CastInt32ToFloat2dTest)
1689ARMNN_AUTO_TEST_CASE_WITH_THF(CastInt8AsymmToFloat, CastInt8AsymmToFloat2dTest)
1690ARMNN_AUTO_TEST_CASE_WITH_THF(CastUIntToFloat, CastUInt8ToFloat2dTest)
1691ARMNN_AUTO_TEST_CASE_WITH_THF(CastFloatToIn8, CastFloat32ToInt82dTest)
1692ARMNN_AUTO_TEST_CASE_WITH_THF(CastFloatToUInt8, CastFloat32ToUInt82dTest)
1693
Teresa Charline89dd692021-09-01 16:30:34 +0100[diff] [blame]1694// ChannelShuffle
1695ARMNN_AUTO_TEST_CASE_WITH_THF(ChannelShuffle4DFloat32, ChannelShuffle4DTest<DataType::Float32>)
1696ARMNN_AUTO_TEST_CASE_WITH_THF(ChannelShuffle4DQAsymmU8, ChannelShuffle4DTest<DataType::QAsymmU8>)
Teresa Charlineb852bb2021-09-30 12:44:33 +0100[diff] [blame]1697ARMNN_AUTO_TEST_CASE_WITH_THF(ChannelShuffle4DQAsymmS8, ChannelShuffle4DTest<DataType::QAsymmS8>)
Sadik Armagan48f011e2021-04-21 10:50:34 +0100[diff] [blame]1698
Matthew Benthamcfbb2712023-04-27 10:40:32 +0000[diff] [blame]1699// Convert from Float16 to Float32
1700ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvertFp16ToFp32, SimpleConvertFp16ToFp32Test)
1701// Convert from Float32 to Float16
1702ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvertFp32ToFp16, SimpleConvertFp32ToFp16Test)
1703
Matteo Martincighdb16dd32019-08-27 16:41:11 +0100[diff] [blame]1704#if defined(ARMNNREF_ENABLED)
Matteo Martincighe67edb22019-08-14 14:05:46 +0100[diff] [blame]1705
1706// The ARMNN_COMPARE_REF_AUTO_TEST_CASE and the ARMNN_COMPARE_REF_FIXTURE_TEST_CASE test units are not available
1707// if the reference backend is not built
1708
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1709// ============================================================================
1710// COMPARE tests
1711
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]1712ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareConv2dWithReference, CompareConvolution2dTest)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1713
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]1714ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareDepthwiseConv2dWithReferenceFloat32,
Nattapat Chaimanowong649dd952019-01-22 16:10:44 +0000[diff] [blame]1715 CompareDepthwiseConvolution2dFloatTest,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]1716 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]1717ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareDepthwiseConv2dWithReferenceUint8,
Nattapat Chaimanowong649dd952019-01-22 16:10:44 +0000[diff] [blame]1718 CompareDepthwiseConvolution2dUint8Test,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]1719 DataLayout::NCHW)
jimfly017af00da2018-10-31 14:43:53 +0000[diff] [blame]1720
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]1721ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareDepthwiseConv2dWithReferenceFloat32Nhwc,
Nattapat Chaimanowong649dd952019-01-22 16:10:44 +0000[diff] [blame]1722 CompareDepthwiseConvolution2dFloatTest,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]1723 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]1724ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareDepthwiseConv2dWithReferenceUint8Nhwc,
Nattapat Chaimanowong649dd952019-01-22 16:10:44 +0000[diff] [blame]1725 CompareDepthwiseConvolution2dUint8Test,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]1726 DataLayout::NHWC)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1727
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]1728ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareNormalizationWithinWithReference, CompareNormalizationTest,
1729 NormalizationAlgorithmChannel::Within,
1730 NormalizationAlgorithmMethod::LocalBrightness)
1731ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareNormalizationAcrossWithReference, CompareNormalizationTest,
1732 NormalizationAlgorithmChannel::Across,
1733 NormalizationAlgorithmMethod::LocalBrightness)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1734
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]1735ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareMaxPooling2dWithReference, ComparePooling2dTest, PoolingAlgorithm::Max)
1736ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareMaxPooling2dWithReferenceUint8, ComparePooling2dUint8Test,
1737 PoolingAlgorithm::Max)
1738ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareAveragePooling2dWithReference, ComparePooling2dTest,
1739 PoolingAlgorithm::Average)
1740ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareAveragePooling2dWithReferenceUint8, ComparePooling2dUint8Test,
1741 PoolingAlgorithm::Average)
1742ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareL2Pooling2dWithReference, ComparePooling2dTest, PoolingAlgorithm::L2)
1743ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(UNSUPPORTED_CompareL2Pooling2dWithReferenceUint8, ComparePooling2dUint8Test,
1744 PoolingAlgorithm::L2)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1745
Ryan OShea19e79422022-05-04 00:38:03 +0100[diff] [blame]1746ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareMaxPooling3dWithRef, ComparePooling3dTest, PoolingAlgorithm::Max,
1747 DataLayout::NDHWC)
1748ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareAveragePooling3dWithRef, ComparePooling3dTest,
1749 PoolingAlgorithm::Average, DataLayout::NDHWC)
1750ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareL2Pooling3dWithRef, ComparePooling3dTest, PoolingAlgorithm::L2,
1751 DataLayout::NDHWC)
1752
Colm Donelanade8c1d2021-03-24 14:51:31 +0000[diff] [blame]1753// Moved to NeonLayerTests_NDK_Bug.cpp
1754//ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareSoftmaxBeta1WithReference, CompareSoftmaxTest, 1.0f)
1755//ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareSoftmaxBeta2WithReference, CompareSoftmaxTest, 2.0f)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1756
Sadik Armagan56785c72020-08-27 12:57:20 +0100[diff] [blame]1757ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareSoftmaxUint8Beta1WithReference, CompareSoftmaxUint8Test, 1.0f)
1758ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareSoftmaxUint8Beta2WithReference, CompareSoftmaxUint8Test, 2.0f)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1759
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]1760ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareAddition, CompareAdditionTest)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1761
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]1762ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareMultiplicationWithReference, CompareMultiplicationTest)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1763
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]1764ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareBatchNorm, CompareBatchNormTest)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1765
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]1766ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareReLu1, CompareBoundedReLuTest, 1.0f, -1.0f)
1767ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareReLu6, CompareBoundedReLuTest, 6.0f, 0.0f)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1768
1769// ============================================================================
1770// FIXTURE tests
1771
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]1772ARMNN_COMPARE_REF_FIXTURE_TEST_CASE_WITH_THF(CompareSigmoidActivationWithReference, ActivationFixture,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]1773 CompareActivationTest, ActivationFunction::Sigmoid, 5u)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1774
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]1775ARMNN_COMPARE_REF_FIXTURE_TEST_CASE_WITH_THF(CompareTanhActivationWithReference, ActivationFixture,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]1776 CompareActivationTest, ActivationFunction::TanH, 5u)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1777
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]1778ARMNN_COMPARE_REF_FIXTURE_TEST_CASE_WITH_THF(CompareLinearActivationWithReference, ActivationFixture,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]1779 CompareActivationTest, ActivationFunction::Linear, 5u)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1780
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]1781ARMNN_COMPARE_REF_FIXTURE_TEST_CASE_WITH_THF(CompareReLuActivationWithReference, ActivationFixture,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]1782 CompareActivationTest, ActivationFunction::ReLu, 5u)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1783
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]1784ARMNN_COMPARE_REF_FIXTURE_TEST_CASE_WITH_THF(CompareBoundedReLuActivationWithReference, ActivationFixture,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]1785 CompareActivationTest, ActivationFunction::BoundedReLu, 5u)
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]1786ARMNN_COMPARE_REF_FIXTURE_TEST_CASE_WITH_THF(CompareBoundedReLuActivationWithReferenceUint8, ActivationFixture,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]1787 CompareActivationUint8Test, ActivationFunction::BoundedReLu)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1788
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]1789ARMNN_COMPARE_REF_FIXTURE_TEST_CASE_WITH_THF(CompareSoftReLuActivationWithReference, ActivationFixture,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]1790 CompareActivationTest, ActivationFunction::SoftReLu, 1u)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1791
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]1792ARMNN_COMPARE_REF_FIXTURE_TEST_CASE_WITH_THF(CompareLeakyReLuActivationWithReference, ActivationFixture,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]1793 CompareActivationTest, ActivationFunction::LeakyReLu, 5u)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1794
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]1795ARMNN_COMPARE_REF_FIXTURE_TEST_CASE_WITH_THF(CompareAbsActivationWithReference, ActivationFixture,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]1796 CompareActivationTest, ActivationFunction::Abs, 5u)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1797
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]1798ARMNN_COMPARE_REF_FIXTURE_TEST_CASE_WITH_THF(CompareSqrtActivationWithReference, PositiveActivationFixture,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]1799 CompareActivationTest, ActivationFunction::Sqrt, 5u)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1800
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]1801ARMNN_COMPARE_REF_FIXTURE_TEST_CASE_WITH_THF(CompareSquareActivationWithReference, ActivationFixture,
Sadik Armagan581742d2019-08-12 14:11:37 +0100[diff] [blame]1802 CompareActivationTest, ActivationFunction::Square, 5u)
Matteo Martincighe67edb22019-08-14 14:05:46 +0100[diff] [blame]1803
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]1804ARMNN_COMPARE_REF_FIXTURE_TEST_CASE_WITH_THF(CompareEluActivationWithReference, ActivationFixture,
Teresa Charline32c8442020-03-05 14:41:50 +0000[diff] [blame]1805 CompareActivationTest, ActivationFunction::Elu, 5u)
1806
Matteo Martincighe67edb22019-08-14 14:05:46 +0100[diff] [blame]1807#endif
1808
Sadik Armagan1625efc2021-06-10 18:24:34 +0100[diff] [blame]1809}