Gitiles
Code Review Sign In
review.mlplatform.org / ml / armnn / 1b417c4970462d08e8e80d2f2462230ca5b91515 / . / src / backends / test / LayerTests.hpp
blob: 392d3bf34e9e31defa376ae49155e691d7683b96 [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//
5#pragma once
6
7#include "armnn/ArmNN.hpp"
8#include "armnn/Tensor.hpp"
arovir01616e7752018-10-01 17:08:59 +0100[diff] [blame]9#include "armnnUtils/Half.hpp"
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]10
11#include <boost/multi_array.hpp>
12#include <boost/assert.hpp>
13#include <array>
14
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]15// Layer callables.
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]16
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 +0100[diff] [blame]53 bool biasEnabled,
54 const armnn::DataLayoutIndexed& layout);
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]55
56LayerTestResult<float, 4> SimpleConvolution2d3x3Test(armnn::IWorkloadFactory& workloadFactory,
narpra015f703182018-10-26 16:24:58 +0100[diff] [blame]57 bool biasEnabled,
58 const armnn::DataLayoutIndexed& layout);
Francis Murtaghd59116e2018-10-04 16:03:07 +0100[diff] [blame]59
60LayerTestResult<float, 4> SimpleConvolution2d3x3NhwcTest(armnn::IWorkloadFactory& workloadFactory,
61 bool biasEnabled);
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]62
63LayerTestResult<float, 4>
narpra015f703182018-10-26 16:24:58 +0100[diff] [blame]64Convolution2dAsymmetricPaddingLargerThanHalfKernelSizeTest(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 +0000[diff] [blame]68
69
narpra015f703182018-10-26 16:24:58 +0100[diff] [blame]70LayerTestResult<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 +0000[diff] [blame]74
jimfly01d84216a2018-10-26 12:56:21 +0100[diff] [blame]75LayerTestResult<float, 4> DepthwiseConvolution2dTest(armnn::IWorkloadFactory& workloadFactory,
76 bool biasEnabled,
77 const armnn::DataLayoutIndexed& layout);
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]78
Nikhil Rajcec6b652018-10-12 13:51:57 +0100[diff] [blame]79LayerTestResult<float, 4> DepthwiseConvolution2dDepthNhwcTest(armnn::IWorkloadFactory& workloadFactory,
80 bool biasEnabled);
81
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]82LayerTestResult<float, 4> DepthwiseConvolution2dDepthMul1Test(armnn::IWorkloadFactory& workloadFactory,
83 bool biasEnabled);
84
surmeh013537c2c2018-05-18 16:31:43 +0100[diff] [blame]85LayerTestResult<float, 4> DepthwiseConvolution2dAsymmetricTest(armnn::IWorkloadFactory& workloadFactory,
jimfly01382a91d2018-10-26 15:55:50 +0100[diff] [blame]86 bool biasEnabled,
87 const armnn::DataLayoutIndexed& layout);
surmeh013537c2c2018-05-18 16:31:43 +0100[diff] [blame]88
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]89LayerTestResult<float, 4> SimpleMaxPooling2dSize2x2Stride2x2Test(armnn::IWorkloadFactory& workloadFactory,
90 bool forceNoPadding);
91LayerTestResult<uint8_t, 4> SimpleMaxPooling2dSize2x2Stride2x2Uint8Test(armnn::IWorkloadFactory& workloadFactory,
92 bool forceNoPadding);
93LayerTestResult<float, 4> SimpleMaxPooling2dSize3x3Stride2x4Test(armnn::IWorkloadFactory& workloadFactory,
94 bool forceNoPadding);
95LayerTestResult<uint8_t, 4> SimpleMaxPooling2dSize3x3Stride2x4Uint8Test(armnn::IWorkloadFactory& workloadFactory,
96 bool forceNoPadding );
97LayerTestResult<float, 4> IgnorePaddingSimpleMaxPooling2dTest(armnn::IWorkloadFactory& workloadFactory);
98LayerTestResult<uint8_t, 4> IgnorePaddingSimpleMaxPooling2dUint8Test(armnn::IWorkloadFactory& workloadFactory);
99LayerTestResult<float, 4> IgnorePaddingMaxPooling2dSize3Test(armnn::IWorkloadFactory& workloadFactory);
100LayerTestResult<uint8_t, 4> IgnorePaddingMaxPooling2dSize3Uint8Test(armnn::IWorkloadFactory& workloadFactory);
101
James Conroy45a9b772018-10-31 11:47:53 +0000[diff] [blame]102LayerTestResult<float, 4> SimpleMaxPooling2dTest(armnn::IWorkloadFactory& workloadFactory,
103 const armnn::DataLayoutIndexed& dataLayout);
104LayerTestResult<uint8_t, 4> SimpleMaxPooling2dUint8Test(armnn::IWorkloadFactory& workloadFactory,
105 const armnn::DataLayoutIndexed& dataLayout);
106
107LayerTestResult<float, 4> SimpleAveragePooling2dTest(armnn::IWorkloadFactory& workloadFactory,
108 const armnn::DataLayoutIndexed& dataLayout);
109LayerTestResult<uint8_t, 4> SimpleAveragePooling2dUint8Test(armnn::IWorkloadFactory& workloadFactory,
110 const armnn::DataLayoutIndexed& dataLayout);
111
surmeh01bceff2f2018-03-29 16:29:27 +0100[diff] [blame]112LayerTestResult<float, 4> IgnorePaddingAveragePooling2dSize3x2Stride2x2Test(armnn::IWorkloadFactory& workloadFactory,
113 bool forceNoPadding);
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]114LayerTestResult<float, 4> IgnorePaddingSimpleAveragePooling2dTest(armnn::IWorkloadFactory& workloadFactory);
115LayerTestResult<uint8_t, 4> IgnorePaddingSimpleAveragePooling2dUint8Test(armnn::IWorkloadFactory& workloadFactory);
116LayerTestResult<float, 4> IgnorePaddingSimpleAveragePooling2dNoPaddingTest(armnn::IWorkloadFactory& workloadFactory);
117LayerTestResult<uint8_t, 4> IgnorePaddingSimpleAveragePooling2dNoPaddingUint8Test(
118 armnn::IWorkloadFactory& workloadFactory);
119LayerTestResult<float, 4> IgnorePaddingAveragePooling2dSize3Test(armnn::IWorkloadFactory& workloadFactory);
120LayerTestResult<uint8_t, 4> IgnorePaddingAveragePooling2dSize3Uint8Test(armnn::IWorkloadFactory& workloadFactory);
121
James Conroy45a9b772018-10-31 11:47:53 +0000[diff] [blame]122LayerTestResult<float, 4> SimpleL2Pooling2dTest(armnn::IWorkloadFactory& workloadFactory,
123 const armnn::DataLayoutIndexed& dataLayout);
124LayerTestResult<uint8_t, 4> SimpleL2Pooling2dUint8Test(armnn::IWorkloadFactory& workloadFactory,
125 const armnn::DataLayoutIndexed& dataLayout);
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]126
127LayerTestResult<float, 4> L2Pooling2dSize3Stride1Test(armnn::IWorkloadFactory& workloadFactory);
128LayerTestResult<uint8_t, 4> L2Pooling2dSize3Stride1Uint8Test(armnn::IWorkloadFactory& workloadFactory);
129LayerTestResult<float, 4> L2Pooling2dSize3Stride3Test(armnn::IWorkloadFactory& workloadFactory);
130LayerTestResult<uint8_t, 4> L2Pooling2dSize3Stride3Uint8Test(armnn::IWorkloadFactory& workloadFactory);
131LayerTestResult<float, 4> L2Pooling2dSize3Stride4Test(armnn::IWorkloadFactory& workloadFactory);
132LayerTestResult<uint8_t, 4> L2Pooling2dSize3Stride4Uint8Test(armnn::IWorkloadFactory& workloadFactory);
133LayerTestResult<float, 4> L2Pooling2dSize7Test(armnn::IWorkloadFactory& workloadFactory);
134LayerTestResult<uint8_t, 4> L2Pooling2dSize7Uint8Test(armnn::IWorkloadFactory& workloadFactory);
135LayerTestResult<float, 4> L2Pooling2dSize9Test(armnn::IWorkloadFactory& workloadFactory);
136LayerTestResult<uint8_t, 4> L2Pooling2dSize9Uint8Test(armnn::IWorkloadFactory& workloadFactory);
137LayerTestResult<float, 4> LargeTensorsAveragePooling2dTest(armnn::IWorkloadFactory& workloadFactory);
138LayerTestResult<uint8_t, 4> LargeTensorsAveragePooling2dUint8Test(armnn::IWorkloadFactory& workloadFactory);
139
140LayerTestResult<float, 4> IgnorePaddingSimpleL2Pooling2dTest(armnn::IWorkloadFactory& workloadFactory);
141LayerTestResult<uint8_t, 4> IgnorePaddingSimpleL2Pooling2dUint8Test(armnn::IWorkloadFactory& workloadFactory);
142LayerTestResult<float, 4> IgnorePaddingL2Pooling2dSize3Test(armnn::IWorkloadFactory& workloadFactory);
143LayerTestResult<uint8_t, 4> IgnorePaddingL2Pooling2dSize3Uint8Test(armnn::IWorkloadFactory& workloadFactory);
144
145LayerTestResult<float, 4> AsymmetricNonSquarePooling2dTest(armnn::IWorkloadFactory& workloadFactory);
146LayerTestResult<uint8_t, 4> AsymmetricNonSquarePooling2dUint8Test(armnn::IWorkloadFactory& workloadFactory);
147
148LayerTestResult<float, 4> ComparePooling2dTest(armnn::IWorkloadFactory& workloadFactory,
149 armnn::IWorkloadFactory& refWorkloadFactory,
150 armnn::PoolingAlgorithm poolingType);
151LayerTestResult<uint8_t, 4> ComparePooling2dUint8Test(armnn::IWorkloadFactory& workloadFactory,
152 armnn::IWorkloadFactory& refWorkloadFactory,
153 armnn::PoolingAlgorithm poolingType);
154
155LayerTestResult<float, 4> ConstantLinearActivationTest(armnn::IWorkloadFactory& workloadFactory);
156
157LayerTestResult<float, 4> SimpleNormalizationAcrossTest(armnn::IWorkloadFactory& workloadFactory);
158LayerTestResult<float, 4> SimpleNormalizationWithinTest(armnn::IWorkloadFactory& workloadFactory);
narpra0155a97bc2018-10-02 14:35:53 +0100[diff] [blame]159LayerTestResult<float,4> SimpleNormalizationAcrossNhwcTest(armnn::IWorkloadFactory& workloadFactory);
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]160
161LayerTestResult<float, 2> SimpleSoftmaxTest(armnn::IWorkloadFactory& workloadFactory, float beta);
162LayerTestResult<uint8_t, 2> SimpleSoftmaxUint8Test(armnn::IWorkloadFactory& workloadFactory, float beta);
163
164LayerTestResult<float, 4> SimpleSigmoidTest(armnn::IWorkloadFactory& workloadFactory);
165
166LayerTestResult<float, 4> SimpleReshapeFloat32Test(armnn::IWorkloadFactory& workloadFactory);
167LayerTestResult<uint8_t, 4> SimpleReshapeUint8Test(armnn::IWorkloadFactory& workloadFactory);
168
169LayerTestResult<float, 4> SimpleFloorTest(armnn::IWorkloadFactory& workloadFactory);
170
171LayerTestResult<float, 1> Concatenation1dTest(armnn::IWorkloadFactory& workloadFactory);
172LayerTestResult<float, 2> Concatenation2dDim0Test(armnn::IWorkloadFactory& workloadFactory);
173LayerTestResult<float, 2> Concatenation2dDim1Test(armnn::IWorkloadFactory& workloadFactory);
174LayerTestResult<float, 2> Concatenation2dDim0DiffInputDimsTest(armnn::IWorkloadFactory& workloadFactory);
175LayerTestResult<float, 2> Concatenation2dDim1DiffInputDimsTest(armnn::IWorkloadFactory& workloadFactory);
176LayerTestResult<float, 3> Concatenation3dDim0Test(armnn::IWorkloadFactory& workloadFactory);
177LayerTestResult<float, 3> Concatenation3dDim1Test(armnn::IWorkloadFactory& workloadFactory);
178LayerTestResult<float, 3> Concatenation3dDim2Test(armnn::IWorkloadFactory& workloadFactory);
179LayerTestResult<float, 3> Concatenation3dDim0DiffInputDimsTest(armnn::IWorkloadFactory& workloadFactory);
180LayerTestResult<float, 3> Concatenation3dDim1DiffInputDimsTest(armnn::IWorkloadFactory& workloadFactory);
181LayerTestResult<float, 3> Concatenation3dDim2DiffInputDimsTest(armnn::IWorkloadFactory& workloadFactory);
182
183LayerTestResult<uint8_t, 4> SimpleSigmoidUint8Test(armnn::IWorkloadFactory& workloadFactory);
184
185LayerTestResult<float, 4> CompareConvolution2dTest(armnn::IWorkloadFactory& workloadFactory,
186 armnn::IWorkloadFactory& refWorkloadFactory);
187
188template<typename T>
189LayerTestResult<T, 4> CompareDepthwiseConvolution2dTest(armnn::IWorkloadFactory& workloadFactory,
190 armnn::IWorkloadFactory& refWorkloadFactory);
191
192LayerTestResult<float, 4> CompareNormalizationTest(armnn::IWorkloadFactory& workloadFactory,
193 armnn::IWorkloadFactory& refWorkloadFactory,
194 armnn::NormalizationAlgorithmChannel normChannel,
195 armnn::NormalizationAlgorithmMethod normMethod);
196
197LayerTestResult<float, 2> CompareSoftmaxTest(armnn::IWorkloadFactory& workloadFactory,
198 armnn::IWorkloadFactory& refWorkloadFactory, float beta);
199
200LayerTestResult<float, 2> FullyConnectedFloat32Test(armnn::IWorkloadFactory& workloadFactory,
201 bool biasEnabled,
202 bool transposeWeights);
203
204std::vector<LayerTestResult<float, 3>> SplitterTest(armnn::IWorkloadFactory& workloadFactory);
205LayerTestResult<float, 3> CopyViaSplitterTest(armnn::IWorkloadFactory& workloadFactory);
206
207LayerTestResult<float, 3> MergerTest(armnn::IWorkloadFactory& workloadFactory);
208
209LayerTestResult<float, 4> AdditionTest(armnn::IWorkloadFactory& workloadFactory);
210LayerTestResult<float, 4> AdditionBroadcast1ElementTest(armnn::IWorkloadFactory& workloadFactory);
211LayerTestResult<float, 4> AdditionBroadcastTest(armnn::IWorkloadFactory& workloadFactory);
212
213LayerTestResult<float, 4> CompareAdditionTest(armnn::IWorkloadFactory& workloadFactory,
David Beckf195f032018-09-06 16:46:34 +0100[diff] [blame]214 armnn::IWorkloadFactory& refWorkloadFactory);
215
216LayerTestResult<float, 4> SubtractionTest(armnn::IWorkloadFactory& workloadFactory);
217LayerTestResult<float, 4> SubtractionBroadcast1ElementTest(armnn::IWorkloadFactory& workloadFactory);
218LayerTestResult<float, 4> SubtractionBroadcastTest(armnn::IWorkloadFactory& workloadFactory);
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]219
220LayerTestResult<float, 4> CompareActivationTest(armnn::IWorkloadFactory& workloadFactory,
221 armnn::IWorkloadFactory& refWorkloadFactory,
222 armnn::ActivationFunction f,
223 unsigned int batchSize);
224
Francis Murtaghe7a86a42018-08-29 12:42:10 +0100[diff] [blame]225LayerTestResult<float, 4> DivisionTest(armnn::IWorkloadFactory& workloadFactory);
Francis Murtagh8c5e3dc2018-08-30 17:18:37 +0100[diff] [blame]226LayerTestResult<float, 4> DivisionByZeroTest(armnn::IWorkloadFactory& workloadFactory);
Francis Murtaghe7a86a42018-08-29 12:42:10 +0100[diff] [blame]227LayerTestResult<float, 4> DivisionBroadcast1ElementTest(armnn::IWorkloadFactory& workloadFactory);
228LayerTestResult<float, 4> DivisionBroadcast1DVectorTest(armnn::IWorkloadFactory& workloadFactory);
229
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]230LayerTestResult<float, 4> MultiplicationTest(armnn::IWorkloadFactory& workloadFactory);
surmeh01bceff2f2018-03-29 16:29:27 +0100[diff] [blame]231LayerTestResult<float, 4> MultiplicationBroadcast1ElementTest(armnn::IWorkloadFactory& workloadFactory);
232LayerTestResult<float, 4> MultiplicationBroadcast1DVectorTest(armnn::IWorkloadFactory& workloadFactory);
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]233
234LayerTestResult<float, 4> CompareMultiplicationTest(armnn::IWorkloadFactory& workloadFactory,
235 armnn::IWorkloadFactory& refWorkloadFactory);
236
237LayerTestResult<float, 4> BatchNormTest(armnn::IWorkloadFactory& workloadFactory);
Matteo Martincigh8eb675e2018-10-17 14:43:29 +0100[diff] [blame]238LayerTestResult<float, 4> BatchNormNhwcTest(armnn::IWorkloadFactory& workloadFactory);
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]239
240LayerTestResult<float, 4> CompareBatchNormTest(armnn::IWorkloadFactory& workloadFactory,
241 armnn::IWorkloadFactory& refWorkloadFactory);
242
243LayerTestResult<float, 4> BoundedReLuUpperAndLowerBoundTest(armnn::IWorkloadFactory& workloadFactory);
244LayerTestResult<uint8_t, 4> BoundedReLuUint8UpperAndLowerBoundTest(armnn::IWorkloadFactory& workloadFactory);
245LayerTestResult<float, 4> BoundedReLuUpperBoundOnlyTest(armnn::IWorkloadFactory& workloadFactory);
246LayerTestResult<uint8_t, 4> BoundedReLuUint8UpperBoundOnlyTest(armnn::IWorkloadFactory& workloadFactory);
247
248LayerTestResult<float, 4> CompareBoundedReLuTest(armnn::IWorkloadFactory& workloadFactory,
249 armnn::IWorkloadFactory& refWorkloadFactory,
250 float upperBound,
251 float lowerBound);
252
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]253// Tests that the output should be identical to the input when the output dimensions match the input ones.
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]254LayerTestResult<float, 4> ResizeBilinearNopTest(armnn::IWorkloadFactory& workloadFactory);
255
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]256// Tests the behaviour of the resize bilinear operation when rescaling a 2x2 image into a 1x1 image.
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]257LayerTestResult<float, 4> SimpleResizeBilinearTest(armnn::IWorkloadFactory& workloadFactory);
258
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]259// Tests the resize bilinear for minification of a square input matrix (also: input dimensions are a
260// multiple of output dimensions).
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]261LayerTestResult<float, 4> ResizeBilinearSqMinTest(armnn::IWorkloadFactory& workloadFactory);
262
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]263// Tests the resize bilinear for minification (output dimensions smaller than input dimensions).
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]264LayerTestResult<float, 4> ResizeBilinearMinTest(armnn::IWorkloadFactory& workloadFactory);
265
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]266// Tests the resize bilinear for magnification (output dimensions bigger than input dimensions).
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]267LayerTestResult<float, 4> ResizeBilinearMagTest(armnn::IWorkloadFactory& workloadFactory);
268
James Conroy074f3712018-10-03 09:32:03 +0100[diff] [blame]269// Tests that execute Resize Bilinear with NHWC data layout
270LayerTestResult<float, 4> ResizeBilinearNopNhwcTest(armnn::IWorkloadFactory& workloadFactory);
271LayerTestResult<float, 4> SimpleResizeBilinearNhwcTest(armnn::IWorkloadFactory& workloadFactory);
272LayerTestResult<float, 4> ResizeBilinearSqMinNhwcTest(armnn::IWorkloadFactory& workloadFactory);
273LayerTestResult<float, 4> ResizeBilinearMinNhwcTest(armnn::IWorkloadFactory& workloadFactory);
274LayerTestResult<float, 4> ResizeBilinearMagNhwcTest(armnn::IWorkloadFactory& workloadFactory);
275
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]276LayerTestResult<float, 4> BatchNormTest(armnn::IWorkloadFactory& workloadFactory);
Nikhil Rajd1340932018-10-18 14:27:50 +0100[diff] [blame]277LayerTestResult<float, 4> BatchNormNhwcTest(armnn::IWorkloadFactory& workloadFactory);
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]278
279LayerTestResult<float, 2> FakeQuantizationTest(armnn::IWorkloadFactory& workloadFactory);
280
281LayerTestResult<float, 4> L2Normalization1dTest(armnn::IWorkloadFactory& workloadFactory);
282LayerTestResult<float, 4> L2Normalization2dTest(armnn::IWorkloadFactory& workloadFactory);
283LayerTestResult<float, 4> L2Normalization3dTest(armnn::IWorkloadFactory& workloadFactory);
284LayerTestResult<float, 4> L2Normalization4dTest(armnn::IWorkloadFactory& workloadFactory);
285
Matteo Martincigh539b44d2018-10-01 09:26:39 +0100[diff] [blame]286LayerTestResult<float, 4> L2Normalization1dNhwcTest(armnn::IWorkloadFactory& workloadFactory);
287LayerTestResult<float, 4> L2Normalization2dNhwcTest(armnn::IWorkloadFactory& workloadFactory);
288LayerTestResult<float, 4> L2Normalization3dNhwcTest(armnn::IWorkloadFactory& workloadFactory);
289LayerTestResult<float, 4> L2Normalization4dNhwcTest(armnn::IWorkloadFactory& workloadFactory);
290
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]291LayerTestResult<float, 4> ConstantTest(armnn::IWorkloadFactory& workloadFactory);
292
293LayerTestResult<uint8_t, 4> ConstantTestUint8(armnn::IWorkloadFactory& workloadFactory);
294
295LayerTestResult<uint8_t, 4> BoundedReLuUint8Test(armnn::IWorkloadFactory& workloadFactory, float upperBound);
296LayerTestResult<uint8_t, 4> BoundedReLuUint8Test(armnn::IWorkloadFactory& workloadFactory,
297 float upperBound,
298 float lowerBound);
299
300LayerTestResult<uint8_t, 2> FullyConnectedUint8Test(armnn::IWorkloadFactory& workloadFactory, bool biasEnabled);
301
302std::vector<LayerTestResult<uint8_t, 3>> SplitterUint8Test(armnn::IWorkloadFactory& workloadFactory);
303LayerTestResult<uint8_t, 3> CopyViaSplitterUint8Test(armnn::IWorkloadFactory& workloadFactory);
304
305LayerTestResult<uint8_t, 3> MergerUint8Test(armnn::IWorkloadFactory& workloadFactory);
306
307LayerTestResult<uint8_t, 4> AdditionUint8Test(armnn::IWorkloadFactory& workloadFactory);
308LayerTestResult<uint8_t, 4> AdditionBroadcast1ElementUint8Test(armnn::IWorkloadFactory& workloadFactory);
309LayerTestResult<uint8_t, 4> AdditionBroadcastUint8Test(armnn::IWorkloadFactory& workloadFactory);
310
David Beckf195f032018-09-06 16:46:34 +0100[diff] [blame]311LayerTestResult<uint8_t, 4> SubtractionUint8Test(armnn::IWorkloadFactory& workloadFactory);
312LayerTestResult<uint8_t, 4> SubtractionBroadcast1ElementUint8Test(armnn::IWorkloadFactory& workloadFactory);
313LayerTestResult<uint8_t, 4> SubtractionBroadcastUint8Test(armnn::IWorkloadFactory& workloadFactory);
314
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]315LayerTestResult<uint8_t, 4> CompareActivationUint8Test(armnn::IWorkloadFactory& workloadFactory,
316 armnn::IWorkloadFactory& refWorkloadFactory,
317 armnn::ActivationFunction f);
318
319LayerTestResult<uint8_t, 2> CompareSoftmaxUint8Test(armnn::IWorkloadFactory& workloadFactory,
320 armnn::IWorkloadFactory& refWorkloadFactory,
321 float beta);
322
323LayerTestResult<uint8_t, 4> MultiplicationUint8Test(armnn::IWorkloadFactory& workloadFactory);
surmeh01bceff2f2018-03-29 16:29:27 +0100[diff] [blame]324LayerTestResult<uint8_t, 4> MultiplicationBroadcast1ElementUint8Test(armnn::IWorkloadFactory& workloadFactory);
325LayerTestResult<uint8_t, 4> MultiplicationBroadcast1DVectorUint8Test(armnn::IWorkloadFactory& workloadFactory);
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]326
David Beck5cd01f32018-09-12 16:00:08 +0100[diff] [blame]327LayerTestResult<uint8_t, 4> DivisionUint8Test(armnn::IWorkloadFactory& workloadFactory);
328LayerTestResult<uint8_t, 4> DivisionBroadcast1ElementUint8Test(armnn::IWorkloadFactory& workloadFactory);
329LayerTestResult<uint8_t, 4> DivisionBroadcast1DVectorUint8Test(armnn::IWorkloadFactory& workloadFactory);
330
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]331LayerTestResult<uint8_t, 4> SimpleConvolution2d3x5Uint8Test(armnn::IWorkloadFactory& workloadFactory,
jimfly010a088a62018-10-25 17:05:05 +0100[diff] [blame]332 bool biasEnabled,
333 const armnn::DataLayoutIndexed& layout);
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]334
335LayerTestResult<uint8_t, 4> SimpleConvolution2d3x3Uint8Test(armnn::IWorkloadFactory& workloadFactory,
narpra015f703182018-10-26 16:24:58 +0100[diff] [blame]336 bool biasEnabled,
337 const armnn::DataLayoutIndexed& layout);
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]338
339LayerTestResult<uint8_t, 4> DepthwiseConvolution2dUint8Test(armnn::IWorkloadFactory& workloadFactory,
jimfly01d84216a2018-10-26 12:56:21 +0100[diff] [blame]340 bool biasEnabled,
341 const armnn::DataLayoutIndexed& layout);
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]342
343LayerTestResult<uint8_t, 4> DepthwiseConvolution2dDepthMul1Uint8Test(armnn::IWorkloadFactory& workloadFactory,
344 bool biasEnabled);
345
346LayerTestResult<uint8_t, 4> ConstantLinearActivationUint8Test(armnn::IWorkloadFactory& workloadFactory);
347
348LayerTestResult<uint8_t, 4> ResizeBilinearNopUint8Test(armnn::IWorkloadFactory& workloadFactory);
349LayerTestResult<uint8_t, 4> SimpleResizeBilinearUint8Test(armnn::IWorkloadFactory& workloadFactory);
350LayerTestResult<uint8_t, 4> ResizeBilinearSqMinUint8Test(armnn::IWorkloadFactory& workloadFactory);
351LayerTestResult<uint8_t, 4> ResizeBilinearMinUint8Test(armnn::IWorkloadFactory& workloadFactory);
352LayerTestResult<uint8_t, 4> ResizeBilinearMagUint8Test(armnn::IWorkloadFactory& workloadFactory);
353
354LayerTestResult<uint8_t, 4> BatchNormUint8Test(armnn::IWorkloadFactory& workloadFactory);
Matteo Martincigh8eb675e2018-10-17 14:43:29 +0100[diff] [blame]355LayerTestResult<uint8_t, 4> BatchNormUint8NhwcTest(armnn::IWorkloadFactory& workloadFactory);
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]356
357LayerTestResult<uint8_t, 4> ConstantUint8Test(armnn::IWorkloadFactory& workloadFactory);
358
359LayerTestResult<uint8_t, 1> Concatenation1dUint8Test(armnn::IWorkloadFactory& workloadFactory);
360LayerTestResult<uint8_t, 2> Concatenation2dDim0Uint8Test(armnn::IWorkloadFactory& workloadFactory);
361LayerTestResult<uint8_t, 2> Concatenation2dDim1Uint8Test(armnn::IWorkloadFactory& workloadFactory);
362LayerTestResult<uint8_t, 2> Concatenation2dDim0DiffInputDimsUint8Test(armnn::IWorkloadFactory& workloadFactory);
363LayerTestResult<uint8_t, 2> Concatenation2dDim1DiffInputDimsUint8Test(armnn::IWorkloadFactory& workloadFactory);
364LayerTestResult<uint8_t, 3> Concatenation3dDim0Uint8Test(armnn::IWorkloadFactory& workloadFactory);
365LayerTestResult<uint8_t, 3> Concatenation3dDim1Uint8Test(armnn::IWorkloadFactory& workloadFactory);
366LayerTestResult<uint8_t, 3> Concatenation3dDim2Uint8Test(armnn::IWorkloadFactory& workloadFactory);
367LayerTestResult<uint8_t, 3> Concatenation3dDim0DiffInputDimsUint8Test(armnn::IWorkloadFactory& workloadFactory);
368LayerTestResult<uint8_t, 3> Concatenation3dDim1DiffInputDimsUint8Test(armnn::IWorkloadFactory& workloadFactory);
369LayerTestResult<uint8_t, 3> Concatenation3dDim2DiffInputDimsUint8Test(armnn::IWorkloadFactory& workloadFactory);
370
371
372LayerTestResult<float, 2> FullyConnectedLargeTest(armnn::IWorkloadFactory& workloadFactory,
373 bool transposeWeights);
374LayerTestResult<float, 4> SimplePermuteFloat32Test(armnn::IWorkloadFactory& workloadFactory);
375LayerTestResult<uint8_t, 4> SimplePermuteUint8Test(armnn::IWorkloadFactory& workloadFactory);
376
Mohamed Nour Abouelseouddd6acea2018-10-18 12:26:19 +0100[diff] [blame]377LayerTestResult<uint8_t, 2> PadUint82dTest(armnn::IWorkloadFactory& workloadFactory);
378LayerTestResult<uint8_t, 3> PadUint83dTest(armnn::IWorkloadFactory& workloadFactory);
379LayerTestResult<uint8_t, 4> PadUint84dTest(armnn::IWorkloadFactory& workloadFactory);
Mohamed Nour Abouelseoud7420e552018-10-12 12:26:24 +0100[diff] [blame]380
Mohamed Nour Abouelseouddd6acea2018-10-18 12:26:19 +0100[diff] [blame]381LayerTestResult<float, 2> PadFloat322dTest(armnn::IWorkloadFactory& workloadFactory);
382LayerTestResult<float, 3> PadFloat323dTest(armnn::IWorkloadFactory& workloadFactory);
383LayerTestResult<float, 4> PadFloat324dTest(armnn::IWorkloadFactory& workloadFactory);
Mohamed Nour Abouelseoud7420e552018-10-12 12:26:24 +0100[diff] [blame]384
surmeh01bceff2f2018-03-29 16:29:27 +0100[diff] [blame]385LayerTestResult<float, 4> PermuteFloat32ValueSet1Test(armnn::IWorkloadFactory& workloadFactory);
386LayerTestResult<float, 4> PermuteFloat32ValueSet2Test(armnn::IWorkloadFactory& workloadFactory);
387LayerTestResult<float, 4> PermuteFloat32ValueSet3Test(armnn::IWorkloadFactory& workloadFactory);
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]388
389LayerTestResult<float, 2> LstmLayerFloat32WithCifgWithPeepholeNoProjectionTest
390 (armnn::IWorkloadFactory& workloadFactory);
391LayerTestResult<float, 2>
392 LstmLayerFloat32NoCifgNoPeepholeNoProjectionTest(armnn::IWorkloadFactory& workloadFactory);
393LayerTestResult<float, 2>
394LstmLayerFloat32NoCifgWithPeepholeWithProjectionTest(armnn::IWorkloadFactory& workloadFactory);
395
396LayerTestResult<float, 4> SimpleConvertFp16ToFp32Test(armnn::IWorkloadFactory& workloadFactory);
397LayerTestResult<armnn::Half, 4> SimpleConvertFp32ToFp16Test(armnn::IWorkloadFactory& workloadFactory);
narpra011e4c31d2018-09-28 11:07:51 +0100[diff] [blame]398
399
400LayerTestResult<uint8_t, 1> MeanUint8SimpleTest(armnn::IWorkloadFactory& workloadFactory);
401LayerTestResult<uint8_t, 3> MeanUint8SimpleAxisTest(armnn::IWorkloadFactory& workloadFactory);
402LayerTestResult<uint8_t, 4> MeanUint8KeepDimsTest(armnn::IWorkloadFactory& workloadFactory);
403LayerTestResult<uint8_t, 4> MeanUint8MultipleDimsTest(armnn::IWorkloadFactory& workloadFactory);
404LayerTestResult<uint8_t, 1> MeanVtsUint8Test(armnn::IWorkloadFactory& workloadFactory);
405LayerTestResult<float, 1> MeanFloatSimpleTest(armnn::IWorkloadFactory& workloadFactory);
406LayerTestResult<float, 3> MeanFloatSimpleAxisTest(armnn::IWorkloadFactory& workloadFactory);
407LayerTestResult<float, 4> MeanFloatKeepDimsTest(armnn::IWorkloadFactory& workloadFactory);
408LayerTestResult<float, 4> MeanFloatMultipleDimsTest(armnn::IWorkloadFactory& workloadFactory);
409LayerTestResult<float, 1> MeanVtsFloat1Test(armnn::IWorkloadFactory& workloadFactory);
410LayerTestResult<float, 3> MeanVtsFloat2Test(armnn::IWorkloadFactory& workloadFactory);
Matteo Martincigh28dcab62018-10-19 16:40:03 +0100[diff] [blame]411LayerTestResult<float, 3> MeanVtsFloat3Test(armnn::IWorkloadFactory& workloadFactory);
412
Matteo Martincigh8eb675e2018-10-17 14:43:29 +0100[diff] [blame]413LayerTestResult<float, 4> AdditionAfterMaxPoolTest(armnn::IWorkloadFactory& workloadFactory);