Gitiles
Code Review Sign In
review.mlplatform.org / ml / armnn / e05e38e782f6c12301d069845fbfe353ab58da55 / . / src / backends / reference / test / RefLayerTests.cpp
blob: cbc56d14b7789a9162db49395423392cc117528c [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
170// Activation
171ARMNN_AUTO_TEST_CASE(ConstantLinearActivation, ConstantLinearActivationTest)
172ARMNN_AUTO_TEST_CASE(ConstantLinearActivationUint8, ConstantLinearActivationUint8Test)
173
Matteo Martincigh8e6f92d2018-10-18 08:45:39 +0100[diff] [blame]174// Normalization
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]175ARMNN_AUTO_TEST_CASE(SimpleNormalizationAcross, SimpleNormalizationAcrossTest)
176ARMNN_AUTO_TEST_CASE(SimpleNormalizationWithin, SimpleNormalizationWithinTest)
Matteo Martincigh8e6f92d2018-10-18 08:45:39 +0100[diff] [blame]177ARMNN_AUTO_TEST_CASE(SimpleNormalizationAcrossNhwc, SimpleNormalizationAcrossNhwcTest)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]178
Narumol Prangnawarat65d30962019-03-14 11:55:03 +0000[diff] [blame]179// Softmax
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]180ARMNN_AUTO_TEST_CASE(SimpleSoftmaxBeta1, SimpleSoftmaxTest, 1.0f)
181ARMNN_AUTO_TEST_CASE(SimpleSoftmaxBeta2, SimpleSoftmaxTest, 2.0f)
182ARMNN_AUTO_TEST_CASE(SimpleSoftmaxBeta1Uint8, SimpleSoftmaxUint8Test, 1.0f)
183ARMNN_AUTO_TEST_CASE(SimpleSoftmaxBeta2Uint8, SimpleSoftmaxUint8Test, 2.0f)
184
Narumol Prangnawarat65d30962019-03-14 11:55:03 +0000[diff] [blame]185ARMNN_AUTO_TEST_CASE(Simple3dSoftmax, Simple3dSoftmaxTest, 1.0f)
186ARMNN_AUTO_TEST_CASE(Simple3dSoftmaxUint8, Simple3dSoftmaxUint8Test, 1.0f)
187
188ARMNN_AUTO_TEST_CASE(Simple4dSoftmax, Simple4dSoftmaxTest, 1.0f)
189ARMNN_AUTO_TEST_CASE(Simple4dSoftmaxUint8, Simple4dSoftmaxUint8Test, 1.0f)
190
191// Sigmoid
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]192ARMNN_AUTO_TEST_CASE(SimpleSigmoid, SimpleSigmoidTest)
193ARMNN_AUTO_TEST_CASE(SimpleSigmoidUint8, SimpleSigmoidUint8Test)
194
195ARMNN_AUTO_TEST_CASE(ReLu1, BoundedReLuUpperAndLowerBoundTest)
196ARMNN_AUTO_TEST_CASE(ReLu6, BoundedReLuUpperBoundOnlyTest)
197ARMNN_AUTO_TEST_CASE(ReLu1Uint8, BoundedReLuUint8UpperAndLowerBoundTest)
198ARMNN_AUTO_TEST_CASE(ReLu6Uint8, BoundedReLuUint8UpperBoundOnlyTest)
199
200// Fully Conected
201ARMNN_AUTO_TEST_CASE(SimpleFullyConnected, FullyConnectedFloat32Test, false, false)
202ARMNN_AUTO_TEST_CASE(FullyConnectedUint8, FullyConnectedUint8Test, false)
203ARMNN_AUTO_TEST_CASE(SimpleFullyConnectedWithBias, FullyConnectedFloat32Test, true, false)
204ARMNN_AUTO_TEST_CASE(FullyConnectedBiasedUint8, FullyConnectedUint8Test, true)
205ARMNN_AUTO_TEST_CASE(SimpleFullyConnectedWithTranspose, FullyConnectedFloat32Test, false, true)
206
207ARMNN_AUTO_TEST_CASE(FullyConnectedLarge, FullyConnectedLargeTest, false)
208ARMNN_AUTO_TEST_CASE(FullyConnectedLargeTransposed, FullyConnectedLargeTest, true)
209
210// Splitter
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]211ARMNN_AUTO_TEST_CASE(SimpleSplitter, SplitterTest)
212ARMNN_AUTO_TEST_CASE(SimpleSplitterUint8, SplitterUint8Test)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]213
214ARMNN_AUTO_TEST_CASE(CopyViaSplitter, CopyViaSplitterTest)
215ARMNN_AUTO_TEST_CASE(CopyViaSplitterUint8, CopyViaSplitterUint8Test)
216
217// Merger
218ARMNN_AUTO_TEST_CASE(SimpleMerger, MergerTest)
219ARMNN_AUTO_TEST_CASE(MergerUint8, MergerUint8Test)
Ferran Balaguerb2845652019-02-27 09:42:06 +0000[diff] [blame]220ARMNN_AUTO_TEST_CASE(MergerUint8DifferentQParams, MergerUint8DifferentQParamsTest)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]221
222// Add
223ARMNN_AUTO_TEST_CASE(SimpleAdd, AdditionTest)
224ARMNN_AUTO_TEST_CASE(AddBroadcast1Element, AdditionBroadcast1ElementTest)
225ARMNN_AUTO_TEST_CASE(AddBroadcast, AdditionBroadcastTest)
226
227ARMNN_AUTO_TEST_CASE(AdditionUint8, AdditionUint8Test)
228ARMNN_AUTO_TEST_CASE(AddBroadcastUint8, AdditionBroadcastUint8Test)
229ARMNN_AUTO_TEST_CASE(AddBroadcast1ElementUint8, AdditionBroadcast1ElementUint8Test)
230
Sadik Armagan2999a022019-04-09 14:20:12 +0100[diff] [blame]231ARMNN_AUTO_TEST_CASE(AdditionInt16, AdditionInt16Test)
232ARMNN_AUTO_TEST_CASE(AddBroadcastInt16, AdditionBroadcastInt16Test)
233ARMNN_AUTO_TEST_CASE(AddBroadcast1ElementInt16, AdditionBroadcast1ElementInt16Test)
234
David Beckf195f032018-09-06 16:46:34 +0100[diff] [blame]235// Sub
236ARMNN_AUTO_TEST_CASE(SimpleSub, SubtractionTest)
237ARMNN_AUTO_TEST_CASE(SubBroadcast1Element, SubtractionBroadcast1ElementTest)
238ARMNN_AUTO_TEST_CASE(SubBroadcast, SubtractionBroadcastTest)
239
David Beck4a8692c2018-09-07 16:19:24 +0100[diff] [blame]240ARMNN_AUTO_TEST_CASE(SubtractionUint8, SubtractionUint8Test)
David Beckf195f032018-09-06 16:46:34 +0100[diff] [blame]241ARMNN_AUTO_TEST_CASE(SubBroadcastUint8, SubtractionBroadcastUint8Test)
242ARMNN_AUTO_TEST_CASE(SubBroadcast1ElementUint8, SubtractionBroadcast1ElementUint8Test)
243
Sadik Armagan2999a022019-04-09 14:20:12 +0100[diff] [blame]244ARMNN_AUTO_TEST_CASE(SubtractionInt16, SubtractionInt16Test)
245ARMNN_AUTO_TEST_CASE(SubBroadcastInt16, SubtractionBroadcastInt16Test)
246ARMNN_AUTO_TEST_CASE(SubBroadcast1ElementInt16, SubtractionBroadcast1ElementInt16Test)
247
Francis Murtaghe7a86a42018-08-29 12:42:10 +0100[diff] [blame]248// Div
249ARMNN_AUTO_TEST_CASE(SimpleDivision, DivisionTest)
Francis Murtagh8c5e3dc2018-08-30 17:18:37 +0100[diff] [blame]250ARMNN_AUTO_TEST_CASE(DivisionByZero, DivisionByZeroTest)
Francis Murtaghe7a86a42018-08-29 12:42:10 +0100[diff] [blame]251ARMNN_AUTO_TEST_CASE(DivisionBroadcast1Element, DivisionBroadcast1ElementTest)
252ARMNN_AUTO_TEST_CASE(DivisionBroadcast1DVector, DivisionBroadcast1DVectorTest)
David Beck5cd01f32018-09-12 16:00:08 +0100[diff] [blame]253// NOTE: division by zero for quantized div needs more attention
254// see IVGCVSW-1849
255ARMNN_AUTO_TEST_CASE(DivisionUint8, DivisionUint8Test)
256ARMNN_AUTO_TEST_CASE(DivisionUint8Broadcast1Element, DivisionBroadcast1ElementUint8Test)
257ARMNN_AUTO_TEST_CASE(DivisionUint8Broadcast1DVector, DivisionBroadcast1DVectorUint8Test)
Francis Murtaghe7a86a42018-08-29 12:42:10 +0100[diff] [blame]258
Sadik Armagan2999a022019-04-09 14:20:12 +0100[diff] [blame]259ARMNN_AUTO_TEST_CASE(DivisionInt16, DivisionInt16Test)
260ARMNN_AUTO_TEST_CASE(DivisionInt16Broadcast1Element, DivisionBroadcast1ElementInt16Test)
261ARMNN_AUTO_TEST_CASE(DivisionInt16Broadcast1DVector, DivisionBroadcast1DVectorInt16Test)
262
FrancisMurtagh30cdfca2018-12-18 12:57:35 +0000[diff] [blame]263// Equal
264ARMNN_AUTO_TEST_CASE(SimpleEqual, EqualSimpleTest)
265ARMNN_AUTO_TEST_CASE(EqualBroadcast1Element, EqualBroadcast1ElementTest)
266ARMNN_AUTO_TEST_CASE(EqualBroadcast1DVector, EqualBroadcast1DVectorTest)
267ARMNN_AUTO_TEST_CASE(EqualUint8, EqualUint8Test)
268ARMNN_AUTO_TEST_CASE(EqualBroadcast1ElementUint8, EqualBroadcast1ElementUint8Test)
269ARMNN_AUTO_TEST_CASE(EqualBroadcast1DVectorUint8, EqualBroadcast1DVectorUint8Test)
270
FrancisMurtagh878f0232018-12-19 10:56:15 +0000[diff] [blame]271// Greater
272ARMNN_AUTO_TEST_CASE(SimpleGreater, GreaterSimpleTest)
273ARMNN_AUTO_TEST_CASE(GreaterBroadcast1Element, GreaterBroadcast1ElementTest)
274ARMNN_AUTO_TEST_CASE(GreaterBroadcast1DVector, GreaterBroadcast1DVectorTest)
275ARMNN_AUTO_TEST_CASE(GreaterUint8, GreaterUint8Test)
276ARMNN_AUTO_TEST_CASE(GreaterBroadcast1ElementUint8, GreaterBroadcast1ElementUint8Test)
277ARMNN_AUTO_TEST_CASE(GreaterBroadcast1DVectorUint8, GreaterBroadcast1DVectorUint8Test)
278
Éanna Ó Catháin20e58802018-12-04 10:29:06 +0000[diff] [blame]279// Max
Éanna Ó Catháinde705582018-12-03 13:04:22 +0000[diff] [blame]280ARMNN_AUTO_TEST_CASE(SimpleMaximum, MaximumSimpleTest)
281ARMNN_AUTO_TEST_CASE(MaximumBroadcast1Element, MaximumBroadcast1ElementTest)
282ARMNN_AUTO_TEST_CASE(MaximumBroadcast1DVector, MaximumBroadcast1DVectorTest)
283ARMNN_AUTO_TEST_CASE(MaximumUint8, MaximumUint8Test)
284ARMNN_AUTO_TEST_CASE(MaximumBroadcast1ElementUint8, MaximumBroadcast1ElementUint8Test)
285ARMNN_AUTO_TEST_CASE(MaximumBroadcast1DVectorUint8, MaximumBroadcast1DVectorUint8Test)
Sadik Armagan2999a022019-04-09 14:20:12 +0100[diff] [blame]286ARMNN_AUTO_TEST_CASE(MaximumInt16, MaximumInt16Test)
287ARMNN_AUTO_TEST_CASE(MaximumBroadcast1ElementInt16, MaximumBroadcast1ElementInt16Test)
288ARMNN_AUTO_TEST_CASE(MaximumBroadcast1DVectorInt16, MaximumBroadcast1DVectorInt16Test)
Éanna Ó Catháinde705582018-12-03 13:04:22 +0000[diff] [blame]289
Éanna Ó Catháin20e58802018-12-04 10:29:06 +0000[diff] [blame]290// Min
291ARMNN_AUTO_TEST_CASE(SimpleMinimum1, MinimumBroadcast1ElementTest1)
292ARMNN_AUTO_TEST_CASE(SimpleMinimum2, MinimumBroadcast1ElementTest2)
293ARMNN_AUTO_TEST_CASE(Minimum1DVectorUint8, MinimumBroadcast1DVectorUint8Test)
Sadik Armagan2999a022019-04-09 14:20:12 +0100[diff] [blame]294ARMNN_AUTO_TEST_CASE(MinimumInt16, MinimumInt16Test)
295ARMNN_AUTO_TEST_CASE(MinimumBroadcast1ElementInt16, MinimumBroadcast1ElementInt16Test)
296ARMNN_AUTO_TEST_CASE(MinimumBroadcast1DVectorInt16, MinimumBroadcast1DVectorInt16Test)
Éanna Ó Catháin20e58802018-12-04 10:29:06 +0000[diff] [blame]297
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]298// Mul
299ARMNN_AUTO_TEST_CASE(SimpleMultiplication, MultiplicationTest)
surmeh01bceff2f2018-03-29 16:29:27 +0100[diff] [blame]300ARMNN_AUTO_TEST_CASE(MultiplicationBroadcast1Element, MultiplicationBroadcast1ElementTest)
301ARMNN_AUTO_TEST_CASE(MultiplicationBroadcast1DVector, MultiplicationBroadcast1DVectorTest)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]302ARMNN_AUTO_TEST_CASE(MultiplicationUint8, MultiplicationUint8Test)
surmeh01bceff2f2018-03-29 16:29:27 +0100[diff] [blame]303ARMNN_AUTO_TEST_CASE(MultiplicationBroadcast1ElementUint8, MultiplicationBroadcast1ElementUint8Test)
304ARMNN_AUTO_TEST_CASE(MultiplicationBroadcast1DVectorUint8, MultiplicationBroadcast1DVectorUint8Test)
Sadik Armagan2999a022019-04-09 14:20:12 +0100[diff] [blame]305ARMNN_AUTO_TEST_CASE(MultiplicationInt16, MultiplicationInt16Test)
306ARMNN_AUTO_TEST_CASE(MultiplicationBroadcast1ElementInt16, MultiplicationBroadcast1ElementInt16Test)
307ARMNN_AUTO_TEST_CASE(MultiplicationBroadcast1DVectorInt16, MultiplicationBroadcast1DVectorInt16Test)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]308
309// Batch Norm
310ARMNN_AUTO_TEST_CASE(BatchNorm, BatchNormTest)
Matteo Martincigh8eb675e2018-10-17 14:43:29 +0100[diff] [blame]311ARMNN_AUTO_TEST_CASE(BatchNormNhwc, BatchNormNhwcTest)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]312ARMNN_AUTO_TEST_CASE(BatchNormUint8, BatchNormUint8Test)
Matteo Martincigh8eb675e2018-10-17 14:43:29 +0100[diff] [blame]313ARMNN_AUTO_TEST_CASE(BatchNormUint8Nhwc, BatchNormUint8NhwcTest)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]314
James Conroy59540822018-10-11 12:39:05 +0100[diff] [blame]315// Resize Bilinear - NCHW
James Conroy6b965822018-11-01 11:33:09 +0000[diff] [blame]316ARMNN_AUTO_TEST_CASE(SimpleResizeBilinear, SimpleResizeBilinearTest, armnn::DataLayout::NCHW)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]317ARMNN_AUTO_TEST_CASE(SimpleResizeBilinearUint8, SimpleResizeBilinearUint8Test)
James Conroy6b965822018-11-01 11:33:09 +0000[diff] [blame]318ARMNN_AUTO_TEST_CASE(ResizeBilinearNop, ResizeBilinearNopTest, armnn::DataLayout::NCHW)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]319ARMNN_AUTO_TEST_CASE(ResizeBilinearNopUint8, ResizeBilinearNopUint8Test)
James Conroy6b965822018-11-01 11:33:09 +0000[diff] [blame]320ARMNN_AUTO_TEST_CASE(ResizeBilinearSqMin, ResizeBilinearSqMinTest, armnn::DataLayout::NCHW)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]321ARMNN_AUTO_TEST_CASE(ResizeBilinearSqMinUint8, ResizeBilinearSqMinUint8Test)
James Conroy6b965822018-11-01 11:33:09 +0000[diff] [blame]322ARMNN_AUTO_TEST_CASE(ResizeBilinearMin, ResizeBilinearMinTest, armnn::DataLayout::NCHW)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]323ARMNN_AUTO_TEST_CASE(ResizeBilinearMinUint8, ResizeBilinearMinUint8Test)
James Conroy6b965822018-11-01 11:33:09 +0000[diff] [blame]324ARMNN_AUTO_TEST_CASE(ResizeBilinearMag, ResizeBilinearMagTest, armnn::DataLayout::NCHW)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]325ARMNN_AUTO_TEST_CASE(ResizeBilinearMagUint8, ResizeBilinearMagUint8Test)
326
James Conroy59540822018-10-11 12:39:05 +0100[diff] [blame]327// Resize Bilinear - NHWC
James Conroy6b965822018-11-01 11:33:09 +0000[diff] [blame]328ARMNN_AUTO_TEST_CASE(ResizeBilinearNopNhwc, ResizeBilinearNopTest, armnn::DataLayout::NHWC)
329ARMNN_AUTO_TEST_CASE(SimpleResizeBilinearNhwc, SimpleResizeBilinearTest, armnn::DataLayout::NHWC)
330ARMNN_AUTO_TEST_CASE(ResizeBilinearSqMinNhwc, ResizeBilinearSqMinTest, armnn::DataLayout::NHWC)
331ARMNN_AUTO_TEST_CASE(ResizeBilinearMinNhwc, ResizeBilinearMinTest, armnn::DataLayout::NHWC)
332ARMNN_AUTO_TEST_CASE(ResizeBilinearMagNhwc, ResizeBilinearMagTest, armnn::DataLayout::NHWC)
James Conroy59540822018-10-11 12:39:05 +0100[diff] [blame]333
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]334// Fake Quantization
335ARMNN_AUTO_TEST_CASE(FakeQuantization, FakeQuantizationTest)
336
Matteo Martincigh539b44d2018-10-01 09:26:39 +0100[diff] [blame]337// L2 Normalization
jimfly013aab7c32018-11-12 13:32:08 +0000[diff] [blame]338ARMNN_AUTO_TEST_CASE(L2Normalization1d, L2Normalization1dTest, armnn::DataLayout::NCHW)
339ARMNN_AUTO_TEST_CASE(L2Normalization2d, L2Normalization2dTest, armnn::DataLayout::NCHW)
340ARMNN_AUTO_TEST_CASE(L2Normalization3d, L2Normalization3dTest, armnn::DataLayout::NCHW)
341ARMNN_AUTO_TEST_CASE(L2Normalization4d, L2Normalization4dTest, armnn::DataLayout::NCHW)
342
343ARMNN_AUTO_TEST_CASE(L2Normalization1dNhwc, L2Normalization1dTest, armnn::DataLayout::NHWC)
344ARMNN_AUTO_TEST_CASE(L2Normalization2dNhwc, L2Normalization2dTest, armnn::DataLayout::NHWC)
345ARMNN_AUTO_TEST_CASE(L2Normalization3dNhwc, L2Normalization3dTest, armnn::DataLayout::NHWC)
346ARMNN_AUTO_TEST_CASE(L2Normalization4dNhwc, L2Normalization4dTest, armnn::DataLayout::NHWC)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]347
Mohamed Nour Abouelseoud7420e552018-10-12 12:26:24 +0100[diff] [blame]348// Pad
Mohamed Nour Abouelseouddd6acea2018-10-18 12:26:19 +0100[diff] [blame]349ARMNN_AUTO_TEST_CASE(PadFloat322d, PadFloat322dTest)
350ARMNN_AUTO_TEST_CASE(PadFloat323d, PadFloat323dTest)
351ARMNN_AUTO_TEST_CASE(PadFloat324d, PadFloat324dTest)
352
353ARMNN_AUTO_TEST_CASE(PadUint82d, PadUint82dTest)
354ARMNN_AUTO_TEST_CASE(PadUint83d, PadUint83dTest)
355ARMNN_AUTO_TEST_CASE(PadUint84d, PadUint84dTest)
Mohamed Nour Abouelseoud7420e552018-10-12 12:26:24 +0100[diff] [blame]356
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]357// Constant
358ARMNN_AUTO_TEST_CASE(Constant, ConstantTest)
359ARMNN_AUTO_TEST_CASE(ConstantUint8, ConstantUint8Test)
360
361// Concat
362ARMNN_AUTO_TEST_CASE(Concatenation1d, Concatenation1dTest)
363ARMNN_AUTO_TEST_CASE(Concatenation1dUint8, Concatenation1dUint8Test)
364
365ARMNN_AUTO_TEST_CASE(Concatenation2dDim0, Concatenation2dDim0Test)
366ARMNN_AUTO_TEST_CASE(Concatenation2dDim0Uint8, Concatenation2dDim0Uint8Test)
367ARMNN_AUTO_TEST_CASE(Concatenation2dDim1, Concatenation2dDim1Test)
368ARMNN_AUTO_TEST_CASE(Concatenation2dDim1Uint8, Concatenation2dDim1Uint8Test)
369
370ARMNN_AUTO_TEST_CASE(Concatenation2dDim0DiffInputDims, Concatenation2dDim0DiffInputDimsTest)
371ARMNN_AUTO_TEST_CASE(Concatenation2dDim0DiffInputDimsUint8, Concatenation2dDim0DiffInputDimsUint8Test)
372ARMNN_AUTO_TEST_CASE(Concatenation2dDim1DiffInputDims, Concatenation2dDim1DiffInputDimsTest)
373ARMNN_AUTO_TEST_CASE(Concatenation2dDim1DiffInputDimsUint8, Concatenation2dDim1DiffInputDimsUint8Test)
374
375ARMNN_AUTO_TEST_CASE(Concatenation3dDim0, Concatenation3dDim0Test)
376ARMNN_AUTO_TEST_CASE(Concatenation3dDim0Uint8, Concatenation3dDim0Uint8Test)
377ARMNN_AUTO_TEST_CASE(Concatenation3dDim1, Concatenation3dDim1Test)
378ARMNN_AUTO_TEST_CASE(Concatenation3dDim1Uint8, Concatenation3dDim1Uint8Test)
narpra015cdda352018-11-19 15:30:27 +0000[diff] [blame]379ARMNN_AUTO_TEST_CASE(Concatenation3dDim2, Concatenation3dDim2Test, true)
380ARMNN_AUTO_TEST_CASE(Concatenation3dDim2Uint8, Concatenation3dDim2Uint8Test, true)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]381
382ARMNN_AUTO_TEST_CASE(Concatenation3dDim0DiffInputDims, Concatenation3dDim0DiffInputDimsTest)
383ARMNN_AUTO_TEST_CASE(Concatenation3dDim0DiffInputDimsUint8, Concatenation3dDim0DiffInputDimsUint8Test)
384ARMNN_AUTO_TEST_CASE(Concatenation3dDim1DiffInputDims, Concatenation3dDim1DiffInputDimsTest)
385ARMNN_AUTO_TEST_CASE(Concatenation3dDim1DiffInputDimsUint8, Concatenation3dDim1DiffInputDimsUint8Test)
narpra015cdda352018-11-19 15:30:27 +0000[diff] [blame]386ARMNN_AUTO_TEST_CASE(Concatenation3dDim2DiffInputDims, Concatenation3dDim2DiffInputDimsTest, true)
387ARMNN_AUTO_TEST_CASE(Concatenation3dDim2DiffInputDimsUint8, Concatenation3dDim2DiffInputDimsUint8Test, true)
388
389ARMNN_AUTO_TEST_CASE(Concatenation4dDim0, Concatenation4dDim0Test)
390ARMNN_AUTO_TEST_CASE(Concatenation4dDim1, Concatenation4dDim1Test)
391ARMNN_AUTO_TEST_CASE(Concatenation4dDim2, Concatenation4dDim2Test)
392ARMNN_AUTO_TEST_CASE(Concatenation4dDim3, Concatenation4dDim3Test, true)
393ARMNN_AUTO_TEST_CASE(Concatenation4dDim0Uint8, Concatenation4dDim0Uint8Test)
394ARMNN_AUTO_TEST_CASE(Concatenation4dDim1Uint8, Concatenation4dDim1Uint8Test)
395ARMNN_AUTO_TEST_CASE(Concatenation4dDim2Uint8, Concatenation4dDim2Uint8Test)
396ARMNN_AUTO_TEST_CASE(Concatenation4dDim3Uint8, Concatenation4dDim3Uint8Test, true)
397
398ARMNN_AUTO_TEST_CASE(Concatenation4dDiffShapeDim0, Concatenation4dDiffShapeDim0Test)
399ARMNN_AUTO_TEST_CASE(Concatenation4dDiffShapeDim1, Concatenation4dDiffShapeDim1Test)
400ARMNN_AUTO_TEST_CASE(Concatenation4dDiffShapeDim2, Concatenation4dDiffShapeDim2Test)
401ARMNN_AUTO_TEST_CASE(Concatenation4dDiffShapeDim3, Concatenation4dDiffShapeDim3Test, true)
402ARMNN_AUTO_TEST_CASE(Concatenation4dDiffShapeDim0Uint8, Concatenation4dDiffShapeDim0Uint8Test)
403ARMNN_AUTO_TEST_CASE(Concatenation4dDiffShapeDim1Uint8, Concatenation4dDiffShapeDim1Uint8Test)
404ARMNN_AUTO_TEST_CASE(Concatenation4dDiffShapeDim2Uint8, Concatenation4dDiffShapeDim2Uint8Test)
405ARMNN_AUTO_TEST_CASE(Concatenation4dDiffShapeDim3Uint8, Concatenation4dDiffShapeDim3Uint8Test, true)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]406
407// Floor
408ARMNN_AUTO_TEST_CASE(SimpleFloor, SimpleFloorTest)
409
410// Reshape
411ARMNN_AUTO_TEST_CASE(SimpleReshapeFloat32, SimpleReshapeFloat32Test)
412ARMNN_AUTO_TEST_CASE(SimpleReshapeUint8, SimpleReshapeUint8Test)
413
Mohamed Nour Abouelseouda1d3c6a2018-12-27 12:39:16 +0000[diff] [blame]414// Rsqrt
415ARMNN_AUTO_TEST_CASE(Rsqrt2d, Rsqrt2dTest)
416ARMNN_AUTO_TEST_CASE(Rsqrt3d, Rsqrt3dTest)
417ARMNN_AUTO_TEST_CASE(RsqrtZero, RsqrtZeroTest)
418ARMNN_AUTO_TEST_CASE(RsqrtNegative, RsqrtNegativeTest)
419
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]420// Permute
421ARMNN_AUTO_TEST_CASE(SimplePermuteFloat32, SimplePermuteFloat32Test)
422ARMNN_AUTO_TEST_CASE(SimplePermuteUint8, SimplePermuteUint8Test)
surmeh01bceff2f2018-03-29 16:29:27 +0100[diff] [blame]423ARMNN_AUTO_TEST_CASE(PermuteFloat32ValueSet1, PermuteFloat32ValueSet1Test)
424ARMNN_AUTO_TEST_CASE(PermuteFloat32ValueSet2, PermuteFloat32ValueSet2Test)
425ARMNN_AUTO_TEST_CASE(PermuteFloat32ValueSet3, PermuteFloat32ValueSet3Test)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]426
Matteo Martincigha65b7ae2018-11-14 12:39:55 +0000[diff] [blame]427// Lstm
428ARMNN_AUTO_TEST_CASE(LstmLayerFloat32WithCifgWithPeepholeNoProjection,
429 LstmLayerFloat32WithCifgWithPeepholeNoProjectionTest)
430ARMNN_AUTO_TEST_CASE(LstmLayerFloat32NoCifgNoPeepholeNoProjection,
431 LstmLayerFloat32NoCifgNoPeepholeNoProjectionTest)
432ARMNN_AUTO_TEST_CASE(LstmLayerFloat32NoCifgWithPeepholeWithProjection,
433 LstmLayerFloat32NoCifgWithPeepholeWithProjectionTest)
434
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]435// Convert from Float16 to Float32
436ARMNN_AUTO_TEST_CASE(SimpleConvertFp16ToFp32, SimpleConvertFp16ToFp32Test)
437// Convert from Float32 to Float16
438ARMNN_AUTO_TEST_CASE(SimpleConvertFp32ToFp16, SimpleConvertFp32ToFp16Test)
439
narpra011e4c31d2018-09-28 11:07:51 +0100[diff] [blame]440// Mean
441ARMNN_AUTO_TEST_CASE(MeanUint8Simple, MeanUint8SimpleTest)
442ARMNN_AUTO_TEST_CASE(MeanUint8SimpleAxis, MeanUint8SimpleAxisTest)
443ARMNN_AUTO_TEST_CASE(MeanUint8KeepDims, MeanUint8KeepDimsTest)
444ARMNN_AUTO_TEST_CASE(MeanUint8MultipleDims, MeanUint8MultipleDimsTest)
445ARMNN_AUTO_TEST_CASE(MeanVtsUint8, MeanVtsUint8Test)
446
447ARMNN_AUTO_TEST_CASE(MeanFloatSimple, MeanFloatSimpleTest)
448ARMNN_AUTO_TEST_CASE(MeanFloatSimpleAxis, MeanFloatSimpleAxisTest)
449ARMNN_AUTO_TEST_CASE(MeanFloatKeepDims, MeanFloatKeepDimsTest)
450ARMNN_AUTO_TEST_CASE(MeanFloatMultipleDims, MeanFloatMultipleDimsTest)
451ARMNN_AUTO_TEST_CASE(MeanVtsFloat1, MeanVtsFloat1Test)
452ARMNN_AUTO_TEST_CASE(MeanVtsFloat2, MeanVtsFloat2Test)
Matteo Martincigh28dcab62018-10-19 16:40:03 +0100[diff] [blame]453ARMNN_AUTO_TEST_CASE(MeanVtsFloat3, MeanVtsFloat3Test)
narpra011e4c31d2018-09-28 11:07:51 +0100[diff] [blame]454
Éanna Ó Catháin47c1ddb2018-10-12 14:24:13 +0100[diff] [blame]455ARMNN_AUTO_TEST_CASE(AdditionAfterMaxPool, AdditionAfterMaxPoolTest)
456
Nattapat Chaimanowong3ea76d52018-11-09 14:10:38 +0000[diff] [blame]457// Space To Batch Nd
458ARMNN_AUTO_TEST_CASE(SpaceToBatchNdSimpleFloat32, SpaceToBatchNdSimpleFloat32Test)
459ARMNN_AUTO_TEST_CASE(SpaceToBatchNdMultiChannelsFloat32, SpaceToBatchNdMultiChannelsFloat32Test)
460ARMNN_AUTO_TEST_CASE(SpaceToBatchNdMultiBlockFloat32, SpaceToBatchNdMultiBlockFloat32Test)
461ARMNN_AUTO_TEST_CASE(SpaceToBatchNdPaddingFloat32, SpaceToBatchNdPaddingFloat32Test)
462
463ARMNN_AUTO_TEST_CASE(SpaceToBatchNdSimpleUint8, SpaceToBatchNdSimpleUint8Test)
464ARMNN_AUTO_TEST_CASE(SpaceToBatchNdMultiChannelsUint8, SpaceToBatchNdMultiChannelsUint8Test)
465ARMNN_AUTO_TEST_CASE(SpaceToBatchNdMultiBlockUint8, SpaceToBatchNdMultiBlockUint8Test)
466ARMNN_AUTO_TEST_CASE(SpaceToBatchNdPaddingUint8, SpaceToBatchNdPaddingUint8Test)
467
468ARMNN_AUTO_TEST_CASE(SpaceToBatchNdSimpleNHWCFloat32, SpaceToBatchNdSimpleNHWCFloat32Test)
469ARMNN_AUTO_TEST_CASE(SpaceToBatchNdMultiChannelsNHWCFloat32, SpaceToBatchNdMultiChannelsNHWCFloat32Test)
470ARMNN_AUTO_TEST_CASE(SpaceToBatchNdMultiBlockNHWCFloat32, SpaceToBatchNdMultiBlockNHWCFloat32Test)
471ARMNN_AUTO_TEST_CASE(SpaceToBatchNdPaddingNHWCFloat32, SpaceToBatchNdPaddingNHWCFloat32Test)
472
473ARMNN_AUTO_TEST_CASE(SpaceToBatchNdSimpleNHWCUint8, SpaceToBatchNdSimpleNHWCUint8Test)
474ARMNN_AUTO_TEST_CASE(SpaceToBatchNdMultiChannelsNHWCUint8, SpaceToBatchNdMultiChannelsNHWCUint8Test)
475ARMNN_AUTO_TEST_CASE(SpaceToBatchNdMultiBlockNHWCUint8, SpaceToBatchNdMultiBlockNHWCUint8Test)
476ARMNN_AUTO_TEST_CASE(SpaceToBatchNdPaddingNHWCUint8, SpaceToBatchNdPaddingNHWCUint8Test)
477
Éanna Ó Catháin4e1e1362018-11-12 11:36:34 +0000[diff] [blame]478ARMNN_AUTO_TEST_CASE(BatchToSpaceNdNhwcFloat321, BatchToSpaceNdNhwcFloat32Test1)
479ARMNN_AUTO_TEST_CASE(BatchToSpaceNdNhwcFloat322, BatchToSpaceNdNhwcFloat32Test2)
480ARMNN_AUTO_TEST_CASE(BatchToSpaceNdNhwcFloat323, BatchToSpaceNdNhwcFloat32Test3)
Nattapat Chaimanowong3ee14222019-02-27 10:28:09 +0000[diff] [blame]481ARMNN_AUTO_TEST_CASE(BatchToSpaceNdNhwcFloat324, BatchToSpaceNdNhwcFloat32Test4)
Éanna Ó Catháin4e1e1362018-11-12 11:36:34 +0000[diff] [blame]482
483ARMNN_AUTO_TEST_CASE(BatchToSpaceNdNchwFloat321, BatchToSpaceNdNchwFloat32Test1)
Mike Kelly831faed2018-11-28 11:52:08 +0000[diff] [blame]484ARMNN_AUTO_TEST_CASE(BatchToSpaceNdNchwFloat322, BatchToSpaceNdNchwFloat32Test2)
485ARMNN_AUTO_TEST_CASE(BatchToSpaceNdNchwFloat323, BatchToSpaceNdNchwFloat32Test3)
Éanna Ó Catháin4e1e1362018-11-12 11:36:34 +0000[diff] [blame]486
Éanna Ó Catháin262553e2018-11-14 11:26:23 +0000[diff] [blame]487ARMNN_AUTO_TEST_CASE(BatchToSpaceNdNhwcUint1, BatchToSpaceNdNhwcUintTest1)
Mike Kelly831faed2018-11-28 11:52:08 +0000[diff] [blame]488ARMNN_AUTO_TEST_CASE(BatchToSpaceNdNhwcUint2, BatchToSpaceNdNhwcUintTest2)
489ARMNN_AUTO_TEST_CASE(BatchToSpaceNdNhwcUint3, BatchToSpaceNdNhwcUintTest3)
490
491ARMNN_AUTO_TEST_CASE(BatchToSpaceNdNchwUint1, BatchToSpaceNdNchwUintTest1)
492ARMNN_AUTO_TEST_CASE(BatchToSpaceNdNchwUint2, BatchToSpaceNdNchwUintTest2)
493ARMNN_AUTO_TEST_CASE(BatchToSpaceNdNchwUint3, BatchToSpaceNdNchwUintTest3)
Nattapat Chaimanowong3ee14222019-02-27 10:28:09 +0000[diff] [blame]494ARMNN_AUTO_TEST_CASE(BatchToSpaceNdNchwUint4, BatchToSpaceNdNchwUintTest4)
Éanna Ó Catháin262553e2018-11-14 11:26:23 +0000[diff] [blame]495
Nattapat Chaimanowong1216b582018-11-23 15:33:41 +0000[diff] [blame]496// Strided Slice
497ARMNN_AUTO_TEST_CASE(StridedSlice4DFloat32, StridedSlice4DFloat32Test)
498ARMNN_AUTO_TEST_CASE(StridedSlice4DReverseFloat32, StridedSlice4DReverseFloat32Test)
499ARMNN_AUTO_TEST_CASE(StridedSliceSimpleStrideFloat32, StridedSliceSimpleStrideFloat32Test)
500ARMNN_AUTO_TEST_CASE(StridedSliceSimpleRangeMaskFloat32, StridedSliceSimpleRangeMaskFloat32Test)
501ARMNN_AUTO_TEST_CASE(StridedSliceShrinkAxisMaskFloat32, StridedSliceShrinkAxisMaskFloat32Test)
502ARMNN_AUTO_TEST_CASE(StridedSlice3DFloat32, StridedSlice3DFloat32Test)
503ARMNN_AUTO_TEST_CASE(StridedSlice3DReverseFloat32, StridedSlice3DReverseFloat32Test)
504ARMNN_AUTO_TEST_CASE(StridedSlice2DFloat32, StridedSlice2DFloat32Test)
505ARMNN_AUTO_TEST_CASE(StridedSlice2DReverseFloat32, StridedSlice2DReverseFloat32Test)
506
507ARMNN_AUTO_TEST_CASE(StridedSlice4DUint8, StridedSlice4DUint8Test)
508ARMNN_AUTO_TEST_CASE(StridedSlice4DReverseUint8, StridedSlice4DReverseUint8Test)
509ARMNN_AUTO_TEST_CASE(StridedSliceSimpleStrideUint8, StridedSliceSimpleStrideUint8Test)
510ARMNN_AUTO_TEST_CASE(StridedSliceSimpleRangeMaskUint8, StridedSliceSimpleRangeMaskUint8Test)
511ARMNN_AUTO_TEST_CASE(StridedSliceShrinkAxisMaskUint8, StridedSliceShrinkAxisMaskUint8Test)
512ARMNN_AUTO_TEST_CASE(StridedSlice3DUint8, StridedSlice3DUint8Test)
513ARMNN_AUTO_TEST_CASE(StridedSlice3DReverseUint8, StridedSlice3DReverseUint8Test)
514ARMNN_AUTO_TEST_CASE(StridedSlice2DUint8, StridedSlice2DUint8Test)
515ARMNN_AUTO_TEST_CASE(StridedSlice2DReverseUint8, StridedSlice2DReverseUint8Test)
516
Nattapat Chaimanowongcfdcadf2018-12-06 11:54:33 +0000[diff] [blame]517// Debug
518ARMNN_AUTO_TEST_CASE(Debug4DFloat32, Debug4DFloat32Test)
519ARMNN_AUTO_TEST_CASE(Debug3DFloat32, Debug3DFloat32Test)
520ARMNN_AUTO_TEST_CASE(Debug2DFloat32, Debug2DFloat32Test)
521ARMNN_AUTO_TEST_CASE(Debug1DFloat32, Debug1DFloat32Test)
522
523ARMNN_AUTO_TEST_CASE(Debug4DUint8, Debug4DUint8Test)
524ARMNN_AUTO_TEST_CASE(Debug3DUint8, Debug3DUint8Test)
525ARMNN_AUTO_TEST_CASE(Debug2DUint8, Debug2DUint8Test)
526ARMNN_AUTO_TEST_CASE(Debug1DUint8, Debug1DUint8Test)
527
narpra014951d842019-01-18 16:53:53 +0000[diff] [blame]528// Gather
529ARMNN_AUTO_TEST_CASE(Gather1DParamsFloat, Gather1DParamsFloatTest)
530ARMNN_AUTO_TEST_CASE(Gather1DParamsUint8, Gather1DParamsUint8Test)
531ARMNN_AUTO_TEST_CASE(GatherMultiDimParamsFloat, GatherMultiDimParamsFloatTest)
532ARMNN_AUTO_TEST_CASE(GatherMultiDimParamsUint8, GatherMultiDimParamsUint8Test)
533ARMNN_AUTO_TEST_CASE(GatherMultiDimParamsMultiDimIndicesFloat, GatherMultiDimParamsMultiDimIndicesFloatTest)
534ARMNN_AUTO_TEST_CASE(GatherMultiDimParamsMultiDimIndicesUint8, GatherMultiDimParamsMultiDimIndicesUint8Test)
535
Narumol Prangnawarate0a4ad82019-02-04 19:05:27 +0000[diff] [blame]536// Detection PostProcess
537BOOST_AUTO_TEST_CASE(DetectionPostProcessRegularNmsFloat)
538{
539 DetectionPostProcessRegularNmsFloatTest<armnn::RefWorkloadFactory>();
540}
541BOOST_AUTO_TEST_CASE(DetectionPostProcessFastNmsFloat)
542{
543 DetectionPostProcessFastNmsFloatTest<armnn::RefWorkloadFactory>();
544}
545BOOST_AUTO_TEST_CASE(DetectionPostProcessRegularNmsUint8)
546{
547 DetectionPostProcessRegularNmsUint8Test<armnn::RefWorkloadFactory>();
548}
549BOOST_AUTO_TEST_CASE(DetectionPostProcessFastNmsUint8)
550{
551 DetectionPostProcessFastNmsUint8Test<armnn::RefWorkloadFactory>();
552}
553
Nattapat Chaimanowong8a54ac02019-03-29 15:25:04 +0000[diff] [blame]554// Dequantize
Nattapat Chaimanowongafa4e3a2019-04-02 11:41:45 +0100[diff] [blame]555ARMNN_AUTO_TEST_CASE(DequantizeSimpleUint8, DequantizeSimpleUint8Test)
556ARMNN_AUTO_TEST_CASE(DequantizeOffsetUint8, DequantizeOffsetUint8Test)
557ARMNN_AUTO_TEST_CASE(DequantizeSimpleInt16, DequantizeSimpleInt16Test)
Narumol Prangnawarate0a4ad82019-02-04 19:05:27 +0000[diff] [blame]558
Nattapat Chaimanowonga0beb3b2019-04-01 17:04:53 +0100[diff] [blame]559// Quantize
560ARMNN_AUTO_TEST_CASE(QuantizeSimpleUint8, QuantizeSimpleUint8Test)
561ARMNN_AUTO_TEST_CASE(QuantizeClampUint8, QuantizeClampUint8Test)
562ARMNN_AUTO_TEST_CASE(QuantizeClampInt16, QuantizeClampInt16Test)
563
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]564BOOST_AUTO_TEST_SUITE_END()