blob: 39cd4c4a0b0a0c6a3c471d0626a1cd4b7f706e3e [file] [log] [blame]
telsoa014fcda012018-03-09 14:13:49 +00001//
2// Copyright © 2017 Arm Ltd. All rights reserved.
David Beckecb56cd2018-09-05 12:52:57 +01003// SPDX-License-Identifier: MIT
telsoa014fcda012018-03-09 14:13:49 +00004//
5#pragma once
6
7#include "armnn/ArmNN.hpp"
8#include "armnn/Tensor.hpp"
arovir01616e7752018-10-01 17:08:59 +01009#include "armnnUtils/Half.hpp"
telsoa014fcda012018-03-09 14:13:49 +000010
11#include <boost/multi_array.hpp>
12#include <boost/assert.hpp>
13#include <array>
14
telsoa01c577f2c2018-08-31 09:22:23 +010015// Layer callables.
telsoa014fcda012018-03-09 14:13:49 +000016
17namespace armnn
18{
19class IWorkloadFactory;
20}
21
22template <std::size_t n>
23boost::array<unsigned int, n> GetTensorShapeAsArray(const armnn::TensorInfo& tensorInfo)
24{
25 BOOST_ASSERT_MSG(n == tensorInfo.GetNumDimensions(),
26 "Attempting to construct a shape array of mismatching size");
27
28 boost::array<unsigned int, n> shape;
29 for (unsigned int i = 0; i < n; i++)
30 {
31 shape[i] = tensorInfo.GetShape()[i];
32 }
33 return shape;
34}
35
36template <typename T, std::size_t n>
37struct LayerTestResult
38{
39 LayerTestResult(const armnn::TensorInfo& outputInfo)
40 {
41 auto shape( GetTensorShapeAsArray<n>(outputInfo) );
42 output.resize(shape);
43 outputExpected.resize(shape);
44 supported = true;
45 }
46
47 boost::multi_array<T, n> output;
48 boost::multi_array<T, n> outputExpected;
49 bool supported;
50};
51
52LayerTestResult<float, 4> SimpleConvolution2d3x5Test(armnn::IWorkloadFactory& workloadFactory,
jimfly010a088a62018-10-25 17:05:05 +010053 bool biasEnabled,
54 const armnn::DataLayoutIndexed& layout);
telsoa014fcda012018-03-09 14:13:49 +000055
56LayerTestResult<float, 4> SimpleConvolution2d3x3Test(armnn::IWorkloadFactory& workloadFactory,
narpra015f703182018-10-26 16:24:58 +010057 bool biasEnabled,
58 const armnn::DataLayoutIndexed& layout);
Francis Murtaghd59116e2018-10-04 16:03:07 +010059
60LayerTestResult<float, 4> SimpleConvolution2d3x3NhwcTest(armnn::IWorkloadFactory& workloadFactory,
61 bool biasEnabled);
telsoa014fcda012018-03-09 14:13:49 +000062
63LayerTestResult<float, 4>
narpra015f703182018-10-26 16:24:58 +010064Convolution2dAsymmetricPaddingLargerThanHalfKernelSizeTest(armnn::IWorkloadFactory& workloadFactory,
65 const armnn::DataLayoutIndexed& layout);
66LayerTestResult<float, 4> Convolution2dAsymmetricPaddingTest(armnn::IWorkloadFactory& workloadFactory,
67 const armnn::DataLayoutIndexed& layout);
telsoa014fcda012018-03-09 14:13:49 +000068
69
narpra015f703182018-10-26 16:24:58 +010070LayerTestResult<float, 4> Convolution1dTest(armnn::IWorkloadFactory& workloadFactory,
71 bool biasEnabled);
72LayerTestResult<uint8_t, 4> Convolution1dUint8Test(armnn::IWorkloadFactory& workloadFactory,
73 bool biasEnabled);
telsoa014fcda012018-03-09 14:13:49 +000074
jimfly01d84216a2018-10-26 12:56:21 +010075LayerTestResult<float, 4> DepthwiseConvolution2dTest(armnn::IWorkloadFactory& workloadFactory,
76 bool biasEnabled,
77 const armnn::DataLayoutIndexed& layout);
telsoa014fcda012018-03-09 14:13:49 +000078
Nikhil Rajcec6b652018-10-12 13:51:57 +010079LayerTestResult<float, 4> DepthwiseConvolution2dDepthNhwcTest(armnn::IWorkloadFactory& workloadFactory,
80 bool biasEnabled);
81
telsoa014fcda012018-03-09 14:13:49 +000082LayerTestResult<float, 4> DepthwiseConvolution2dDepthMul1Test(armnn::IWorkloadFactory& workloadFactory,
jimfly01b9c89632018-10-26 16:50:13 +010083 bool biasEnabled,
84 const armnn::DataLayoutIndexed& layout);
telsoa014fcda012018-03-09 14:13:49 +000085
surmeh013537c2c2018-05-18 16:31:43 +010086LayerTestResult<float, 4> DepthwiseConvolution2dAsymmetricTest(armnn::IWorkloadFactory& workloadFactory,
jimfly01382a91d2018-10-26 15:55:50 +010087 bool biasEnabled,
88 const armnn::DataLayoutIndexed& layout);
surmeh013537c2c2018-05-18 16:31:43 +010089
telsoa014fcda012018-03-09 14:13:49 +000090LayerTestResult<float, 4> SimpleMaxPooling2dSize2x2Stride2x2Test(armnn::IWorkloadFactory& workloadFactory,
91 bool forceNoPadding);
92LayerTestResult<uint8_t, 4> SimpleMaxPooling2dSize2x2Stride2x2Uint8Test(armnn::IWorkloadFactory& workloadFactory,
93 bool forceNoPadding);
94LayerTestResult<float, 4> SimpleMaxPooling2dSize3x3Stride2x4Test(armnn::IWorkloadFactory& workloadFactory,
95 bool forceNoPadding);
96LayerTestResult<uint8_t, 4> SimpleMaxPooling2dSize3x3Stride2x4Uint8Test(armnn::IWorkloadFactory& workloadFactory,
97 bool forceNoPadding );
98LayerTestResult<float, 4> IgnorePaddingSimpleMaxPooling2dTest(armnn::IWorkloadFactory& workloadFactory);
99LayerTestResult<uint8_t, 4> IgnorePaddingSimpleMaxPooling2dUint8Test(armnn::IWorkloadFactory& workloadFactory);
100LayerTestResult<float, 4> IgnorePaddingMaxPooling2dSize3Test(armnn::IWorkloadFactory& workloadFactory);
101LayerTestResult<uint8_t, 4> IgnorePaddingMaxPooling2dSize3Uint8Test(armnn::IWorkloadFactory& workloadFactory);
102
James Conroy45a9b772018-10-31 11:47:53 +0000103LayerTestResult<float, 4> SimpleMaxPooling2dTest(armnn::IWorkloadFactory& workloadFactory,
104 const armnn::DataLayoutIndexed& dataLayout);
105LayerTestResult<uint8_t, 4> SimpleMaxPooling2dUint8Test(armnn::IWorkloadFactory& workloadFactory,
106 const armnn::DataLayoutIndexed& dataLayout);
107
108LayerTestResult<float, 4> SimpleAveragePooling2dTest(armnn::IWorkloadFactory& workloadFactory,
109 const armnn::DataLayoutIndexed& dataLayout);
110LayerTestResult<uint8_t, 4> SimpleAveragePooling2dUint8Test(armnn::IWorkloadFactory& workloadFactory,
111 const armnn::DataLayoutIndexed& dataLayout);
112
surmeh01bceff2f2018-03-29 16:29:27 +0100113LayerTestResult<float, 4> IgnorePaddingAveragePooling2dSize3x2Stride2x2Test(armnn::IWorkloadFactory& workloadFactory,
114 bool forceNoPadding);
telsoa014fcda012018-03-09 14:13:49 +0000115LayerTestResult<float, 4> IgnorePaddingSimpleAveragePooling2dTest(armnn::IWorkloadFactory& workloadFactory);
116LayerTestResult<uint8_t, 4> IgnorePaddingSimpleAveragePooling2dUint8Test(armnn::IWorkloadFactory& workloadFactory);
117LayerTestResult<float, 4> IgnorePaddingSimpleAveragePooling2dNoPaddingTest(armnn::IWorkloadFactory& workloadFactory);
118LayerTestResult<uint8_t, 4> IgnorePaddingSimpleAveragePooling2dNoPaddingUint8Test(
119 armnn::IWorkloadFactory& workloadFactory);
120LayerTestResult<float, 4> IgnorePaddingAveragePooling2dSize3Test(armnn::IWorkloadFactory& workloadFactory);
121LayerTestResult<uint8_t, 4> IgnorePaddingAveragePooling2dSize3Uint8Test(armnn::IWorkloadFactory& workloadFactory);
122
James Conroy45a9b772018-10-31 11:47:53 +0000123LayerTestResult<float, 4> SimpleL2Pooling2dTest(armnn::IWorkloadFactory& workloadFactory,
124 const armnn::DataLayoutIndexed& dataLayout);
125LayerTestResult<uint8_t, 4> SimpleL2Pooling2dUint8Test(armnn::IWorkloadFactory& workloadFactory,
126 const armnn::DataLayoutIndexed& dataLayout);
telsoa014fcda012018-03-09 14:13:49 +0000127
128LayerTestResult<float, 4> L2Pooling2dSize3Stride1Test(armnn::IWorkloadFactory& workloadFactory);
129LayerTestResult<uint8_t, 4> L2Pooling2dSize3Stride1Uint8Test(armnn::IWorkloadFactory& workloadFactory);
130LayerTestResult<float, 4> L2Pooling2dSize3Stride3Test(armnn::IWorkloadFactory& workloadFactory);
131LayerTestResult<uint8_t, 4> L2Pooling2dSize3Stride3Uint8Test(armnn::IWorkloadFactory& workloadFactory);
132LayerTestResult<float, 4> L2Pooling2dSize3Stride4Test(armnn::IWorkloadFactory& workloadFactory);
133LayerTestResult<uint8_t, 4> L2Pooling2dSize3Stride4Uint8Test(armnn::IWorkloadFactory& workloadFactory);
134LayerTestResult<float, 4> L2Pooling2dSize7Test(armnn::IWorkloadFactory& workloadFactory);
135LayerTestResult<uint8_t, 4> L2Pooling2dSize7Uint8Test(armnn::IWorkloadFactory& workloadFactory);
136LayerTestResult<float, 4> L2Pooling2dSize9Test(armnn::IWorkloadFactory& workloadFactory);
137LayerTestResult<uint8_t, 4> L2Pooling2dSize9Uint8Test(armnn::IWorkloadFactory& workloadFactory);
138LayerTestResult<float, 4> LargeTensorsAveragePooling2dTest(armnn::IWorkloadFactory& workloadFactory);
139LayerTestResult<uint8_t, 4> LargeTensorsAveragePooling2dUint8Test(armnn::IWorkloadFactory& workloadFactory);
140
141LayerTestResult<float, 4> IgnorePaddingSimpleL2Pooling2dTest(armnn::IWorkloadFactory& workloadFactory);
142LayerTestResult<uint8_t, 4> IgnorePaddingSimpleL2Pooling2dUint8Test(armnn::IWorkloadFactory& workloadFactory);
143LayerTestResult<float, 4> IgnorePaddingL2Pooling2dSize3Test(armnn::IWorkloadFactory& workloadFactory);
144LayerTestResult<uint8_t, 4> IgnorePaddingL2Pooling2dSize3Uint8Test(armnn::IWorkloadFactory& workloadFactory);
145
146LayerTestResult<float, 4> AsymmetricNonSquarePooling2dTest(armnn::IWorkloadFactory& workloadFactory);
147LayerTestResult<uint8_t, 4> AsymmetricNonSquarePooling2dUint8Test(armnn::IWorkloadFactory& workloadFactory);
148
149LayerTestResult<float, 4> ComparePooling2dTest(armnn::IWorkloadFactory& workloadFactory,
150 armnn::IWorkloadFactory& refWorkloadFactory,
151 armnn::PoolingAlgorithm poolingType);
152LayerTestResult<uint8_t, 4> ComparePooling2dUint8Test(armnn::IWorkloadFactory& workloadFactory,
153 armnn::IWorkloadFactory& refWorkloadFactory,
154 armnn::PoolingAlgorithm poolingType);
155
156LayerTestResult<float, 4> ConstantLinearActivationTest(armnn::IWorkloadFactory& workloadFactory);
157
158LayerTestResult<float, 4> SimpleNormalizationAcrossTest(armnn::IWorkloadFactory& workloadFactory);
159LayerTestResult<float, 4> SimpleNormalizationWithinTest(armnn::IWorkloadFactory& workloadFactory);
narpra0155a97bc2018-10-02 14:35:53 +0100160LayerTestResult<float,4> SimpleNormalizationAcrossNhwcTest(armnn::IWorkloadFactory& workloadFactory);
telsoa014fcda012018-03-09 14:13:49 +0000161
162LayerTestResult<float, 2> SimpleSoftmaxTest(armnn::IWorkloadFactory& workloadFactory, float beta);
163LayerTestResult<uint8_t, 2> SimpleSoftmaxUint8Test(armnn::IWorkloadFactory& workloadFactory, float beta);
164
165LayerTestResult<float, 4> SimpleSigmoidTest(armnn::IWorkloadFactory& workloadFactory);
166
167LayerTestResult<float, 4> SimpleReshapeFloat32Test(armnn::IWorkloadFactory& workloadFactory);
168LayerTestResult<uint8_t, 4> SimpleReshapeUint8Test(armnn::IWorkloadFactory& workloadFactory);
169
170LayerTestResult<float, 4> SimpleFloorTest(armnn::IWorkloadFactory& workloadFactory);
171
172LayerTestResult<float, 1> Concatenation1dTest(armnn::IWorkloadFactory& workloadFactory);
173LayerTestResult<float, 2> Concatenation2dDim0Test(armnn::IWorkloadFactory& workloadFactory);
174LayerTestResult<float, 2> Concatenation2dDim1Test(armnn::IWorkloadFactory& workloadFactory);
175LayerTestResult<float, 2> Concatenation2dDim0DiffInputDimsTest(armnn::IWorkloadFactory& workloadFactory);
176LayerTestResult<float, 2> Concatenation2dDim1DiffInputDimsTest(armnn::IWorkloadFactory& workloadFactory);
177LayerTestResult<float, 3> Concatenation3dDim0Test(armnn::IWorkloadFactory& workloadFactory);
178LayerTestResult<float, 3> Concatenation3dDim1Test(armnn::IWorkloadFactory& workloadFactory);
179LayerTestResult<float, 3> Concatenation3dDim2Test(armnn::IWorkloadFactory& workloadFactory);
180LayerTestResult<float, 3> Concatenation3dDim0DiffInputDimsTest(armnn::IWorkloadFactory& workloadFactory);
181LayerTestResult<float, 3> Concatenation3dDim1DiffInputDimsTest(armnn::IWorkloadFactory& workloadFactory);
182LayerTestResult<float, 3> Concatenation3dDim2DiffInputDimsTest(armnn::IWorkloadFactory& workloadFactory);
183
184LayerTestResult<uint8_t, 4> SimpleSigmoidUint8Test(armnn::IWorkloadFactory& workloadFactory);
185
186LayerTestResult<float, 4> CompareConvolution2dTest(armnn::IWorkloadFactory& workloadFactory,
187 armnn::IWorkloadFactory& refWorkloadFactory);
188
189template<typename T>
190LayerTestResult<T, 4> CompareDepthwiseConvolution2dTest(armnn::IWorkloadFactory& workloadFactory,
jimfly017af00da2018-10-31 14:43:53 +0000191 armnn::IWorkloadFactory& refWorkloadFactory,
192 const armnn::DataLayoutIndexed& layout);
telsoa014fcda012018-03-09 14:13:49 +0000193
194LayerTestResult<float, 4> CompareNormalizationTest(armnn::IWorkloadFactory& workloadFactory,
195 armnn::IWorkloadFactory& refWorkloadFactory,
196 armnn::NormalizationAlgorithmChannel normChannel,
197 armnn::NormalizationAlgorithmMethod normMethod);
198
199LayerTestResult<float, 2> CompareSoftmaxTest(armnn::IWorkloadFactory& workloadFactory,
200 armnn::IWorkloadFactory& refWorkloadFactory, float beta);
201
202LayerTestResult<float, 2> FullyConnectedFloat32Test(armnn::IWorkloadFactory& workloadFactory,
203 bool biasEnabled,
204 bool transposeWeights);
205
206std::vector<LayerTestResult<float, 3>> SplitterTest(armnn::IWorkloadFactory& workloadFactory);
207LayerTestResult<float, 3> CopyViaSplitterTest(armnn::IWorkloadFactory& workloadFactory);
208
209LayerTestResult<float, 3> MergerTest(armnn::IWorkloadFactory& workloadFactory);
210
211LayerTestResult<float, 4> AdditionTest(armnn::IWorkloadFactory& workloadFactory);
212LayerTestResult<float, 4> AdditionBroadcast1ElementTest(armnn::IWorkloadFactory& workloadFactory);
213LayerTestResult<float, 4> AdditionBroadcastTest(armnn::IWorkloadFactory& workloadFactory);
214
215LayerTestResult<float, 4> CompareAdditionTest(armnn::IWorkloadFactory& workloadFactory,
David Beckf195f032018-09-06 16:46:34 +0100216 armnn::IWorkloadFactory& refWorkloadFactory);
217
218LayerTestResult<float, 4> SubtractionTest(armnn::IWorkloadFactory& workloadFactory);
219LayerTestResult<float, 4> SubtractionBroadcast1ElementTest(armnn::IWorkloadFactory& workloadFactory);
220LayerTestResult<float, 4> SubtractionBroadcastTest(armnn::IWorkloadFactory& workloadFactory);
telsoa014fcda012018-03-09 14:13:49 +0000221
222LayerTestResult<float, 4> CompareActivationTest(armnn::IWorkloadFactory& workloadFactory,
223 armnn::IWorkloadFactory& refWorkloadFactory,
224 armnn::ActivationFunction f,
225 unsigned int batchSize);
226
Francis Murtaghe7a86a42018-08-29 12:42:10 +0100227LayerTestResult<float, 4> DivisionTest(armnn::IWorkloadFactory& workloadFactory);
Francis Murtagh8c5e3dc2018-08-30 17:18:37 +0100228LayerTestResult<float, 4> DivisionByZeroTest(armnn::IWorkloadFactory& workloadFactory);
Francis Murtaghe7a86a42018-08-29 12:42:10 +0100229LayerTestResult<float, 4> DivisionBroadcast1ElementTest(armnn::IWorkloadFactory& workloadFactory);
230LayerTestResult<float, 4> DivisionBroadcast1DVectorTest(armnn::IWorkloadFactory& workloadFactory);
231
telsoa014fcda012018-03-09 14:13:49 +0000232LayerTestResult<float, 4> MultiplicationTest(armnn::IWorkloadFactory& workloadFactory);
surmeh01bceff2f2018-03-29 16:29:27 +0100233LayerTestResult<float, 4> MultiplicationBroadcast1ElementTest(armnn::IWorkloadFactory& workloadFactory);
234LayerTestResult<float, 4> MultiplicationBroadcast1DVectorTest(armnn::IWorkloadFactory& workloadFactory);
telsoa014fcda012018-03-09 14:13:49 +0000235
236LayerTestResult<float, 4> CompareMultiplicationTest(armnn::IWorkloadFactory& workloadFactory,
237 armnn::IWorkloadFactory& refWorkloadFactory);
238
239LayerTestResult<float, 4> BatchNormTest(armnn::IWorkloadFactory& workloadFactory);
Matteo Martincigh8eb675e2018-10-17 14:43:29 +0100240LayerTestResult<float, 4> BatchNormNhwcTest(armnn::IWorkloadFactory& workloadFactory);
telsoa014fcda012018-03-09 14:13:49 +0000241
242LayerTestResult<float, 4> CompareBatchNormTest(armnn::IWorkloadFactory& workloadFactory,
243 armnn::IWorkloadFactory& refWorkloadFactory);
244
245LayerTestResult<float, 4> BoundedReLuUpperAndLowerBoundTest(armnn::IWorkloadFactory& workloadFactory);
246LayerTestResult<uint8_t, 4> BoundedReLuUint8UpperAndLowerBoundTest(armnn::IWorkloadFactory& workloadFactory);
247LayerTestResult<float, 4> BoundedReLuUpperBoundOnlyTest(armnn::IWorkloadFactory& workloadFactory);
248LayerTestResult<uint8_t, 4> BoundedReLuUint8UpperBoundOnlyTest(armnn::IWorkloadFactory& workloadFactory);
249
250LayerTestResult<float, 4> CompareBoundedReLuTest(armnn::IWorkloadFactory& workloadFactory,
251 armnn::IWorkloadFactory& refWorkloadFactory,
252 float upperBound,
253 float lowerBound);
254
telsoa01c577f2c2018-08-31 09:22:23 +0100255// Tests that the output should be identical to the input when the output dimensions match the input ones.
telsoa014fcda012018-03-09 14:13:49 +0000256LayerTestResult<float, 4> ResizeBilinearNopTest(armnn::IWorkloadFactory& workloadFactory);
257
telsoa01c577f2c2018-08-31 09:22:23 +0100258// Tests the behaviour of the resize bilinear operation when rescaling a 2x2 image into a 1x1 image.
telsoa014fcda012018-03-09 14:13:49 +0000259LayerTestResult<float, 4> SimpleResizeBilinearTest(armnn::IWorkloadFactory& workloadFactory);
260
telsoa01c577f2c2018-08-31 09:22:23 +0100261// Tests the resize bilinear for minification of a square input matrix (also: input dimensions are a
262// multiple of output dimensions).
telsoa014fcda012018-03-09 14:13:49 +0000263LayerTestResult<float, 4> ResizeBilinearSqMinTest(armnn::IWorkloadFactory& workloadFactory);
264
telsoa01c577f2c2018-08-31 09:22:23 +0100265// Tests the resize bilinear for minification (output dimensions smaller than input dimensions).
telsoa014fcda012018-03-09 14:13:49 +0000266LayerTestResult<float, 4> ResizeBilinearMinTest(armnn::IWorkloadFactory& workloadFactory);
267
telsoa01c577f2c2018-08-31 09:22:23 +0100268// Tests the resize bilinear for magnification (output dimensions bigger than input dimensions).
telsoa014fcda012018-03-09 14:13:49 +0000269LayerTestResult<float, 4> ResizeBilinearMagTest(armnn::IWorkloadFactory& workloadFactory);
270
James Conroy074f3712018-10-03 09:32:03 +0100271// Tests that execute Resize Bilinear with NHWC data layout
272LayerTestResult<float, 4> ResizeBilinearNopNhwcTest(armnn::IWorkloadFactory& workloadFactory);
273LayerTestResult<float, 4> SimpleResizeBilinearNhwcTest(armnn::IWorkloadFactory& workloadFactory);
274LayerTestResult<float, 4> ResizeBilinearSqMinNhwcTest(armnn::IWorkloadFactory& workloadFactory);
275LayerTestResult<float, 4> ResizeBilinearMinNhwcTest(armnn::IWorkloadFactory& workloadFactory);
276LayerTestResult<float, 4> ResizeBilinearMagNhwcTest(armnn::IWorkloadFactory& workloadFactory);
277
telsoa014fcda012018-03-09 14:13:49 +0000278LayerTestResult<float, 4> BatchNormTest(armnn::IWorkloadFactory& workloadFactory);
Nikhil Rajd1340932018-10-18 14:27:50 +0100279LayerTestResult<float, 4> BatchNormNhwcTest(armnn::IWorkloadFactory& workloadFactory);
telsoa014fcda012018-03-09 14:13:49 +0000280
281LayerTestResult<float, 2> FakeQuantizationTest(armnn::IWorkloadFactory& workloadFactory);
282
283LayerTestResult<float, 4> L2Normalization1dTest(armnn::IWorkloadFactory& workloadFactory);
284LayerTestResult<float, 4> L2Normalization2dTest(armnn::IWorkloadFactory& workloadFactory);
285LayerTestResult<float, 4> L2Normalization3dTest(armnn::IWorkloadFactory& workloadFactory);
286LayerTestResult<float, 4> L2Normalization4dTest(armnn::IWorkloadFactory& workloadFactory);
287
Matteo Martincigh539b44d2018-10-01 09:26:39 +0100288LayerTestResult<float, 4> L2Normalization1dNhwcTest(armnn::IWorkloadFactory& workloadFactory);
289LayerTestResult<float, 4> L2Normalization2dNhwcTest(armnn::IWorkloadFactory& workloadFactory);
290LayerTestResult<float, 4> L2Normalization3dNhwcTest(armnn::IWorkloadFactory& workloadFactory);
291LayerTestResult<float, 4> L2Normalization4dNhwcTest(armnn::IWorkloadFactory& workloadFactory);
292
telsoa014fcda012018-03-09 14:13:49 +0000293LayerTestResult<float, 4> ConstantTest(armnn::IWorkloadFactory& workloadFactory);
294
295LayerTestResult<uint8_t, 4> ConstantTestUint8(armnn::IWorkloadFactory& workloadFactory);
296
297LayerTestResult<uint8_t, 4> BoundedReLuUint8Test(armnn::IWorkloadFactory& workloadFactory, float upperBound);
298LayerTestResult<uint8_t, 4> BoundedReLuUint8Test(armnn::IWorkloadFactory& workloadFactory,
299 float upperBound,
300 float lowerBound);
301
302LayerTestResult<uint8_t, 2> FullyConnectedUint8Test(armnn::IWorkloadFactory& workloadFactory, bool biasEnabled);
303
304std::vector<LayerTestResult<uint8_t, 3>> SplitterUint8Test(armnn::IWorkloadFactory& workloadFactory);
305LayerTestResult<uint8_t, 3> CopyViaSplitterUint8Test(armnn::IWorkloadFactory& workloadFactory);
306
307LayerTestResult<uint8_t, 3> MergerUint8Test(armnn::IWorkloadFactory& workloadFactory);
308
309LayerTestResult<uint8_t, 4> AdditionUint8Test(armnn::IWorkloadFactory& workloadFactory);
310LayerTestResult<uint8_t, 4> AdditionBroadcast1ElementUint8Test(armnn::IWorkloadFactory& workloadFactory);
311LayerTestResult<uint8_t, 4> AdditionBroadcastUint8Test(armnn::IWorkloadFactory& workloadFactory);
312
David Beckf195f032018-09-06 16:46:34 +0100313LayerTestResult<uint8_t, 4> SubtractionUint8Test(armnn::IWorkloadFactory& workloadFactory);
314LayerTestResult<uint8_t, 4> SubtractionBroadcast1ElementUint8Test(armnn::IWorkloadFactory& workloadFactory);
315LayerTestResult<uint8_t, 4> SubtractionBroadcastUint8Test(armnn::IWorkloadFactory& workloadFactory);
316
telsoa014fcda012018-03-09 14:13:49 +0000317LayerTestResult<uint8_t, 4> CompareActivationUint8Test(armnn::IWorkloadFactory& workloadFactory,
318 armnn::IWorkloadFactory& refWorkloadFactory,
319 armnn::ActivationFunction f);
320
321LayerTestResult<uint8_t, 2> CompareSoftmaxUint8Test(armnn::IWorkloadFactory& workloadFactory,
322 armnn::IWorkloadFactory& refWorkloadFactory,
323 float beta);
324
325LayerTestResult<uint8_t, 4> MultiplicationUint8Test(armnn::IWorkloadFactory& workloadFactory);
surmeh01bceff2f2018-03-29 16:29:27 +0100326LayerTestResult<uint8_t, 4> MultiplicationBroadcast1ElementUint8Test(armnn::IWorkloadFactory& workloadFactory);
327LayerTestResult<uint8_t, 4> MultiplicationBroadcast1DVectorUint8Test(armnn::IWorkloadFactory& workloadFactory);
telsoa014fcda012018-03-09 14:13:49 +0000328
David Beck5cd01f32018-09-12 16:00:08 +0100329LayerTestResult<uint8_t, 4> DivisionUint8Test(armnn::IWorkloadFactory& workloadFactory);
330LayerTestResult<uint8_t, 4> DivisionBroadcast1ElementUint8Test(armnn::IWorkloadFactory& workloadFactory);
331LayerTestResult<uint8_t, 4> DivisionBroadcast1DVectorUint8Test(armnn::IWorkloadFactory& workloadFactory);
332
telsoa014fcda012018-03-09 14:13:49 +0000333LayerTestResult<uint8_t, 4> SimpleConvolution2d3x5Uint8Test(armnn::IWorkloadFactory& workloadFactory,
jimfly010a088a62018-10-25 17:05:05 +0100334 bool biasEnabled,
335 const armnn::DataLayoutIndexed& layout);
telsoa014fcda012018-03-09 14:13:49 +0000336
337LayerTestResult<uint8_t, 4> SimpleConvolution2d3x3Uint8Test(armnn::IWorkloadFactory& workloadFactory,
narpra015f703182018-10-26 16:24:58 +0100338 bool biasEnabled,
339 const armnn::DataLayoutIndexed& layout);
telsoa014fcda012018-03-09 14:13:49 +0000340
341LayerTestResult<uint8_t, 4> DepthwiseConvolution2dUint8Test(armnn::IWorkloadFactory& workloadFactory,
jimfly01d84216a2018-10-26 12:56:21 +0100342 bool biasEnabled,
343 const armnn::DataLayoutIndexed& layout);
telsoa014fcda012018-03-09 14:13:49 +0000344
345LayerTestResult<uint8_t, 4> DepthwiseConvolution2dDepthMul1Uint8Test(armnn::IWorkloadFactory& workloadFactory,
jimfly01b9c89632018-10-26 16:50:13 +0100346 bool biasEnabled,
347 const armnn::DataLayoutIndexed& layout);
telsoa014fcda012018-03-09 14:13:49 +0000348
349LayerTestResult<uint8_t, 4> ConstantLinearActivationUint8Test(armnn::IWorkloadFactory& workloadFactory);
350
351LayerTestResult<uint8_t, 4> ResizeBilinearNopUint8Test(armnn::IWorkloadFactory& workloadFactory);
352LayerTestResult<uint8_t, 4> SimpleResizeBilinearUint8Test(armnn::IWorkloadFactory& workloadFactory);
353LayerTestResult<uint8_t, 4> ResizeBilinearSqMinUint8Test(armnn::IWorkloadFactory& workloadFactory);
354LayerTestResult<uint8_t, 4> ResizeBilinearMinUint8Test(armnn::IWorkloadFactory& workloadFactory);
355LayerTestResult<uint8_t, 4> ResizeBilinearMagUint8Test(armnn::IWorkloadFactory& workloadFactory);
356
357LayerTestResult<uint8_t, 4> BatchNormUint8Test(armnn::IWorkloadFactory& workloadFactory);
Matteo Martincigh8eb675e2018-10-17 14:43:29 +0100358LayerTestResult<uint8_t, 4> BatchNormUint8NhwcTest(armnn::IWorkloadFactory& workloadFactory);
telsoa014fcda012018-03-09 14:13:49 +0000359
360LayerTestResult<uint8_t, 4> ConstantUint8Test(armnn::IWorkloadFactory& workloadFactory);
361
362LayerTestResult<uint8_t, 1> Concatenation1dUint8Test(armnn::IWorkloadFactory& workloadFactory);
363LayerTestResult<uint8_t, 2> Concatenation2dDim0Uint8Test(armnn::IWorkloadFactory& workloadFactory);
364LayerTestResult<uint8_t, 2> Concatenation2dDim1Uint8Test(armnn::IWorkloadFactory& workloadFactory);
365LayerTestResult<uint8_t, 2> Concatenation2dDim0DiffInputDimsUint8Test(armnn::IWorkloadFactory& workloadFactory);
366LayerTestResult<uint8_t, 2> Concatenation2dDim1DiffInputDimsUint8Test(armnn::IWorkloadFactory& workloadFactory);
367LayerTestResult<uint8_t, 3> Concatenation3dDim0Uint8Test(armnn::IWorkloadFactory& workloadFactory);
368LayerTestResult<uint8_t, 3> Concatenation3dDim1Uint8Test(armnn::IWorkloadFactory& workloadFactory);
369LayerTestResult<uint8_t, 3> Concatenation3dDim2Uint8Test(armnn::IWorkloadFactory& workloadFactory);
370LayerTestResult<uint8_t, 3> Concatenation3dDim0DiffInputDimsUint8Test(armnn::IWorkloadFactory& workloadFactory);
371LayerTestResult<uint8_t, 3> Concatenation3dDim1DiffInputDimsUint8Test(armnn::IWorkloadFactory& workloadFactory);
372LayerTestResult<uint8_t, 3> Concatenation3dDim2DiffInputDimsUint8Test(armnn::IWorkloadFactory& workloadFactory);
373
374
375LayerTestResult<float, 2> FullyConnectedLargeTest(armnn::IWorkloadFactory& workloadFactory,
376 bool transposeWeights);
377LayerTestResult<float, 4> SimplePermuteFloat32Test(armnn::IWorkloadFactory& workloadFactory);
378LayerTestResult<uint8_t, 4> SimplePermuteUint8Test(armnn::IWorkloadFactory& workloadFactory);
379
Mohamed Nour Abouelseouddd6acea2018-10-18 12:26:19 +0100380LayerTestResult<uint8_t, 2> PadUint82dTest(armnn::IWorkloadFactory& workloadFactory);
381LayerTestResult<uint8_t, 3> PadUint83dTest(armnn::IWorkloadFactory& workloadFactory);
382LayerTestResult<uint8_t, 4> PadUint84dTest(armnn::IWorkloadFactory& workloadFactory);
Mohamed Nour Abouelseoud7420e552018-10-12 12:26:24 +0100383
Mohamed Nour Abouelseouddd6acea2018-10-18 12:26:19 +0100384LayerTestResult<float, 2> PadFloat322dTest(armnn::IWorkloadFactory& workloadFactory);
385LayerTestResult<float, 3> PadFloat323dTest(armnn::IWorkloadFactory& workloadFactory);
386LayerTestResult<float, 4> PadFloat324dTest(armnn::IWorkloadFactory& workloadFactory);
Mohamed Nour Abouelseoud7420e552018-10-12 12:26:24 +0100387
surmeh01bceff2f2018-03-29 16:29:27 +0100388LayerTestResult<float, 4> PermuteFloat32ValueSet1Test(armnn::IWorkloadFactory& workloadFactory);
389LayerTestResult<float, 4> PermuteFloat32ValueSet2Test(armnn::IWorkloadFactory& workloadFactory);
390LayerTestResult<float, 4> PermuteFloat32ValueSet3Test(armnn::IWorkloadFactory& workloadFactory);
telsoa01c577f2c2018-08-31 09:22:23 +0100391
392LayerTestResult<float, 2> LstmLayerFloat32WithCifgWithPeepholeNoProjectionTest
393 (armnn::IWorkloadFactory& workloadFactory);
394LayerTestResult<float, 2>
395 LstmLayerFloat32NoCifgNoPeepholeNoProjectionTest(armnn::IWorkloadFactory& workloadFactory);
396LayerTestResult<float, 2>
397LstmLayerFloat32NoCifgWithPeepholeWithProjectionTest(armnn::IWorkloadFactory& workloadFactory);
398
399LayerTestResult<float, 4> SimpleConvertFp16ToFp32Test(armnn::IWorkloadFactory& workloadFactory);
400LayerTestResult<armnn::Half, 4> SimpleConvertFp32ToFp16Test(armnn::IWorkloadFactory& workloadFactory);
narpra011e4c31d2018-09-28 11:07:51 +0100401
402
403LayerTestResult<uint8_t, 1> MeanUint8SimpleTest(armnn::IWorkloadFactory& workloadFactory);
404LayerTestResult<uint8_t, 3> MeanUint8SimpleAxisTest(armnn::IWorkloadFactory& workloadFactory);
405LayerTestResult<uint8_t, 4> MeanUint8KeepDimsTest(armnn::IWorkloadFactory& workloadFactory);
406LayerTestResult<uint8_t, 4> MeanUint8MultipleDimsTest(armnn::IWorkloadFactory& workloadFactory);
407LayerTestResult<uint8_t, 1> MeanVtsUint8Test(armnn::IWorkloadFactory& workloadFactory);
408LayerTestResult<float, 1> MeanFloatSimpleTest(armnn::IWorkloadFactory& workloadFactory);
409LayerTestResult<float, 3> MeanFloatSimpleAxisTest(armnn::IWorkloadFactory& workloadFactory);
410LayerTestResult<float, 4> MeanFloatKeepDimsTest(armnn::IWorkloadFactory& workloadFactory);
411LayerTestResult<float, 4> MeanFloatMultipleDimsTest(armnn::IWorkloadFactory& workloadFactory);
412LayerTestResult<float, 1> MeanVtsFloat1Test(armnn::IWorkloadFactory& workloadFactory);
413LayerTestResult<float, 3> MeanVtsFloat2Test(armnn::IWorkloadFactory& workloadFactory);
Matteo Martincigh28dcab62018-10-19 16:40:03 +0100414LayerTestResult<float, 3> MeanVtsFloat3Test(armnn::IWorkloadFactory& workloadFactory);
415
Matteo Martincigh8eb675e2018-10-17 14:43:29 +0100416LayerTestResult<float, 4> AdditionAfterMaxPoolTest(armnn::IWorkloadFactory& workloadFactory);