Gitiles
Code Review Sign In
review.mlplatform.org / ml / armnn / 18515e2e77d04e5065aa0d2e850972d67446260d / . / src / backends / reference / test / RefLayerTests.cpp
blob: 04b215a9276727ee322e40e8da6cfb05cfade5d2 [file] [log] [blame]
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1//
2// Copyright © 2017 Arm Ltd. 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 "RefWorkloadFactoryHelper.hpp"
7
Aron Virginas-Tarc9cc8042018-11-01 16:15:57 +0000[diff] [blame]8#include <test/TensorHelpers.hpp>
9#include <test/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 <reference/RefWorkloadFactory.hpp>
Narumol Prangnawarate0a4ad82019-02-04 19:05:27 +0000[diff] [blame]12#include <backendsCommon/test/DetectionPostProcessLayerTestImpl.hpp>
Aron Virginas-Tarc9cc8042018-11-01 16:15:57 +0000[diff] [blame]13#include <backendsCommon/test/LayerTests.hpp>
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]14
arovir0143095f32018-10-09 18:04:24 +0100[diff] [blame]15#include <boost/test/unit_test.hpp>
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]16
17BOOST_AUTO_TEST_SUITE(Compute_Reference)
18using FactoryType = armnn::RefWorkloadFactory;
19
20// ============================================================================
21// UNIT tests
22
23// Convolution
jimfly010a088a62018-10-25 17:05:05 +0100[diff] [blame]24ARMNN_AUTO_TEST_CASE(SimpleConvolution2d3x5, SimpleConvolution2d3x5Test, true, armnn::DataLayout::NCHW)
25ARMNN_AUTO_TEST_CASE(SimpleConvolution2d3x5Uint8, SimpleConvolution2d3x5Uint8Test, true, armnn::DataLayout::NCHW)
narpra015f703182018-10-26 16:24:58 +0100[diff] [blame]26ARMNN_AUTO_TEST_CASE(SimpleConvolution2d3x5Nhwc, SimpleConvolution2d3x5Test, true, armnn::DataLayout::NHWC)
27ARMNN_AUTO_TEST_CASE(SimpleConvolution2d3x5Uint8Nhwc, SimpleConvolution2d3x5Uint8Test, true, armnn::DataLayout::NHWC)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]28
jimfly010a088a62018-10-25 17:05:05 +0100[diff] [blame]29ARMNN_AUTO_TEST_CASE(UnbiasedConvolution2d, SimpleConvolution2d3x5Test, false, armnn::DataLayout::NCHW)
30ARMNN_AUTO_TEST_CASE(UnbiasedConvolutionUint8, SimpleConvolution2d3x5Uint8Test, false, armnn::DataLayout::NCHW)
narpra015f703182018-10-26 16:24:58 +0100[diff] [blame]31ARMNN_AUTO_TEST_CASE(UnbiasedConvolution2dNhwc, SimpleConvolution2d3x5Test, false, armnn::DataLayout::NHWC)
32ARMNN_AUTO_TEST_CASE(UnbiasedConvolutionUint8Nhwc, SimpleConvolution2d3x5Uint8Test, false, armnn::DataLayout::NHWC)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]33
34ARMNN_AUTO_TEST_CASE(SimpleConvolution1d, Convolution1dTest, true)
35ARMNN_AUTO_TEST_CASE(SimpleConvolution1dUint8, Convolution1dUint8Test, true)
36
narpra015f703182018-10-26 16:24:58 +0100[diff] [blame]37ARMNN_AUTO_TEST_CASE(SimpleConvolution2d3x3, SimpleConvolution2d3x3Test, true, armnn::DataLayout::NCHW)
38ARMNN_AUTO_TEST_CASE(SimpleConvolution2d3x3Uint8, SimpleConvolution2d3x3Uint8Test, true, armnn::DataLayout::NCHW)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]39
narpra015f703182018-10-26 16:24:58 +0100[diff] [blame]40ARMNN_AUTO_TEST_CASE(SimpleConvolution2d3x3Nhwc, SimpleConvolution2d3x3Test, true, armnn::DataLayout::NHWC)
41ARMNN_AUTO_TEST_CASE(SimpleConvolution2d3x3Uint8Nhwc, SimpleConvolution2d3x3Uint8Test, true, armnn::DataLayout::NHWC)
42
43ARMNN_AUTO_TEST_CASE(UnbiasedConvolution2dSquare, SimpleConvolution2d3x3Test, false, armnn::DataLayout::NCHW)
44ARMNN_AUTO_TEST_CASE(UnbiasedConvolution2dSquareNhwc, SimpleConvolution2d3x3Test, false, armnn::DataLayout::NHWC)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]45
Mike Kelly7332ed82018-12-20 17:03:06 +0000[diff] [blame]46ARMNN_AUTO_TEST_CASE(UnbiasedConvolution2dSquareStride2x2Nhwc,
47 SimpleConvolution2d3x3Stride2x2Test,
48 false,
49 armnn::DataLayout::NHWC)
50
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]51ARMNN_AUTO_TEST_CASE(SimpleConvolution2dAsymmetricPaddingLargerThanHalfKernelSize,
narpra015f703182018-10-26 16:24:58 +0100[diff] [blame]52 Convolution2dAsymmetricPaddingLargerThanHalfKernelSizeTest,
53 armnn::DataLayout::NCHW)
54ARMNN_AUTO_TEST_CASE(SimpleConvolution2dAsymmetricPadding, Convolution2dAsymmetricPaddingTest, armnn::DataLayout::NCHW)
55
56ARMNN_AUTO_TEST_CASE(SimpleConvolution2dAsymmetricPaddingLargerThanHalfKernelSizeNhwc,
57 Convolution2dAsymmetricPaddingLargerThanHalfKernelSizeTest,
58 armnn::DataLayout::NHWC)
59ARMNN_AUTO_TEST_CASE(SimpleConvolution2dAsymmetricPaddingNhwc,
60 Convolution2dAsymmetricPaddingTest,
61 armnn::DataLayout::NHWC)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]62
Nikhil Raje4dfd6e2018-10-18 10:11:04 +0100[diff] [blame]63ARMNN_AUTO_TEST_CASE(SimpleConvolution2dSquareNhwc, SimpleConvolution2d3x3NhwcTest, false)
64
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]65// Depthwise Convolution
jimfly01d84216a2018-10-26 12:56:21 +0100[diff] [blame]66ARMNN_AUTO_TEST_CASE(DepthwiseConvolution2d, DepthwiseConvolution2dTest, true, armnn::DataLayout::NCHW)
67ARMNN_AUTO_TEST_CASE(DepthwiseConvolution2dUint8, DepthwiseConvolution2dUint8Test, true, armnn::DataLayout::NCHW)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]68
jimfly01d84216a2018-10-26 12:56:21 +0100[diff] [blame]69ARMNN_AUTO_TEST_CASE(UnbiasedDepthwiseConvolution2d, DepthwiseConvolution2dTest, false, armnn::DataLayout::NCHW)
70ARMNN_AUTO_TEST_CASE(UnbiasedDepthwiseConvolution2dUint8,
71 DepthwiseConvolution2dUint8Test,
72 false,
73 armnn::DataLayout::NCHW)
74
75// NHWC Depthwise Convolution
76ARMNN_AUTO_TEST_CASE(DepthwiseConvolution2dNhwc, DepthwiseConvolution2dTest, true, armnn::DataLayout::NHWC)
77ARMNN_AUTO_TEST_CASE(DepthwiseConvolution2dUint8Nhwc, DepthwiseConvolution2dUint8Test, true, armnn::DataLayout::NHWC)
78
79ARMNN_AUTO_TEST_CASE(UnbiasedDepthwiseConvolution2dNhwc, DepthwiseConvolution2dTest, false, armnn::DataLayout::NHWC)
80ARMNN_AUTO_TEST_CASE(UnbiasedDepthwiseConvolution2dUint8Nhwc,
81 DepthwiseConvolution2dUint8Test,
82 false,
83 armnn::DataLayout::NHWC)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]84
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]85
jimfly01b9c89632018-10-26 16:50:13 +0100[diff] [blame]86ARMNN_AUTO_TEST_CASE(DepthwiseConvolution2dDepthMul1,
87 DepthwiseConvolution2dDepthMul1Test, true, armnn::DataLayout::NCHW)
88ARMNN_AUTO_TEST_CASE(DepthwiseConvolution2dDepthMul1Uint8,
89 DepthwiseConvolution2dDepthMul1Uint8Test, true, armnn::DataLayout::NCHW)
90
91ARMNN_AUTO_TEST_CASE(UnbiasedDepthwiseConvolution2dDepthMul1,
92 DepthwiseConvolution2dDepthMul1Test, false, armnn::DataLayout::NCHW)
93ARMNN_AUTO_TEST_CASE(UnbiasedDepthwiseConvolution2dDepthMul1Uint8,
94 DepthwiseConvolution2dDepthMul1Uint8Test, false, armnn::DataLayout::NCHW)
95
96ARMNN_AUTO_TEST_CASE(DepthwiseConvolution2dDepthMul1Nhwc,
97 DepthwiseConvolution2dDepthMul1Test, true, armnn::DataLayout::NHWC)
98ARMNN_AUTO_TEST_CASE(DepthwiseConvolution2dDepthMul1Uint8Nhwc,
99 DepthwiseConvolution2dDepthMul1Uint8Test, true, armnn::DataLayout::NHWC)
100
101ARMNN_AUTO_TEST_CASE(UnbiasedDepthwiseConvolution2dDepthMul1Nhwc,
102 DepthwiseConvolution2dDepthMul1Test, false, armnn::DataLayout::NHWC)
103ARMNN_AUTO_TEST_CASE(UnbiasedDepthwiseConvolution2dDepthMul1Uint8Nhwc,
104 DepthwiseConvolution2dDepthMul1Uint8Test, false, armnn::DataLayout::NHWC)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]105
jimfly01382a91d2018-10-26 15:55:50 +0100[diff] [blame]106ARMNN_AUTO_TEST_CASE(DepthwiseConvolution2dAsymmetric,
107 DepthwiseConvolution2dAsymmetricTest, true, armnn::DataLayout::NCHW)
108ARMNN_AUTO_TEST_CASE(UnbiasedDepthwiseConvolution2dAsymmetric,
109 DepthwiseConvolution2dAsymmetricTest, false, armnn::DataLayout::NCHW)
110ARMNN_AUTO_TEST_CASE(DepthwiseConvolution2dAsymmetricNhwc,
111 DepthwiseConvolution2dAsymmetricTest, true, armnn::DataLayout::NHWC)
112ARMNN_AUTO_TEST_CASE(UnbiasedDepthwiseConvolution2dAsymmetricNhwc,
113 DepthwiseConvolution2dAsymmetricTest, false, armnn::DataLayout::NHWC)
surmeh013537c2c2018-05-18 16:31:43 +0100[diff] [blame]114
jimfly01b9c89632018-10-26 16:50:13 +0100[diff] [blame]115
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]116// Pooling
117ARMNN_AUTO_TEST_CASE(SimpleMaxPooling2dSize2x2Stride2x2, SimpleMaxPooling2dSize2x2Stride2x2Test, false)
118ARMNN_AUTO_TEST_CASE(SimpleMaxPooling2dSize2x2Stride2x2Uint8, SimpleMaxPooling2dSize2x2Stride2x2Uint8Test, false)
119
120ARMNN_AUTO_TEST_CASE(SimpleMaxPooling2dSize3x3Stride2x4, SimpleMaxPooling2dSize3x3Stride2x4Test, false)
121ARMNN_AUTO_TEST_CASE(SimpleMaxPooling2dSize3x3Stride2x4Uint8, SimpleMaxPooling2dSize3x3Stride2x4Uint8Test, false)
122
123ARMNN_AUTO_TEST_CASE(IgnorePaddingSimpleMaxPooling2d, IgnorePaddingSimpleMaxPooling2dTest)
124ARMNN_AUTO_TEST_CASE(IgnorePaddingSimpleMaxPooling2dUint8, IgnorePaddingSimpleMaxPooling2dUint8Test)
125ARMNN_AUTO_TEST_CASE(IgnorePaddingMaxPooling2dSize3, IgnorePaddingMaxPooling2dSize3Test)
126ARMNN_AUTO_TEST_CASE(IgnorePaddingMaxPooling2dSize3Uint8, IgnorePaddingMaxPooling2dSize3Uint8Test)
127
128ARMNN_AUTO_TEST_CASE(IgnorePaddingSimpleAveragePooling2d, IgnorePaddingSimpleAveragePooling2dTest)
129ARMNN_AUTO_TEST_CASE(IgnorePaddingSimpleAveragePooling2dUint8, IgnorePaddingSimpleAveragePooling2dUint8Test)
130ARMNN_AUTO_TEST_CASE(IgnorePaddingSimpleAveragePooling2dNoPadding, IgnorePaddingSimpleAveragePooling2dNoPaddingTest)
131ARMNN_AUTO_TEST_CASE(IgnorePaddingSimpleAveragePooling2dNoPaddingUint8,
132 IgnorePaddingSimpleAveragePooling2dNoPaddingUint8Test)
133ARMNN_AUTO_TEST_CASE(IgnorePaddingAveragePooling2dSize3, IgnorePaddingAveragePooling2dSize3Test)
134ARMNN_AUTO_TEST_CASE(IgnorePaddingAveragePooling2dSize3Uint8, IgnorePaddingAveragePooling2dSize3Uint8Test)
135
136ARMNN_AUTO_TEST_CASE(IgnorePaddingSimpleL2Pooling2d, IgnorePaddingSimpleL2Pooling2dTest)
137ARMNN_AUTO_TEST_CASE(IgnorePaddingSimpleL2Pooling2dUint8, IgnorePaddingSimpleL2Pooling2dUint8Test)
138ARMNN_AUTO_TEST_CASE(IgnorePaddingL2Pooling2dSize3, IgnorePaddingL2Pooling2dSize3Test)
139ARMNN_AUTO_TEST_CASE(IgnorePaddingL2Pooling2dSize3Uint8, IgnorePaddingL2Pooling2dSize3Uint8Test)
140
James Conroy45a9b772018-10-31 11:47:53 +0000[diff] [blame]141ARMNN_AUTO_TEST_CASE(SimpleMaxPooling2d, SimpleMaxPooling2dTest, armnn::DataLayout::NCHW)
142ARMNN_AUTO_TEST_CASE(SimpleMaxPooling2dNhwc, SimpleMaxPooling2dTest, armnn::DataLayout::NHWC)
143ARMNN_AUTO_TEST_CASE(SimpleMaxPooling2dUint8, SimpleMaxPooling2dUint8Test, armnn::DataLayout::NCHW)
144ARMNN_AUTO_TEST_CASE(SimpleMaxPooling2dUint8Nhwc, SimpleMaxPooling2dUint8Test, armnn::DataLayout::NHWC)
145
146ARMNN_AUTO_TEST_CASE(SimpleAveragePooling2d, SimpleAveragePooling2dTest, armnn::DataLayout::NCHW)
147ARMNN_AUTO_TEST_CASE(SimpleAveragePooling2dNhwc, SimpleAveragePooling2dTest, armnn::DataLayout::NHWC)
148ARMNN_AUTO_TEST_CASE(SimpleAveragePooling2dUint8, SimpleAveragePooling2dUint8Test, armnn::DataLayout::NCHW)
149ARMNN_AUTO_TEST_CASE(SimpleAveragePooling2dUint8Nhwc, SimpleAveragePooling2dUint8Test, armnn::DataLayout::NHWC)
150
surmeh01bceff2f2018-03-29 16:29:27 +0100[diff] [blame]151ARMNN_AUTO_TEST_CASE(IgnorePaddingAveragePooling2dSize3x2Stride2x2,
152 IgnorePaddingAveragePooling2dSize3x2Stride2x2Test, false)
153ARMNN_AUTO_TEST_CASE(IgnorePaddingAveragePooling2dSize3x2Stride2x2NoPadding,
154 IgnorePaddingAveragePooling2dSize3x2Stride2x2Test, true)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]155
156ARMNN_AUTO_TEST_CASE(LargeTensorsAveragePooling2d, LargeTensorsAveragePooling2dTest)
157ARMNN_AUTO_TEST_CASE(LargeTensorsAveragePooling2dUint8, LargeTensorsAveragePooling2dUint8Test)
158
James Conroy45a9b772018-10-31 11:47:53 +0000[diff] [blame]159ARMNN_AUTO_TEST_CASE(SimpleL2Pooling2d, SimpleL2Pooling2dTest, armnn::DataLayout::NCHW)
160ARMNN_AUTO_TEST_CASE(SimpleL2Pooling2dNhwc, SimpleL2Pooling2dTest, armnn::DataLayout::NHWC)
161ARMNN_AUTO_TEST_CASE(SimpleL2Pooling2dUint8, SimpleL2Pooling2dUint8Test, armnn::DataLayout::NCHW)
162ARMNN_AUTO_TEST_CASE(SimpleL2Pooling2dNhwcUint8, SimpleL2Pooling2dUint8Test, armnn::DataLayout::NHWC)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]163
164ARMNN_AUTO_TEST_CASE(L2Pooling2dSize7, L2Pooling2dSize7Test)
165ARMNN_AUTO_TEST_CASE(L2Pooling2dSize7Uint8, L2Pooling2dSize7Uint8Test)
166
167ARMNN_AUTO_TEST_CASE(AsymmNonSquarePooling2d, AsymmetricNonSquarePooling2dTest)
168ARMNN_AUTO_TEST_CASE(AsymmNonSquarePooling2dUint8, AsymmetricNonSquarePooling2dUint8Test)
169
Teresa Charlin18515e22019-04-24 10:17:46 +0100[diff] [blame^]170// Linear Activation
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]171ARMNN_AUTO_TEST_CASE(ConstantLinearActivation, ConstantLinearActivationTest)
172ARMNN_AUTO_TEST_CASE(ConstantLinearActivationUint8, ConstantLinearActivationUint8Test)
Teresa Charlin18515e22019-04-24 10:17:46 +0100[diff] [blame^]173ARMNN_AUTO_TEST_CASE(ConstantLinearActivationInt16, ConstantLinearActivationInt16Test)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]174
Matteo Martincigh8e6f92d2018-10-18 08:45:39 +0100[diff] [blame]175// Normalization
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]176ARMNN_AUTO_TEST_CASE(SimpleNormalizationAcross, SimpleNormalizationAcrossTest)
177ARMNN_AUTO_TEST_CASE(SimpleNormalizationWithin, SimpleNormalizationWithinTest)
Matteo Martincigh8e6f92d2018-10-18 08:45:39 +0100[diff] [blame]178ARMNN_AUTO_TEST_CASE(SimpleNormalizationAcrossNhwc, SimpleNormalizationAcrossNhwcTest)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]179
Narumol Prangnawarat65d30962019-03-14 11:55:03 +0000[diff] [blame]180// Softmax
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]181ARMNN_AUTO_TEST_CASE(SimpleSoftmaxBeta1, SimpleSoftmaxTest, 1.0f)
182ARMNN_AUTO_TEST_CASE(SimpleSoftmaxBeta2, SimpleSoftmaxTest, 2.0f)
183ARMNN_AUTO_TEST_CASE(SimpleSoftmaxBeta1Uint8, SimpleSoftmaxUint8Test, 1.0f)
184ARMNN_AUTO_TEST_CASE(SimpleSoftmaxBeta2Uint8, SimpleSoftmaxUint8Test, 2.0f)
185
Narumol Prangnawarat65d30962019-03-14 11:55:03 +0000[diff] [blame]186ARMNN_AUTO_TEST_CASE(Simple3dSoftmax, Simple3dSoftmaxTest, 1.0f)
187ARMNN_AUTO_TEST_CASE(Simple3dSoftmaxUint8, Simple3dSoftmaxUint8Test, 1.0f)
188
189ARMNN_AUTO_TEST_CASE(Simple4dSoftmax, Simple4dSoftmaxTest, 1.0f)
190ARMNN_AUTO_TEST_CASE(Simple4dSoftmaxUint8, Simple4dSoftmaxUint8Test, 1.0f)
191
Teresa Charlin18515e22019-04-24 10:17:46 +0100[diff] [blame^]192// Sigmoid Activation
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]193ARMNN_AUTO_TEST_CASE(SimpleSigmoid, SimpleSigmoidTest)
194ARMNN_AUTO_TEST_CASE(SimpleSigmoidUint8, SimpleSigmoidUint8Test)
Teresa Charlin18515e22019-04-24 10:17:46 +0100[diff] [blame^]195ARMNN_AUTO_TEST_CASE(SimpleSigmoidInt16, SimpleSigmoidInt16Test)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]196
Teresa Charlin18515e22019-04-24 10:17:46 +0100[diff] [blame^]197// BoundedReLU Activation
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]198ARMNN_AUTO_TEST_CASE(ReLu1, BoundedReLuUpperAndLowerBoundTest)
199ARMNN_AUTO_TEST_CASE(ReLu6, BoundedReLuUpperBoundOnlyTest)
200ARMNN_AUTO_TEST_CASE(ReLu1Uint8, BoundedReLuUint8UpperAndLowerBoundTest)
201ARMNN_AUTO_TEST_CASE(ReLu6Uint8, BoundedReLuUint8UpperBoundOnlyTest)
Teresa Charlin18515e22019-04-24 10:17:46 +0100[diff] [blame^]202ARMNN_AUTO_TEST_CASE(BoundedReLuInt16, BoundedReLuInt16Test)
203
204// ReLU Activation
205ARMNN_AUTO_TEST_CASE(ReLuInt16, ReLuInt16Test)
206
207// SoftReLU Activation
208ARMNN_AUTO_TEST_CASE(SoftReLuInt16, SoftReLuInt16Test)
209
210// LeakyReLU Activation
211ARMNN_AUTO_TEST_CASE(LeakyReLuInt16, LeakyReLuInt16Test)
212
213// Abs Activation
214ARMNN_AUTO_TEST_CASE(AbsInt16, AbsInt16Test)
215
216// Sqrt Activation
217ARMNN_AUTO_TEST_CASE(SqrtInt16, SqrtInt16Test)
218
219// Square Activation
220ARMNN_AUTO_TEST_CASE(SquareInt16, SquareInt16Test)
221
222// Tanh Activation
223ARMNN_AUTO_TEST_CASE(TanhInt16, TanhInt16Test)
224
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]225
226// Fully Conected
227ARMNN_AUTO_TEST_CASE(SimpleFullyConnected, FullyConnectedFloat32Test, false, false)
228ARMNN_AUTO_TEST_CASE(FullyConnectedUint8, FullyConnectedUint8Test, false)
229ARMNN_AUTO_TEST_CASE(SimpleFullyConnectedWithBias, FullyConnectedFloat32Test, true, false)
230ARMNN_AUTO_TEST_CASE(FullyConnectedBiasedUint8, FullyConnectedUint8Test, true)
231ARMNN_AUTO_TEST_CASE(SimpleFullyConnectedWithTranspose, FullyConnectedFloat32Test, false, true)
232
233ARMNN_AUTO_TEST_CASE(FullyConnectedLarge, FullyConnectedLargeTest, false)
234ARMNN_AUTO_TEST_CASE(FullyConnectedLargeTransposed, FullyConnectedLargeTest, true)
235
236// Splitter
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]237ARMNN_AUTO_TEST_CASE(SimpleSplitter, SplitterTest)
238ARMNN_AUTO_TEST_CASE(SimpleSplitterUint8, SplitterUint8Test)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]239
240ARMNN_AUTO_TEST_CASE(CopyViaSplitter, CopyViaSplitterTest)
241ARMNN_AUTO_TEST_CASE(CopyViaSplitterUint8, CopyViaSplitterUint8Test)
242
243// Merger
244ARMNN_AUTO_TEST_CASE(SimpleMerger, MergerTest)
245ARMNN_AUTO_TEST_CASE(MergerUint8, MergerUint8Test)
Ferran Balaguerb2845652019-02-27 09:42:06 +0000[diff] [blame]246ARMNN_AUTO_TEST_CASE(MergerUint8DifferentQParams, MergerUint8DifferentQParamsTest)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]247
248// Add
249ARMNN_AUTO_TEST_CASE(SimpleAdd, AdditionTest)
250ARMNN_AUTO_TEST_CASE(AddBroadcast1Element, AdditionBroadcast1ElementTest)
251ARMNN_AUTO_TEST_CASE(AddBroadcast, AdditionBroadcastTest)
252
253ARMNN_AUTO_TEST_CASE(AdditionUint8, AdditionUint8Test)
254ARMNN_AUTO_TEST_CASE(AddBroadcastUint8, AdditionBroadcastUint8Test)
255ARMNN_AUTO_TEST_CASE(AddBroadcast1ElementUint8, AdditionBroadcast1ElementUint8Test)
256
Sadik Armagan2999a022019-04-09 14:20:12 +0100[diff] [blame]257ARMNN_AUTO_TEST_CASE(AdditionInt16, AdditionInt16Test)
258ARMNN_AUTO_TEST_CASE(AddBroadcastInt16, AdditionBroadcastInt16Test)
259ARMNN_AUTO_TEST_CASE(AddBroadcast1ElementInt16, AdditionBroadcast1ElementInt16Test)
260
David Beckf195f032018-09-06 16:46:34 +0100[diff] [blame]261// Sub
262ARMNN_AUTO_TEST_CASE(SimpleSub, SubtractionTest)
263ARMNN_AUTO_TEST_CASE(SubBroadcast1Element, SubtractionBroadcast1ElementTest)
264ARMNN_AUTO_TEST_CASE(SubBroadcast, SubtractionBroadcastTest)
265
David Beck4a8692c2018-09-07 16:19:24 +0100[diff] [blame]266ARMNN_AUTO_TEST_CASE(SubtractionUint8, SubtractionUint8Test)
David Beckf195f032018-09-06 16:46:34 +0100[diff] [blame]267ARMNN_AUTO_TEST_CASE(SubBroadcastUint8, SubtractionBroadcastUint8Test)
268ARMNN_AUTO_TEST_CASE(SubBroadcast1ElementUint8, SubtractionBroadcast1ElementUint8Test)
269
Sadik Armagan2999a022019-04-09 14:20:12 +0100[diff] [blame]270ARMNN_AUTO_TEST_CASE(SubtractionInt16, SubtractionInt16Test)
271ARMNN_AUTO_TEST_CASE(SubBroadcastInt16, SubtractionBroadcastInt16Test)
272ARMNN_AUTO_TEST_CASE(SubBroadcast1ElementInt16, SubtractionBroadcast1ElementInt16Test)
273
Francis Murtaghe7a86a42018-08-29 12:42:10 +0100[diff] [blame]274// Div
275ARMNN_AUTO_TEST_CASE(SimpleDivision, DivisionTest)
Francis Murtagh8c5e3dc2018-08-30 17:18:37 +0100[diff] [blame]276ARMNN_AUTO_TEST_CASE(DivisionByZero, DivisionByZeroTest)
Francis Murtaghe7a86a42018-08-29 12:42:10 +0100[diff] [blame]277ARMNN_AUTO_TEST_CASE(DivisionBroadcast1Element, DivisionBroadcast1ElementTest)
278ARMNN_AUTO_TEST_CASE(DivisionBroadcast1DVector, DivisionBroadcast1DVectorTest)
David Beck5cd01f32018-09-12 16:00:08 +0100[diff] [blame]279// NOTE: division by zero for quantized div needs more attention
280// see IVGCVSW-1849
281ARMNN_AUTO_TEST_CASE(DivisionUint8, DivisionUint8Test)
282ARMNN_AUTO_TEST_CASE(DivisionUint8Broadcast1Element, DivisionBroadcast1ElementUint8Test)
283ARMNN_AUTO_TEST_CASE(DivisionUint8Broadcast1DVector, DivisionBroadcast1DVectorUint8Test)
Francis Murtaghe7a86a42018-08-29 12:42:10 +0100[diff] [blame]284
Sadik Armagan2999a022019-04-09 14:20:12 +0100[diff] [blame]285ARMNN_AUTO_TEST_CASE(DivisionInt16, DivisionInt16Test)
286ARMNN_AUTO_TEST_CASE(DivisionInt16Broadcast1Element, DivisionBroadcast1ElementInt16Test)
287ARMNN_AUTO_TEST_CASE(DivisionInt16Broadcast1DVector, DivisionBroadcast1DVectorInt16Test)
288
FrancisMurtagh30cdfca2018-12-18 12:57:35 +0000[diff] [blame]289// Equal
290ARMNN_AUTO_TEST_CASE(SimpleEqual, EqualSimpleTest)
291ARMNN_AUTO_TEST_CASE(EqualBroadcast1Element, EqualBroadcast1ElementTest)
292ARMNN_AUTO_TEST_CASE(EqualBroadcast1DVector, EqualBroadcast1DVectorTest)
293ARMNN_AUTO_TEST_CASE(EqualUint8, EqualUint8Test)
294ARMNN_AUTO_TEST_CASE(EqualBroadcast1ElementUint8, EqualBroadcast1ElementUint8Test)
295ARMNN_AUTO_TEST_CASE(EqualBroadcast1DVectorUint8, EqualBroadcast1DVectorUint8Test)
296
FrancisMurtagh878f0232018-12-19 10:56:15 +0000[diff] [blame]297// Greater
298ARMNN_AUTO_TEST_CASE(SimpleGreater, GreaterSimpleTest)
299ARMNN_AUTO_TEST_CASE(GreaterBroadcast1Element, GreaterBroadcast1ElementTest)
300ARMNN_AUTO_TEST_CASE(GreaterBroadcast1DVector, GreaterBroadcast1DVectorTest)
301ARMNN_AUTO_TEST_CASE(GreaterUint8, GreaterUint8Test)
302ARMNN_AUTO_TEST_CASE(GreaterBroadcast1ElementUint8, GreaterBroadcast1ElementUint8Test)
303ARMNN_AUTO_TEST_CASE(GreaterBroadcast1DVectorUint8, GreaterBroadcast1DVectorUint8Test)
304
Éanna Ó Catháin20e58802018-12-04 10:29:06 +0000[diff] [blame]305// Max
Éanna Ó Catháinde705582018-12-03 13:04:22 +0000[diff] [blame]306ARMNN_AUTO_TEST_CASE(SimpleMaximum, MaximumSimpleTest)
307ARMNN_AUTO_TEST_CASE(MaximumBroadcast1Element, MaximumBroadcast1ElementTest)
308ARMNN_AUTO_TEST_CASE(MaximumBroadcast1DVector, MaximumBroadcast1DVectorTest)
309ARMNN_AUTO_TEST_CASE(MaximumUint8, MaximumUint8Test)
310ARMNN_AUTO_TEST_CASE(MaximumBroadcast1ElementUint8, MaximumBroadcast1ElementUint8Test)
311ARMNN_AUTO_TEST_CASE(MaximumBroadcast1DVectorUint8, MaximumBroadcast1DVectorUint8Test)
Sadik Armagan2999a022019-04-09 14:20:12 +0100[diff] [blame]312ARMNN_AUTO_TEST_CASE(MaximumInt16, MaximumInt16Test)
313ARMNN_AUTO_TEST_CASE(MaximumBroadcast1ElementInt16, MaximumBroadcast1ElementInt16Test)
314ARMNN_AUTO_TEST_CASE(MaximumBroadcast1DVectorInt16, MaximumBroadcast1DVectorInt16Test)
Éanna Ó Catháinde705582018-12-03 13:04:22 +0000[diff] [blame]315
Éanna Ó Catháin20e58802018-12-04 10:29:06 +0000[diff] [blame]316// Min
317ARMNN_AUTO_TEST_CASE(SimpleMinimum1, MinimumBroadcast1ElementTest1)
318ARMNN_AUTO_TEST_CASE(SimpleMinimum2, MinimumBroadcast1ElementTest2)
319ARMNN_AUTO_TEST_CASE(Minimum1DVectorUint8, MinimumBroadcast1DVectorUint8Test)
Sadik Armagan2999a022019-04-09 14:20:12 +0100[diff] [blame]320ARMNN_AUTO_TEST_CASE(MinimumInt16, MinimumInt16Test)
321ARMNN_AUTO_TEST_CASE(MinimumBroadcast1ElementInt16, MinimumBroadcast1ElementInt16Test)
322ARMNN_AUTO_TEST_CASE(MinimumBroadcast1DVectorInt16, MinimumBroadcast1DVectorInt16Test)
Éanna Ó Catháin20e58802018-12-04 10:29:06 +0000[diff] [blame]323
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]324// Mul
325ARMNN_AUTO_TEST_CASE(SimpleMultiplication, MultiplicationTest)
surmeh01bceff2f2018-03-29 16:29:27 +0100[diff] [blame]326ARMNN_AUTO_TEST_CASE(MultiplicationBroadcast1Element, MultiplicationBroadcast1ElementTest)
327ARMNN_AUTO_TEST_CASE(MultiplicationBroadcast1DVector, MultiplicationBroadcast1DVectorTest)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]328ARMNN_AUTO_TEST_CASE(MultiplicationUint8, MultiplicationUint8Test)
surmeh01bceff2f2018-03-29 16:29:27 +0100[diff] [blame]329ARMNN_AUTO_TEST_CASE(MultiplicationBroadcast1ElementUint8, MultiplicationBroadcast1ElementUint8Test)
330ARMNN_AUTO_TEST_CASE(MultiplicationBroadcast1DVectorUint8, MultiplicationBroadcast1DVectorUint8Test)
Sadik Armagan2999a022019-04-09 14:20:12 +0100[diff] [blame]331ARMNN_AUTO_TEST_CASE(MultiplicationInt16, MultiplicationInt16Test)
332ARMNN_AUTO_TEST_CASE(MultiplicationBroadcast1ElementInt16, MultiplicationBroadcast1ElementInt16Test)
333ARMNN_AUTO_TEST_CASE(MultiplicationBroadcast1DVectorInt16, MultiplicationBroadcast1DVectorInt16Test)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]334
335// Batch Norm
336ARMNN_AUTO_TEST_CASE(BatchNorm, BatchNormTest)
Matteo Martincigh8eb675e2018-10-17 14:43:29 +0100[diff] [blame]337ARMNN_AUTO_TEST_CASE(BatchNormNhwc, BatchNormNhwcTest)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]338ARMNN_AUTO_TEST_CASE(BatchNormUint8, BatchNormUint8Test)
Matteo Martincigh8eb675e2018-10-17 14:43:29 +0100[diff] [blame]339ARMNN_AUTO_TEST_CASE(BatchNormUint8Nhwc, BatchNormUint8NhwcTest)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]340
James Conroy59540822018-10-11 12:39:05 +0100[diff] [blame]341// Resize Bilinear - NCHW
James Conroy6b965822018-11-01 11:33:09 +0000[diff] [blame]342ARMNN_AUTO_TEST_CASE(SimpleResizeBilinear, SimpleResizeBilinearTest, armnn::DataLayout::NCHW)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]343ARMNN_AUTO_TEST_CASE(SimpleResizeBilinearUint8, SimpleResizeBilinearUint8Test)
James Conroy6b965822018-11-01 11:33:09 +0000[diff] [blame]344ARMNN_AUTO_TEST_CASE(ResizeBilinearNop, ResizeBilinearNopTest, armnn::DataLayout::NCHW)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]345ARMNN_AUTO_TEST_CASE(ResizeBilinearNopUint8, ResizeBilinearNopUint8Test)
James Conroy6b965822018-11-01 11:33:09 +0000[diff] [blame]346ARMNN_AUTO_TEST_CASE(ResizeBilinearSqMin, ResizeBilinearSqMinTest, armnn::DataLayout::NCHW)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]347ARMNN_AUTO_TEST_CASE(ResizeBilinearSqMinUint8, ResizeBilinearSqMinUint8Test)
James Conroy6b965822018-11-01 11:33:09 +0000[diff] [blame]348ARMNN_AUTO_TEST_CASE(ResizeBilinearMin, ResizeBilinearMinTest, armnn::DataLayout::NCHW)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]349ARMNN_AUTO_TEST_CASE(ResizeBilinearMinUint8, ResizeBilinearMinUint8Test)
James Conroy6b965822018-11-01 11:33:09 +0000[diff] [blame]350ARMNN_AUTO_TEST_CASE(ResizeBilinearMag, ResizeBilinearMagTest, armnn::DataLayout::NCHW)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]351ARMNN_AUTO_TEST_CASE(ResizeBilinearMagUint8, ResizeBilinearMagUint8Test)
352
James Conroy59540822018-10-11 12:39:05 +0100[diff] [blame]353// Resize Bilinear - NHWC
James Conroy6b965822018-11-01 11:33:09 +0000[diff] [blame]354ARMNN_AUTO_TEST_CASE(ResizeBilinearNopNhwc, ResizeBilinearNopTest, armnn::DataLayout::NHWC)
355ARMNN_AUTO_TEST_CASE(SimpleResizeBilinearNhwc, SimpleResizeBilinearTest, armnn::DataLayout::NHWC)
356ARMNN_AUTO_TEST_CASE(ResizeBilinearSqMinNhwc, ResizeBilinearSqMinTest, armnn::DataLayout::NHWC)
357ARMNN_AUTO_TEST_CASE(ResizeBilinearMinNhwc, ResizeBilinearMinTest, armnn::DataLayout::NHWC)
358ARMNN_AUTO_TEST_CASE(ResizeBilinearMagNhwc, ResizeBilinearMagTest, armnn::DataLayout::NHWC)
James Conroy59540822018-10-11 12:39:05 +0100[diff] [blame]359
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]360// Fake Quantization
361ARMNN_AUTO_TEST_CASE(FakeQuantization, FakeQuantizationTest)
362
Matteo Martincigh539b44d2018-10-01 09:26:39 +0100[diff] [blame]363// L2 Normalization
jimfly013aab7c32018-11-12 13:32:08 +0000[diff] [blame]364ARMNN_AUTO_TEST_CASE(L2Normalization1d, L2Normalization1dTest, armnn::DataLayout::NCHW)
365ARMNN_AUTO_TEST_CASE(L2Normalization2d, L2Normalization2dTest, armnn::DataLayout::NCHW)
366ARMNN_AUTO_TEST_CASE(L2Normalization3d, L2Normalization3dTest, armnn::DataLayout::NCHW)
367ARMNN_AUTO_TEST_CASE(L2Normalization4d, L2Normalization4dTest, armnn::DataLayout::NCHW)
368
369ARMNN_AUTO_TEST_CASE(L2Normalization1dNhwc, L2Normalization1dTest, armnn::DataLayout::NHWC)
370ARMNN_AUTO_TEST_CASE(L2Normalization2dNhwc, L2Normalization2dTest, armnn::DataLayout::NHWC)
371ARMNN_AUTO_TEST_CASE(L2Normalization3dNhwc, L2Normalization3dTest, armnn::DataLayout::NHWC)
372ARMNN_AUTO_TEST_CASE(L2Normalization4dNhwc, L2Normalization4dTest, armnn::DataLayout::NHWC)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]373
Mohamed Nour Abouelseoud7420e552018-10-12 12:26:24 +0100[diff] [blame]374// Pad
Mohamed Nour Abouelseouddd6acea2018-10-18 12:26:19 +0100[diff] [blame]375ARMNN_AUTO_TEST_CASE(PadFloat322d, PadFloat322dTest)
376ARMNN_AUTO_TEST_CASE(PadFloat323d, PadFloat323dTest)
377ARMNN_AUTO_TEST_CASE(PadFloat324d, PadFloat324dTest)
378
379ARMNN_AUTO_TEST_CASE(PadUint82d, PadUint82dTest)
380ARMNN_AUTO_TEST_CASE(PadUint83d, PadUint83dTest)
381ARMNN_AUTO_TEST_CASE(PadUint84d, PadUint84dTest)
Mohamed Nour Abouelseoud7420e552018-10-12 12:26:24 +0100[diff] [blame]382
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]383// Constant
384ARMNN_AUTO_TEST_CASE(Constant, ConstantTest)
385ARMNN_AUTO_TEST_CASE(ConstantUint8, ConstantUint8Test)
386
387// Concat
388ARMNN_AUTO_TEST_CASE(Concatenation1d, Concatenation1dTest)
389ARMNN_AUTO_TEST_CASE(Concatenation1dUint8, Concatenation1dUint8Test)
390
391ARMNN_AUTO_TEST_CASE(Concatenation2dDim0, Concatenation2dDim0Test)
392ARMNN_AUTO_TEST_CASE(Concatenation2dDim0Uint8, Concatenation2dDim0Uint8Test)
393ARMNN_AUTO_TEST_CASE(Concatenation2dDim1, Concatenation2dDim1Test)
394ARMNN_AUTO_TEST_CASE(Concatenation2dDim1Uint8, Concatenation2dDim1Uint8Test)
395
396ARMNN_AUTO_TEST_CASE(Concatenation2dDim0DiffInputDims, Concatenation2dDim0DiffInputDimsTest)
397ARMNN_AUTO_TEST_CASE(Concatenation2dDim0DiffInputDimsUint8, Concatenation2dDim0DiffInputDimsUint8Test)
398ARMNN_AUTO_TEST_CASE(Concatenation2dDim1DiffInputDims, Concatenation2dDim1DiffInputDimsTest)
399ARMNN_AUTO_TEST_CASE(Concatenation2dDim1DiffInputDimsUint8, Concatenation2dDim1DiffInputDimsUint8Test)
400
401ARMNN_AUTO_TEST_CASE(Concatenation3dDim0, Concatenation3dDim0Test)
402ARMNN_AUTO_TEST_CASE(Concatenation3dDim0Uint8, Concatenation3dDim0Uint8Test)
403ARMNN_AUTO_TEST_CASE(Concatenation3dDim1, Concatenation3dDim1Test)
404ARMNN_AUTO_TEST_CASE(Concatenation3dDim1Uint8, Concatenation3dDim1Uint8Test)
narpra015cdda352018-11-19 15:30:27 +0000[diff] [blame]405ARMNN_AUTO_TEST_CASE(Concatenation3dDim2, Concatenation3dDim2Test, true)
406ARMNN_AUTO_TEST_CASE(Concatenation3dDim2Uint8, Concatenation3dDim2Uint8Test, true)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]407
408ARMNN_AUTO_TEST_CASE(Concatenation3dDim0DiffInputDims, Concatenation3dDim0DiffInputDimsTest)
409ARMNN_AUTO_TEST_CASE(Concatenation3dDim0DiffInputDimsUint8, Concatenation3dDim0DiffInputDimsUint8Test)
410ARMNN_AUTO_TEST_CASE(Concatenation3dDim1DiffInputDims, Concatenation3dDim1DiffInputDimsTest)
411ARMNN_AUTO_TEST_CASE(Concatenation3dDim1DiffInputDimsUint8, Concatenation3dDim1DiffInputDimsUint8Test)
narpra015cdda352018-11-19 15:30:27 +0000[diff] [blame]412ARMNN_AUTO_TEST_CASE(Concatenation3dDim2DiffInputDims, Concatenation3dDim2DiffInputDimsTest, true)
413ARMNN_AUTO_TEST_CASE(Concatenation3dDim2DiffInputDimsUint8, Concatenation3dDim2DiffInputDimsUint8Test, true)
414
415ARMNN_AUTO_TEST_CASE(Concatenation4dDim0, Concatenation4dDim0Test)
416ARMNN_AUTO_TEST_CASE(Concatenation4dDim1, Concatenation4dDim1Test)
417ARMNN_AUTO_TEST_CASE(Concatenation4dDim2, Concatenation4dDim2Test)
418ARMNN_AUTO_TEST_CASE(Concatenation4dDim3, Concatenation4dDim3Test, true)
419ARMNN_AUTO_TEST_CASE(Concatenation4dDim0Uint8, Concatenation4dDim0Uint8Test)
420ARMNN_AUTO_TEST_CASE(Concatenation4dDim1Uint8, Concatenation4dDim1Uint8Test)
421ARMNN_AUTO_TEST_CASE(Concatenation4dDim2Uint8, Concatenation4dDim2Uint8Test)
422ARMNN_AUTO_TEST_CASE(Concatenation4dDim3Uint8, Concatenation4dDim3Uint8Test, true)
423
424ARMNN_AUTO_TEST_CASE(Concatenation4dDiffShapeDim0, Concatenation4dDiffShapeDim0Test)
425ARMNN_AUTO_TEST_CASE(Concatenation4dDiffShapeDim1, Concatenation4dDiffShapeDim1Test)
426ARMNN_AUTO_TEST_CASE(Concatenation4dDiffShapeDim2, Concatenation4dDiffShapeDim2Test)
427ARMNN_AUTO_TEST_CASE(Concatenation4dDiffShapeDim3, Concatenation4dDiffShapeDim3Test, true)
428ARMNN_AUTO_TEST_CASE(Concatenation4dDiffShapeDim0Uint8, Concatenation4dDiffShapeDim0Uint8Test)
429ARMNN_AUTO_TEST_CASE(Concatenation4dDiffShapeDim1Uint8, Concatenation4dDiffShapeDim1Uint8Test)
430ARMNN_AUTO_TEST_CASE(Concatenation4dDiffShapeDim2Uint8, Concatenation4dDiffShapeDim2Uint8Test)
431ARMNN_AUTO_TEST_CASE(Concatenation4dDiffShapeDim3Uint8, Concatenation4dDiffShapeDim3Uint8Test, true)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]432
433// Floor
434ARMNN_AUTO_TEST_CASE(SimpleFloor, SimpleFloorTest)
435
436// Reshape
437ARMNN_AUTO_TEST_CASE(SimpleReshapeFloat32, SimpleReshapeFloat32Test)
438ARMNN_AUTO_TEST_CASE(SimpleReshapeUint8, SimpleReshapeUint8Test)
439
Mohamed Nour Abouelseouda1d3c6a2018-12-27 12:39:16 +0000[diff] [blame]440// Rsqrt
441ARMNN_AUTO_TEST_CASE(Rsqrt2d, Rsqrt2dTest)
442ARMNN_AUTO_TEST_CASE(Rsqrt3d, Rsqrt3dTest)
443ARMNN_AUTO_TEST_CASE(RsqrtZero, RsqrtZeroTest)
444ARMNN_AUTO_TEST_CASE(RsqrtNegative, RsqrtNegativeTest)
445
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]446// Permute
447ARMNN_AUTO_TEST_CASE(SimplePermuteFloat32, SimplePermuteFloat32Test)
448ARMNN_AUTO_TEST_CASE(SimplePermuteUint8, SimplePermuteUint8Test)
surmeh01bceff2f2018-03-29 16:29:27 +0100[diff] [blame]449ARMNN_AUTO_TEST_CASE(PermuteFloat32ValueSet1, PermuteFloat32ValueSet1Test)
450ARMNN_AUTO_TEST_CASE(PermuteFloat32ValueSet2, PermuteFloat32ValueSet2Test)
451ARMNN_AUTO_TEST_CASE(PermuteFloat32ValueSet3, PermuteFloat32ValueSet3Test)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]452
Matteo Martincigha65b7ae2018-11-14 12:39:55 +0000[diff] [blame]453// Lstm
454ARMNN_AUTO_TEST_CASE(LstmLayerFloat32WithCifgWithPeepholeNoProjection,
455 LstmLayerFloat32WithCifgWithPeepholeNoProjectionTest)
456ARMNN_AUTO_TEST_CASE(LstmLayerFloat32NoCifgNoPeepholeNoProjection,
457 LstmLayerFloat32NoCifgNoPeepholeNoProjectionTest)
458ARMNN_AUTO_TEST_CASE(LstmLayerFloat32NoCifgWithPeepholeWithProjection,
459 LstmLayerFloat32NoCifgWithPeepholeWithProjectionTest)
460
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]461// Convert from Float16 to Float32
462ARMNN_AUTO_TEST_CASE(SimpleConvertFp16ToFp32, SimpleConvertFp16ToFp32Test)
463// Convert from Float32 to Float16
464ARMNN_AUTO_TEST_CASE(SimpleConvertFp32ToFp16, SimpleConvertFp32ToFp16Test)
465
narpra011e4c31d2018-09-28 11:07:51 +0100[diff] [blame]466// Mean
467ARMNN_AUTO_TEST_CASE(MeanUint8Simple, MeanUint8SimpleTest)
468ARMNN_AUTO_TEST_CASE(MeanUint8SimpleAxis, MeanUint8SimpleAxisTest)
469ARMNN_AUTO_TEST_CASE(MeanUint8KeepDims, MeanUint8KeepDimsTest)
470ARMNN_AUTO_TEST_CASE(MeanUint8MultipleDims, MeanUint8MultipleDimsTest)
471ARMNN_AUTO_TEST_CASE(MeanVtsUint8, MeanVtsUint8Test)
472
473ARMNN_AUTO_TEST_CASE(MeanFloatSimple, MeanFloatSimpleTest)
474ARMNN_AUTO_TEST_CASE(MeanFloatSimpleAxis, MeanFloatSimpleAxisTest)
475ARMNN_AUTO_TEST_CASE(MeanFloatKeepDims, MeanFloatKeepDimsTest)
476ARMNN_AUTO_TEST_CASE(MeanFloatMultipleDims, MeanFloatMultipleDimsTest)
477ARMNN_AUTO_TEST_CASE(MeanVtsFloat1, MeanVtsFloat1Test)
478ARMNN_AUTO_TEST_CASE(MeanVtsFloat2, MeanVtsFloat2Test)
Matteo Martincigh28dcab62018-10-19 16:40:03 +0100[diff] [blame]479ARMNN_AUTO_TEST_CASE(MeanVtsFloat3, MeanVtsFloat3Test)
narpra011e4c31d2018-09-28 11:07:51 +0100[diff] [blame]480
Éanna Ó Catháin47c1ddb2018-10-12 14:24:13 +0100[diff] [blame]481ARMNN_AUTO_TEST_CASE(AdditionAfterMaxPool, AdditionAfterMaxPoolTest)
482
Nattapat Chaimanowong3ea76d52018-11-09 14:10:38 +0000[diff] [blame]483// Space To Batch Nd
484ARMNN_AUTO_TEST_CASE(SpaceToBatchNdSimpleFloat32, SpaceToBatchNdSimpleFloat32Test)
485ARMNN_AUTO_TEST_CASE(SpaceToBatchNdMultiChannelsFloat32, SpaceToBatchNdMultiChannelsFloat32Test)
486ARMNN_AUTO_TEST_CASE(SpaceToBatchNdMultiBlockFloat32, SpaceToBatchNdMultiBlockFloat32Test)
487ARMNN_AUTO_TEST_CASE(SpaceToBatchNdPaddingFloat32, SpaceToBatchNdPaddingFloat32Test)
488
489ARMNN_AUTO_TEST_CASE(SpaceToBatchNdSimpleUint8, SpaceToBatchNdSimpleUint8Test)
490ARMNN_AUTO_TEST_CASE(SpaceToBatchNdMultiChannelsUint8, SpaceToBatchNdMultiChannelsUint8Test)
491ARMNN_AUTO_TEST_CASE(SpaceToBatchNdMultiBlockUint8, SpaceToBatchNdMultiBlockUint8Test)
492ARMNN_AUTO_TEST_CASE(SpaceToBatchNdPaddingUint8, SpaceToBatchNdPaddingUint8Test)
493
494ARMNN_AUTO_TEST_CASE(SpaceToBatchNdSimpleNHWCFloat32, SpaceToBatchNdSimpleNHWCFloat32Test)
495ARMNN_AUTO_TEST_CASE(SpaceToBatchNdMultiChannelsNHWCFloat32, SpaceToBatchNdMultiChannelsNHWCFloat32Test)
496ARMNN_AUTO_TEST_CASE(SpaceToBatchNdMultiBlockNHWCFloat32, SpaceToBatchNdMultiBlockNHWCFloat32Test)
497ARMNN_AUTO_TEST_CASE(SpaceToBatchNdPaddingNHWCFloat32, SpaceToBatchNdPaddingNHWCFloat32Test)
498
499ARMNN_AUTO_TEST_CASE(SpaceToBatchNdSimpleNHWCUint8, SpaceToBatchNdSimpleNHWCUint8Test)
500ARMNN_AUTO_TEST_CASE(SpaceToBatchNdMultiChannelsNHWCUint8, SpaceToBatchNdMultiChannelsNHWCUint8Test)
501ARMNN_AUTO_TEST_CASE(SpaceToBatchNdMultiBlockNHWCUint8, SpaceToBatchNdMultiBlockNHWCUint8Test)
502ARMNN_AUTO_TEST_CASE(SpaceToBatchNdPaddingNHWCUint8, SpaceToBatchNdPaddingNHWCUint8Test)
503
Éanna Ó Catháin4e1e1362018-11-12 11:36:34 +0000[diff] [blame]504ARMNN_AUTO_TEST_CASE(BatchToSpaceNdNhwcFloat321, BatchToSpaceNdNhwcFloat32Test1)
505ARMNN_AUTO_TEST_CASE(BatchToSpaceNdNhwcFloat322, BatchToSpaceNdNhwcFloat32Test2)
506ARMNN_AUTO_TEST_CASE(BatchToSpaceNdNhwcFloat323, BatchToSpaceNdNhwcFloat32Test3)
Nattapat Chaimanowong3ee14222019-02-27 10:28:09 +0000[diff] [blame]507ARMNN_AUTO_TEST_CASE(BatchToSpaceNdNhwcFloat324, BatchToSpaceNdNhwcFloat32Test4)
Éanna Ó Catháin4e1e1362018-11-12 11:36:34 +0000[diff] [blame]508
509ARMNN_AUTO_TEST_CASE(BatchToSpaceNdNchwFloat321, BatchToSpaceNdNchwFloat32Test1)
Mike Kelly831faed2018-11-28 11:52:08 +0000[diff] [blame]510ARMNN_AUTO_TEST_CASE(BatchToSpaceNdNchwFloat322, BatchToSpaceNdNchwFloat32Test2)
511ARMNN_AUTO_TEST_CASE(BatchToSpaceNdNchwFloat323, BatchToSpaceNdNchwFloat32Test3)
Éanna Ó Catháin4e1e1362018-11-12 11:36:34 +0000[diff] [blame]512
Éanna Ó Catháin262553e2018-11-14 11:26:23 +0000[diff] [blame]513ARMNN_AUTO_TEST_CASE(BatchToSpaceNdNhwcUint1, BatchToSpaceNdNhwcUintTest1)
Mike Kelly831faed2018-11-28 11:52:08 +0000[diff] [blame]514ARMNN_AUTO_TEST_CASE(BatchToSpaceNdNhwcUint2, BatchToSpaceNdNhwcUintTest2)
515ARMNN_AUTO_TEST_CASE(BatchToSpaceNdNhwcUint3, BatchToSpaceNdNhwcUintTest3)
516
517ARMNN_AUTO_TEST_CASE(BatchToSpaceNdNchwUint1, BatchToSpaceNdNchwUintTest1)
518ARMNN_AUTO_TEST_CASE(BatchToSpaceNdNchwUint2, BatchToSpaceNdNchwUintTest2)
519ARMNN_AUTO_TEST_CASE(BatchToSpaceNdNchwUint3, BatchToSpaceNdNchwUintTest3)
Nattapat Chaimanowong3ee14222019-02-27 10:28:09 +0000[diff] [blame]520ARMNN_AUTO_TEST_CASE(BatchToSpaceNdNchwUint4, BatchToSpaceNdNchwUintTest4)
Éanna Ó Catháin262553e2018-11-14 11:26:23 +0000[diff] [blame]521
Nattapat Chaimanowong1216b582018-11-23 15:33:41 +0000[diff] [blame]522// Strided Slice
523ARMNN_AUTO_TEST_CASE(StridedSlice4DFloat32, StridedSlice4DFloat32Test)
524ARMNN_AUTO_TEST_CASE(StridedSlice4DReverseFloat32, StridedSlice4DReverseFloat32Test)
525ARMNN_AUTO_TEST_CASE(StridedSliceSimpleStrideFloat32, StridedSliceSimpleStrideFloat32Test)
526ARMNN_AUTO_TEST_CASE(StridedSliceSimpleRangeMaskFloat32, StridedSliceSimpleRangeMaskFloat32Test)
527ARMNN_AUTO_TEST_CASE(StridedSliceShrinkAxisMaskFloat32, StridedSliceShrinkAxisMaskFloat32Test)
528ARMNN_AUTO_TEST_CASE(StridedSlice3DFloat32, StridedSlice3DFloat32Test)
529ARMNN_AUTO_TEST_CASE(StridedSlice3DReverseFloat32, StridedSlice3DReverseFloat32Test)
530ARMNN_AUTO_TEST_CASE(StridedSlice2DFloat32, StridedSlice2DFloat32Test)
531ARMNN_AUTO_TEST_CASE(StridedSlice2DReverseFloat32, StridedSlice2DReverseFloat32Test)
532
533ARMNN_AUTO_TEST_CASE(StridedSlice4DUint8, StridedSlice4DUint8Test)
534ARMNN_AUTO_TEST_CASE(StridedSlice4DReverseUint8, StridedSlice4DReverseUint8Test)
535ARMNN_AUTO_TEST_CASE(StridedSliceSimpleStrideUint8, StridedSliceSimpleStrideUint8Test)
536ARMNN_AUTO_TEST_CASE(StridedSliceSimpleRangeMaskUint8, StridedSliceSimpleRangeMaskUint8Test)
537ARMNN_AUTO_TEST_CASE(StridedSliceShrinkAxisMaskUint8, StridedSliceShrinkAxisMaskUint8Test)
538ARMNN_AUTO_TEST_CASE(StridedSlice3DUint8, StridedSlice3DUint8Test)
539ARMNN_AUTO_TEST_CASE(StridedSlice3DReverseUint8, StridedSlice3DReverseUint8Test)
540ARMNN_AUTO_TEST_CASE(StridedSlice2DUint8, StridedSlice2DUint8Test)
541ARMNN_AUTO_TEST_CASE(StridedSlice2DReverseUint8, StridedSlice2DReverseUint8Test)
542
Nattapat Chaimanowongcfdcadf2018-12-06 11:54:33 +0000[diff] [blame]543// Debug
544ARMNN_AUTO_TEST_CASE(Debug4DFloat32, Debug4DFloat32Test)
545ARMNN_AUTO_TEST_CASE(Debug3DFloat32, Debug3DFloat32Test)
546ARMNN_AUTO_TEST_CASE(Debug2DFloat32, Debug2DFloat32Test)
547ARMNN_AUTO_TEST_CASE(Debug1DFloat32, Debug1DFloat32Test)
548
549ARMNN_AUTO_TEST_CASE(Debug4DUint8, Debug4DUint8Test)
550ARMNN_AUTO_TEST_CASE(Debug3DUint8, Debug3DUint8Test)
551ARMNN_AUTO_TEST_CASE(Debug2DUint8, Debug2DUint8Test)
552ARMNN_AUTO_TEST_CASE(Debug1DUint8, Debug1DUint8Test)
553
narpra014951d842019-01-18 16:53:53 +0000[diff] [blame]554// Gather
555ARMNN_AUTO_TEST_CASE(Gather1DParamsFloat, Gather1DParamsFloatTest)
556ARMNN_AUTO_TEST_CASE(Gather1DParamsUint8, Gather1DParamsUint8Test)
557ARMNN_AUTO_TEST_CASE(GatherMultiDimParamsFloat, GatherMultiDimParamsFloatTest)
558ARMNN_AUTO_TEST_CASE(GatherMultiDimParamsUint8, GatherMultiDimParamsUint8Test)
559ARMNN_AUTO_TEST_CASE(GatherMultiDimParamsMultiDimIndicesFloat, GatherMultiDimParamsMultiDimIndicesFloatTest)
560ARMNN_AUTO_TEST_CASE(GatherMultiDimParamsMultiDimIndicesUint8, GatherMultiDimParamsMultiDimIndicesUint8Test)
561
Narumol Prangnawarate0a4ad82019-02-04 19:05:27 +0000[diff] [blame]562// Detection PostProcess
563BOOST_AUTO_TEST_CASE(DetectionPostProcessRegularNmsFloat)
564{
565 DetectionPostProcessRegularNmsFloatTest<armnn::RefWorkloadFactory>();
566}
567BOOST_AUTO_TEST_CASE(DetectionPostProcessFastNmsFloat)
568{
569 DetectionPostProcessFastNmsFloatTest<armnn::RefWorkloadFactory>();
570}
571BOOST_AUTO_TEST_CASE(DetectionPostProcessRegularNmsUint8)
572{
573 DetectionPostProcessRegularNmsUint8Test<armnn::RefWorkloadFactory>();
574}
575BOOST_AUTO_TEST_CASE(DetectionPostProcessFastNmsUint8)
576{
577 DetectionPostProcessFastNmsUint8Test<armnn::RefWorkloadFactory>();
578}
579
Nattapat Chaimanowong8a54ac02019-03-29 15:25:04 +0000[diff] [blame]580// Dequantize
Nattapat Chaimanowongafa4e3a2019-04-02 11:41:45 +0100[diff] [blame]581ARMNN_AUTO_TEST_CASE(DequantizeSimpleUint8, DequantizeSimpleUint8Test)
582ARMNN_AUTO_TEST_CASE(DequantizeOffsetUint8, DequantizeOffsetUint8Test)
583ARMNN_AUTO_TEST_CASE(DequantizeSimpleInt16, DequantizeSimpleInt16Test)
Narumol Prangnawarate0a4ad82019-02-04 19:05:27 +0000[diff] [blame]584
Nattapat Chaimanowonga0beb3b2019-04-01 17:04:53 +0100[diff] [blame]585// Quantize
586ARMNN_AUTO_TEST_CASE(QuantizeSimpleUint8, QuantizeSimpleUint8Test)
587ARMNN_AUTO_TEST_CASE(QuantizeClampUint8, QuantizeClampUint8Test)
588ARMNN_AUTO_TEST_CASE(QuantizeClampInt16, QuantizeClampInt16Test)
589
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]590BOOST_AUTO_TEST_SUITE_END()