Gitiles
Code Review Sign In
review.mlplatform.org / ml / armnn / 529195f3ae752192d5f205345235bf537f472cbc / . / src / backends / reference / test / RefLayerTests.cpp
blob: 69694e02752f89bf01a66926f5a35ca3da27b974 [file] [log] [blame]
Laurent Carlier749294b2020-06-01 09:03:17 +0100[diff] [blame]1//
Finn Williams2605b232020-06-10 15:53:46 +0100[diff] [blame]2// Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
David Beckecb56cd2018-09-05 12:52:57 +0100[diff] [blame]3// SPDX-License-Identifier: MIT
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]4//
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]5
Aron Virginas-Tar56055192018-11-12 18:10:43 +0000[diff] [blame]6#include "RefWorkloadFactoryHelper.hpp"
7
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]8#include <backendsCommon/test/LayerTests.hpp>
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]9
Aron Virginas-Tarc9cc8042018-11-01 16:15:57 +0000[diff] [blame]10#include <reference/RefWorkloadFactory.hpp>
Aron Virginas-Tar93820382019-06-27 16:59:36 +0100[diff] [blame]11
Sadik Armagana097d2a2021-11-24 15:47:28 +0000[diff] [blame]12#include <UnitTests.hpp>
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]13
Sadik Armagan1625efc2021-06-10 18:24:34 +0100[diff] [blame]14TEST_SUITE("Compute_Reference")
15{
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]16
17using namespace armnn;
18
19using FactoryType = RefWorkloadFactory;
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]20
21// ============================================================================
22// UNIT tests
23
24// Convolution
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]25ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution2d3x5, SimpleConvolution2d3x5Test, true, DataLayout::NCHW)
26ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution2d3x5Uint8, SimpleConvolution2d3x5Uint8Test, true, DataLayout::NCHW)
27ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution2d3x5Nhwc, SimpleConvolution2d3x5Test, true, DataLayout::NHWC)
28ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution2d3x5Uint8Nhwc, SimpleConvolution2d3x5Uint8Test, true, DataLayout::NHWC)
29ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution2d3x5QSymm16, SimpleConvolution2d3x5QSymm16Test, true, DataLayout::NCHW)
30ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution2d3x5QSymm16Nhwc,
31 SimpleConvolution2d3x5QSymm16Test,
32 true,
33 DataLayout::NHWC)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]34
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]35ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedConvolution2d, SimpleConvolution2d3x5Test, false, DataLayout::NCHW)
36ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedConvolutionUint8, SimpleConvolution2d3x5Uint8Test, false, DataLayout::NCHW)
37ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedConvolution2dNhwc, SimpleConvolution2d3x5Test, false, DataLayout::NHWC)
38ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedConvolutionUint8Nhwc, SimpleConvolution2d3x5Uint8Test, false, DataLayout::NHWC)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]39
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]40ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution1d, Convolution1dTest, true)
41ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution1dUint8, Convolution1dUint8Test, true)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]42
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]43ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution2d3x3, SimpleConvolution2d3x3Test, true, DataLayout::NCHW)
44ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution2d3x3Uint8, SimpleConvolution2d3x3Uint8Test, true, DataLayout::NCHW)
45ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution2d3x3QSymm16, SimpleConvolution2d3x3QSymm16Test, true, DataLayout::NCHW)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]46
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]47ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution2d3x3Nhwc, SimpleConvolution2d3x3Test, true, DataLayout::NHWC)
48ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution2d3x3Uint8Nhwc, SimpleConvolution2d3x3Uint8Test, true, DataLayout::NHWC)
49ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution2d3x3QSymm16Nhwc, SimpleConvolution2d3x3QSymm16Test, true,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]50 DataLayout::NCHW)
narpra015f703182018-10-26 16:24:58 +0100[diff] [blame]51
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]52ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedConvolution2dSquare, SimpleConvolution2d3x3Test, false, DataLayout::NCHW)
53ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedConvolution2dSquareNhwc, SimpleConvolution2d3x3Test, false, DataLayout::NHWC)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]54
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]55ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedConvolution2dSquareStride2x2Nhwc,
Mike Kelly7332ed82018-12-20 17:03:06 +0000[diff] [blame]56 SimpleConvolution2d3x3Stride2x2Test,
57 false,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]58 DataLayout::NHWC)
Mike Kelly7332ed82018-12-20 17:03:06 +0000[diff] [blame]59
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]60ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution2dAsymmetricPaddingLargerThanHalfKernelSize,
narpra015f703182018-10-26 16:24:58 +0100[diff] [blame]61 Convolution2dAsymmetricPaddingLargerThanHalfKernelSizeTest,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]62 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]63ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution2dAsymmetricPadding,
64 Convolution2dAsymmetricPaddingTest, DataLayout::NCHW)
narpra015f703182018-10-26 16:24:58 +0100[diff] [blame]65
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]66ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution2dAsymmetricPaddingLargerThanHalfKernelSizeNhwc,
narpra015f703182018-10-26 16:24:58 +0100[diff] [blame]67 Convolution2dAsymmetricPaddingLargerThanHalfKernelSizeTest,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]68 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]69ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution2dAsymmetricPaddingNhwc,
narpra015f703182018-10-26 16:24:58 +0100[diff] [blame]70 Convolution2dAsymmetricPaddingTest,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]71 DataLayout::NHWC)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]72
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]73ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution2dSquareNhwc, SimpleConvolution2d3x3NhwcTest, false)
Nikhil Raje4dfd6e2018-10-18 10:11:04 +0100[diff] [blame]74
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]75ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d3x3Dilation3x3BFloat16,
Narumol Prangnawarat44179c32020-03-11 14:51:27 +0000[diff] [blame]76 Convolution2d3x3Dilation3x3Test<DataType::BFloat16, DataType::BFloat16>,
77 false,
78 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]79ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d3x3Dilation3x3NhwcBFloat16,
Narumol Prangnawarat44179c32020-03-11 14:51:27 +0000[diff] [blame]80 Convolution2d3x3Dilation3x3Test<DataType::BFloat16, DataType::BFloat16>,
81 false,
82 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]83ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d3x3Dilation3x3,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]84 Convolution2d3x3Dilation3x3Test<DataType::Float32, DataType::Float32>,
Teresa Charlinedeeb162019-06-14 11:09:19 +0100[diff] [blame]85 false,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]86 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]87ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d3x3Dilation3x3Nhwc,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]88 Convolution2d3x3Dilation3x3Test<DataType::Float32, DataType::Float32>,
Teresa Charlinedeeb162019-06-14 11:09:19 +0100[diff] [blame]89 false,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]90 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]91ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d3x3Dilation3x3Int8,
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]92 Convolution2d3x3Dilation3x3Test<DataType::QAsymmS8, DataType::Signed32>,
93 false,
94 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]95ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d3x3Dilation3x3NhwcInt8,
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]96 Convolution2d3x3Dilation3x3Test<DataType::QAsymmS8, DataType::Signed32>,
97 false,
98 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]99ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d3x3Dilation3x3Uint8,
Derek Lambertif90c56d2020-01-10 17:14:08 +0000[diff] [blame]100 Convolution2d3x3Dilation3x3Test<DataType::QAsymmU8, DataType::Signed32>,
Teresa Charlinedeeb162019-06-14 11:09:19 +0100[diff] [blame]101 false,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]102 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]103ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d3x3Dilation3x3NhwcUint8,
Derek Lambertif90c56d2020-01-10 17:14:08 +0000[diff] [blame]104 Convolution2d3x3Dilation3x3Test<DataType::QAsymmU8, DataType::Signed32>,
Teresa Charlinedeeb162019-06-14 11:09:19 +0100[diff] [blame]105 false,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]106 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]107ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d3x3Dilation3x3Int16,
Derek Lambertif90c56d2020-01-10 17:14:08 +0000[diff] [blame]108 Convolution2d3x3Dilation3x3Test<DataType::QSymmS16, DataType::Signed32>,
Teresa Charlinedeeb162019-06-14 11:09:19 +0100[diff] [blame]109 false,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]110 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]111ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d3x3Dilation3x3NhwcInt16,
Derek Lambertif90c56d2020-01-10 17:14:08 +0000[diff] [blame]112 Convolution2d3x3Dilation3x3Test<DataType::QSymmS16, DataType::Signed32>,
Teresa Charlinedeeb162019-06-14 11:09:19 +0100[diff] [blame]113 false,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]114 DataLayout::NHWC)
Teresa Charlinedeeb162019-06-14 11:09:19 +0100[diff] [blame]115
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]116ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x3x3Dilation3x3BFloat16,
Narumol Prangnawarat44179c32020-03-11 14:51:27 +0000[diff] [blame]117 Convolution2d2x3x3Dilation3x3Test<DataType::BFloat16, DataType::BFloat16>,
118 false,
119 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]120ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x3x3Dilation3x3NhwcBFloat16,
Narumol Prangnawarat44179c32020-03-11 14:51:27 +0000[diff] [blame]121 Convolution2d2x3x3Dilation3x3Test<DataType::BFloat16, DataType::BFloat16>,
122 false,
123 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]124ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x3x3Dilation3x3,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]125 Convolution2d2x3x3Dilation3x3Test<DataType::Float32, DataType::Float32>,
Teresa Charlinedeeb162019-06-14 11:09:19 +0100[diff] [blame]126 false,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]127 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]128ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x3x3Dilation3x3Nhwc,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]129 Convolution2d2x3x3Dilation3x3Test<DataType::Float32, DataType::Float32>,
Teresa Charlinedeeb162019-06-14 11:09:19 +0100[diff] [blame]130 false,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]131 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]132ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x3x3Dilation3x3Int8,
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]133 Convolution2d2x3x3Dilation3x3Test<DataType::QAsymmS8, DataType::Signed32>,
134 false,
135 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]136ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x3x3Dilation3x3NhwcInt8,
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]137 Convolution2d2x3x3Dilation3x3Test<DataType::QAsymmS8, DataType::Signed32>,
138 false,
139 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]140ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x3x3Dilation3x3Uint8,
Derek Lambertif90c56d2020-01-10 17:14:08 +0000[diff] [blame]141 Convolution2d2x3x3Dilation3x3Test<DataType::QAsymmU8, DataType::Signed32>,
Teresa Charlinedeeb162019-06-14 11:09:19 +0100[diff] [blame]142 false,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]143 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]144ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x3x3Dilation3x3NhwcUint8,
Derek Lambertif90c56d2020-01-10 17:14:08 +0000[diff] [blame]145 Convolution2d2x3x3Dilation3x3Test<DataType::QAsymmU8, DataType::Signed32>,
Teresa Charlinedeeb162019-06-14 11:09:19 +0100[diff] [blame]146 false,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]147 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]148ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x3x3Dilation3x3Int16,
Derek Lambertif90c56d2020-01-10 17:14:08 +0000[diff] [blame]149 Convolution2d2x3x3Dilation3x3Test<DataType::QSymmS16, DataType::Signed32>,
Teresa Charlinedeeb162019-06-14 11:09:19 +0100[diff] [blame]150 false,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]151 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]152ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x3x3Dilation3x3NhwcInt16,
Derek Lambertif90c56d2020-01-10 17:14:08 +0000[diff] [blame]153 Convolution2d2x3x3Dilation3x3Test<DataType::QSymmS16, DataType::Signed32>,
Teresa Charlinedeeb162019-06-14 11:09:19 +0100[diff] [blame]154 false,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]155 DataLayout::NHWC)
Teresa Charlinedeeb162019-06-14 11:09:19 +0100[diff] [blame]156
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]157ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x2Dilation2x2Padding2x2Stride3x3BFloat16,
Narumol Prangnawarat44179c32020-03-11 14:51:27 +0000[diff] [blame]158 Convolution2d2x2Dilation2x2Padding2x2Stride3x3Test<DataType::BFloat16, DataType::BFloat16>,
159 false,
160 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]161
162ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x2Dilation2x2Padding2x2Stride3x3NhwcBFloat16,
Narumol Prangnawarat44179c32020-03-11 14:51:27 +0000[diff] [blame]163 Convolution2d2x2Dilation2x2Padding2x2Stride3x3Test<DataType::BFloat16, DataType::BFloat16>,
164 false,
165 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]166ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x2Dilation2x2Padding2x2Stride3x3,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]167 Convolution2d2x2Dilation2x2Padding2x2Stride3x3Test<DataType::Float32, DataType::Float32>,
Teresa Charlin2b7519d2019-07-09 15:45:35 +0100[diff] [blame]168 false,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]169 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]170ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x2Dilation2x2Padding2x2Stride3x3Nhwc,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]171 Convolution2d2x2Dilation2x2Padding2x2Stride3x3Test<DataType::Float32, DataType::Float32>,
Teresa Charlin2b7519d2019-07-09 15:45:35 +0100[diff] [blame]172 false,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]173 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]174ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x2Dilation2x2Padding2x2Stride3x3Int8,
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]175 Convolution2d2x2Dilation2x2Padding2x2Stride3x3Test<DataType::QAsymmS8, DataType::Signed32>,
176 false,
177 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]178ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x2Dilation2x2Padding2x2Stride3x3NhwcInt8,
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]179 Convolution2d2x2Dilation2x2Padding2x2Stride3x3Test<DataType::QAsymmS8, DataType::Signed32>,
180 false,
181 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]182ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x2Dilation2x2Padding2x2Stride3x3Uint8,
Derek Lambertif90c56d2020-01-10 17:14:08 +0000[diff] [blame]183 Convolution2d2x2Dilation2x2Padding2x2Stride3x3Test<DataType::QAsymmU8, DataType::Signed32>,
Teresa Charlin2b7519d2019-07-09 15:45:35 +0100[diff] [blame]184 false,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]185 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]186ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x2Dilation2x2Padding2x2Stride3x3NhwcUint8,
Derek Lambertif90c56d2020-01-10 17:14:08 +0000[diff] [blame]187 Convolution2d2x2Dilation2x2Padding2x2Stride3x3Test<DataType::QAsymmU8, DataType::Signed32>,
Teresa Charlin2b7519d2019-07-09 15:45:35 +0100[diff] [blame]188 false,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]189 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]190ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x2Dilation2x2Padding2x2Stride3x3Int16,
Derek Lambertif90c56d2020-01-10 17:14:08 +0000[diff] [blame]191 Convolution2d2x2Dilation2x2Padding2x2Stride3x3Test<DataType::QSymmS16, DataType::Signed32>,
Teresa Charlin2b7519d2019-07-09 15:45:35 +0100[diff] [blame]192 false,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]193 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]194ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d2x2Dilation2x2Padding2x2Stride3x3NhwcInt16,
Derek Lambertif90c56d2020-01-10 17:14:08 +0000[diff] [blame]195 Convolution2d2x2Dilation2x2Padding2x2Stride3x3Test<DataType::QSymmS16, DataType::Signed32>,
Teresa Charlin2b7519d2019-07-09 15:45:35 +0100[diff] [blame]196 false,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]197 DataLayout::NHWC)
Teresa Charlin2b7519d2019-07-09 15:45:35 +0100[diff] [blame]198
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]199ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2dPerAxisQuantTestNchw, Convolution2dPerAxisQuantTest, DataLayout::NCHW);
200ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2dPerAxisQuantTestNhwc, Convolution2dPerAxisQuantTest, DataLayout::NHWC);
Teresa Charlin2b7519d2019-07-09 15:45:35 +0100[diff] [blame]201
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]202ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d3x3Stride2x2Bf16,
203 Convolution2d3x3Stride2x2BFloat16Test,
204 false,
205 DataLayout::NHWC);
206ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution2d3x3Stride2x2BFloat16SmallValue,
Narumol Prangnawarate8cddeb2020-04-01 16:51:23 +0100[diff] [blame]207 Convolution2d3x3Stride2x2BFloat16SmallValueTest,
208 false,
209 DataLayout::NHWC);
210
Matthew Sloyan5d7b0a32021-10-18 13:07:49 +0100[diff] [blame]211// Convolution 3d - NDHWC
212ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution3d3x3x3Float32,
213 SimpleConvolution3d3x3x3Float32Test,
214 false,
215 DataLayout::NDHWC)
216ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution3d3x3x3Int8,
217 SimpleConvolution3d3x3x3Int8Test,
218 false,
219 DataLayout::NDHWC)
220ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution3d3x3x3Uint8,
221 SimpleConvolution3d3x3x3Uint8Test,
222 false,
223 DataLayout::NDHWC)
224ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution3d3x3x3Int16,
225 SimpleConvolution3d3x3x3Int16Test,
226 false,
227 DataLayout::NDHWC)
Matthew Sloyanb63a3112021-09-08 13:05:51 +0100[diff] [blame]228
Matthew Sloyan5d7b0a32021-10-18 13:07:49 +0100[diff] [blame]229ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution3d2x2x2Strides3x5x5Float32,
230 Convolution3d2x2x2Strides3x5x5Float32Test,
231 false,
232 DataLayout::NDHWC)
233ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution3d2x2x2Strides3x5x5TestInt8,
234 Convolution3d2x2x2Strides3x5x5Int8Test,
235 true,
236 DataLayout::NDHWC)
237ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution3d2x2x2Strides3x5x5TestUint8,
238 Convolution3d2x2x2Strides3x5x5Uint8Test,
239 false,
240 DataLayout::NDHWC)
241ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution3d2x2x2Strides3x5x5TestInt16,
242 Convolution3d2x2x2Strides3x5x5Int16Test,
243 true,
244 DataLayout::NDHWC)
Matthew Sloyanb63a3112021-09-08 13:05:51 +0100[diff] [blame]245
Matthew Sloyan5d7b0a32021-10-18 13:07:49 +0100[diff] [blame]246ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution3dPaddingSame3x3x3Float32,
247 Convolution3dPaddingSame3x3x3Float32Test,
248 false,
249 DataLayout::NDHWC)
250ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution3dPaddingSame3x3x3TestInt8,
251 Convolution3dPaddingSame3x3x3Int8Test,
252 false,
253 DataLayout::NDHWC)
254ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution3dPaddingSame3x3x3TestUint8,
255 Convolution3dPaddingSame3x3x3Uint8Test,
256 false,
257 DataLayout::NDHWC)
258ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution3dPaddingSame3x3x3TestInt16,
259 Convolution3dPaddingSame3x3x3Int16Test,
260 false,
261 DataLayout::NDHWC)
Matthew Sloyanb63a3112021-09-08 13:05:51 +0100[diff] [blame]262
Matthew Sloyan5d7b0a32021-10-18 13:07:49 +0100[diff] [blame]263ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution3d2x2x2Dilation2x2x2Float32,
264 Convolution3d2x2x2Dilation2x2x2Float32Test,
265 true,
266 DataLayout::NDHWC)
267ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution3d2x2x2Dilation2x2x2TestInt8,
268 Convolution3d2x2x2Dilation2x2x2Int8Test,
269 true,
270 DataLayout::NDHWC)
271ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution3d2x2x2Dilation2x2x2TestUint8,
272 Convolution3d2x2x2Dilation2x2x2Uint8Test,
273 true,
274 DataLayout::NDHWC)
275ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution3d2x2x2Dilation2x2x2TestInt16,
276 Convolution3d2x2x2Dilation2x2x2Int16Test,
277 true,
278 DataLayout::NDHWC)
Matthew Sloyanb63a3112021-09-08 13:05:51 +0100[diff] [blame]279
280ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution3dStrideDilationPadding3x3x3Float32,
281 Convolution3dStrideDilationPadding3x3x3Float32Test,
Matthew Sloyan5d7b0a32021-10-18 13:07:49 +0100[diff] [blame]282 true,
283 DataLayout::NDHWC)
Matthew Sloyanb63a3112021-09-08 13:05:51 +0100[diff] [blame]284
285ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution3d2x2x2Stride3x3x3SmallTestFloat32,
286 Convolution3d2x2x2Stride3x3x3SmallFloat32Test,
Matthew Sloyan5d7b0a32021-10-18 13:07:49 +0100[diff] [blame]287 false,
288 DataLayout::NDHWC)
Matthew Sloyanb63a3112021-09-08 13:05:51 +0100[diff] [blame]289
Matthew Sloyan5d7b0a32021-10-18 13:07:49 +0100[diff] [blame]290ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution3d2x3x3TestFloat16,
291 Convolution3d2x3x3Float16Test,
292 true,
293 DataLayout::NDHWC)
294ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution3d2x2x2SmallTestFloat16,
295 Convolution3d2x2x2SmallFloat16Test,
296 false,
297 DataLayout::NDHWC)
298
299// Convolution 3d - NCDHW
300ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvolution3d3x3x3NcdhwFloat32,
301 SimpleConvolution3d3x3x3Float32Test,
302 false,
303 DataLayout::NCDHW)
304
305ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution3d2x3x3TestNcdhwFloat16,
306 Convolution3d2x3x3Float16Test,
307 false,
308 DataLayout::NCDHW)
309
310ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution3d2x2x2Strides3x5x5NcdhwTestInt8,
311 Convolution3d2x2x2Strides3x5x5Int8Test,
312 true,
313 DataLayout::NCDHW)
314
315ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution3dPaddingSame3x3x3NcdhwTestUint8,
316 Convolution3dPaddingSame3x3x3Uint8Test,
317 false,
318 DataLayout::NCDHW)
319
320ARMNN_AUTO_TEST_CASE_WITH_THF(Convolution3d2x2x2Dilation2x2x2NcdhwTestInt16,
321 Convolution3d2x2x2Dilation2x2x2Int16Test,
322 true,
323 DataLayout::NCDHW)
Matthew Sloyanb63a3112021-09-08 13:05:51 +0100[diff] [blame]324
325
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]326// Depthwise Convolution
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]327ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2d, DepthwiseConvolution2dTest, true, DataLayout::NCHW)
328ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dUint8, DepthwiseConvolution2dUint8Test, true, DataLayout::NCHW)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]329
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]330ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedDepthwiseConvolution2d, DepthwiseConvolution2dTest, false, DataLayout::NCHW)
331ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedDepthwiseConvolution2dUint8,
jimfly01d84216a2018-10-26 12:56:21 +0100[diff] [blame]332 DepthwiseConvolution2dUint8Test,
333 false,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]334 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]335ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dQSymm16, DepthwiseConvolution2dInt16Test, true, DataLayout::NCHW)
jimfly01d84216a2018-10-26 12:56:21 +0100[diff] [blame]336
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]337ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dNhwc, DepthwiseConvolution2dTest, true, DataLayout::NHWC)
338ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dUint8Nhwc, DepthwiseConvolution2dUint8Test, true, DataLayout::NHWC)
jimfly01d84216a2018-10-26 12:56:21 +0100[diff] [blame]339
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]340ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedDepthwiseConvolution2dNhwc, DepthwiseConvolution2dTest, false, DataLayout::NHWC)
341ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedDepthwiseConvolution2dUint8Nhwc,
jimfly01d84216a2018-10-26 12:56:21 +0100[diff] [blame]342 DepthwiseConvolution2dUint8Test,
343 false,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]344 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]345ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dDepthNhwc, DepthwiseConvolution2dDepthNhwcTest, false)
346ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleDepthwiseConvolution2d3x3Dilation3x3Nhwc,
Matthew Bentham49cb7d02019-05-10 16:55:23 +0100[diff] [blame]347 SimpleDepthwiseConvolution2d3x3Dilation3x3NhwcTest)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]348
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]349ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2d3x3Dilation3x3,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]350 DepthwiseConvolution2d3x3Dilation3x3Test<DataType::Float32, DataType::Float32>,
Teresa Charlin20b1f882019-06-19 09:34:37 +0100[diff] [blame]351 false,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]352 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]353ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2d3x3Dilation3x3Nhwc,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]354 DepthwiseConvolution2d3x3Dilation3x3Test<DataType::Float32, DataType::Float32>,
Teresa Charlin20b1f882019-06-19 09:34:37 +0100[diff] [blame]355 false,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]356 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]357ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2d3x3Dilation3x3BFloat16,
Narumol Prangnawarat44179c32020-03-11 14:51:27 +0000[diff] [blame]358 DepthwiseConvolution2d3x3Dilation3x3Test<DataType::BFloat16, DataType::BFloat16>,
359 false,
360 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]361ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2d3x3Dilation3x3NhwcBFloat16,
Narumol Prangnawarat44179c32020-03-11 14:51:27 +0000[diff] [blame]362 DepthwiseConvolution2d3x3Dilation3x3Test<DataType::BFloat16, DataType::BFloat16>,
363 false,
364 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]365ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2d3x3Dilation3x3Int8,
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]366 DepthwiseConvolution2d3x3Dilation3x3Test<DataType::QAsymmS8, DataType::Signed32>,
367 false,
368 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]369ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2d3x3Dilation3x3NhwcInt8,
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]370 DepthwiseConvolution2d3x3Dilation3x3Test<DataType::QAsymmS8, DataType::Signed32>,
371 false,
372 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]373ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2d3x3Dilation3x3Uint8,
Derek Lambertif90c56d2020-01-10 17:14:08 +0000[diff] [blame]374 DepthwiseConvolution2d3x3Dilation3x3Test<DataType::QAsymmU8, DataType::Signed32>,
Teresa Charlin20b1f882019-06-19 09:34:37 +0100[diff] [blame]375 false,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]376 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]377ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2d3x3Dilation3x3NhwcUint8,
Derek Lambertif90c56d2020-01-10 17:14:08 +0000[diff] [blame]378 DepthwiseConvolution2d3x3Dilation3x3Test<DataType::QAsymmU8, DataType::Signed32>,
Teresa Charlin20b1f882019-06-19 09:34:37 +0100[diff] [blame]379 false,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]380 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]381ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2d3x3Dilation3x3Int16,
Derek Lambertif90c56d2020-01-10 17:14:08 +0000[diff] [blame]382 DepthwiseConvolution2d3x3Dilation3x3Test<DataType::QSymmS16, DataType::Signed32>,
Teresa Charlin20b1f882019-06-19 09:34:37 +0100[diff] [blame]383 false,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]384 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]385ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2d3x3Dilation3x3NhwcInt16,
Derek Lambertif90c56d2020-01-10 17:14:08 +0000[diff] [blame]386 DepthwiseConvolution2d3x3Dilation3x3Test<DataType::QSymmS16, DataType::Signed32>,
Teresa Charlin20b1f882019-06-19 09:34:37 +0100[diff] [blame]387 false,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]388 DataLayout::NHWC)
Teresa Charlin20b1f882019-06-19 09:34:37 +0100[diff] [blame]389
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]390ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2d2x3x3Dilation3x3,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]391 DepthwiseConvolution2d2x3x3Dilation3x3Test<DataType::Float32, DataType::Float32>,
Teresa Charlin20b1f882019-06-19 09:34:37 +0100[diff] [blame]392 false,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]393 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]394ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2d2x3x3Dilation3x3Nhwc,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]395 DepthwiseConvolution2d2x3x3Dilation3x3Test<DataType::Float32, DataType::Float32>,
Teresa Charlin20b1f882019-06-19 09:34:37 +0100[diff] [blame]396 false,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]397 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]398ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2d2x3x3Dilation3x3BFloat16,
Narumol Prangnawarat44179c32020-03-11 14:51:27 +0000[diff] [blame]399 DepthwiseConvolution2d2x3x3Dilation3x3Test<DataType::BFloat16, DataType::BFloat16>,
400 false,
401 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]402ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2d2x3x3Dilation3x3NhwcBFloat16,
Narumol Prangnawarat44179c32020-03-11 14:51:27 +0000[diff] [blame]403 DepthwiseConvolution2d2x3x3Dilation3x3Test<DataType::BFloat16, DataType::BFloat16>,
404 false,
405 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]406ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2d2x3x3Dilation3x3Int8,
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]407 DepthwiseConvolution2d2x3x3Dilation3x3Test<DataType::QAsymmS8, DataType::Signed32>,
408 false,
409 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]410ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2d2x3x3Dilation3x3NhwcInt8,
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]411 DepthwiseConvolution2d2x3x3Dilation3x3Test<DataType::QAsymmS8, DataType::Signed32>,
412 false,
413 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]414ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2d2x3x3Dilation3x3Uint8,
Derek Lambertif90c56d2020-01-10 17:14:08 +0000[diff] [blame]415 DepthwiseConvolution2d2x3x3Dilation3x3Test<DataType::QAsymmU8, DataType::Signed32>,
Teresa Charlin20b1f882019-06-19 09:34:37 +0100[diff] [blame]416 false,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]417 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]418ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2d2x3x3Dilation3x3NhwcUint8,
Derek Lambertif90c56d2020-01-10 17:14:08 +0000[diff] [blame]419 DepthwiseConvolution2d2x3x3Dilation3x3Test<DataType::QAsymmU8, DataType::Signed32>,
Teresa Charlin20b1f882019-06-19 09:34:37 +0100[diff] [blame]420 false,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]421 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]422ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2d2x3x3Dilation3x3Int16,
Derek Lambertif90c56d2020-01-10 17:14:08 +0000[diff] [blame]423 DepthwiseConvolution2d2x3x3Dilation3x3Test<DataType::QSymmS16, DataType::Signed32>,
Teresa Charlin20b1f882019-06-19 09:34:37 +0100[diff] [blame]424 false,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]425 DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]426ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2d2x3x3Dilation3x3NhwcInt16,
Derek Lambertif90c56d2020-01-10 17:14:08 +0000[diff] [blame]427 DepthwiseConvolution2d2x3x3Dilation3x3Test<DataType::QSymmS16, DataType::Signed32>,
Teresa Charlin20b1f882019-06-19 09:34:37 +0100[diff] [blame]428 false,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]429 DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]430ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dMult4,
Kevin May665a964a2019-08-21 16:53:50 +0100[diff] [blame]431 DepthwiseConvolution2dMult4Test<armnn::DataType::Float32, armnn::DataType::Float32>,
432 false,
433 armnn::DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]434ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dMult2,
Kevin May665a964a2019-08-21 16:53:50 +0100[diff] [blame]435 DepthwiseConvolution2dMult2Test<armnn::DataType::Float32, armnn::DataType::Float32>,
436 false,
437 armnn::DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]438ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dMult4BFloat16,
Narumol Prangnawarat44179c32020-03-11 14:51:27 +0000[diff] [blame]439 DepthwiseConvolution2dMult4Test<armnn::DataType::BFloat16, armnn::DataType::BFloat16>,
440 false,
441 armnn::DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]442ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dMult2BFloat16,
Narumol Prangnawarat44179c32020-03-11 14:51:27 +0000[diff] [blame]443 DepthwiseConvolution2dMult2Test<armnn::DataType::BFloat16, armnn::DataType::BFloat16>,
444 false,
445 armnn::DataLayout::NCHW)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]446
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]447ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dDepthMul1,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]448 DepthwiseConvolution2dDepthMul1Test, true, DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]449ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dDepthMul1Uint8,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]450 DepthwiseConvolution2dDepthMul1Uint8Test, true, DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]451ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dDepthMul1Int16,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]452 DepthwiseConvolution2dDepthMul1Int16Test, true, DataLayout::NCHW)
jimfly01b9c89632018-10-26 16:50:13 +0100[diff] [blame]453
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]454ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedDepthwiseConvolution2dDepthMul1,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]455 DepthwiseConvolution2dDepthMul1Test, false, DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]456ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedDepthwiseConvolution2dDepthMul1Uint8,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]457 DepthwiseConvolution2dDepthMul1Uint8Test, false, DataLayout::NCHW)
jimfly01b9c89632018-10-26 16:50:13 +0100[diff] [blame]458
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]459ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dDepthMul1Nhwc,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]460 DepthwiseConvolution2dDepthMul1Test, true, DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]461ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dDepthMul1Uint8Nhwc,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]462 DepthwiseConvolution2dDepthMul1Uint8Test, true, DataLayout::NHWC)
jimfly01b9c89632018-10-26 16:50:13 +0100[diff] [blame]463
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]464ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedDepthwiseConvolution2dDepthMul1Nhwc,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]465 DepthwiseConvolution2dDepthMul1Test, false, DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]466ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedDepthwiseConvolution2dDepthMul1Uint8Nhwc,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]467 DepthwiseConvolution2dDepthMul1Uint8Test, false, DataLayout::NHWC)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]468
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]469ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dAsymmetric,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]470 DepthwiseConvolution2dAsymmetricTest, true, DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]471ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedDepthwiseConvolution2dAsymmetric,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]472 DepthwiseConvolution2dAsymmetricTest, false, DataLayout::NCHW)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]473ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dAsymmetricNhwc,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]474 DepthwiseConvolution2dAsymmetricTest, true, DataLayout::NHWC)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]475ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedDepthwiseConvolution2dAsymmetricNhwc,
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]476 DepthwiseConvolution2dAsymmetricTest, false, DataLayout::NHWC)
surmeh013537c2c2018-05-18 16:31:43 +0100[diff] [blame]477
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]478ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dDepthMul64, DepthwiseConvolution2dDepthMul64Test);
Matthew Jacksond6a9dee2019-07-22 13:53:24 +0100[diff] [blame]479
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]480ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dPerAxisQuantTestNchw, DepthwiseConvolution2dPerAxisQuantTest,
Teresa Charlind8df0262019-11-11 12:28:15 +0000[diff] [blame]481 DataLayout::NCHW);
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]482ARMNN_AUTO_TEST_CASE_WITH_THF(DepthwiseConvolution2dPerAxisQuantTestNhwc, DepthwiseConvolution2dPerAxisQuantTest,
Teresa Charlind8df0262019-11-11 12:28:15 +0000[diff] [blame]483 DataLayout::NHWC);
484
Tamás Nyíri7b885b32021-10-26 14:47:57 +0100[diff] [blame]485// [ Pooling 2D
Teresa Charlin0434df62019-06-06 13:40:35 +0100[diff] [blame]486//MaxPooling
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]487ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling2dSize2x2Stride2x2, SimpleMaxPooling2dSize2x2Stride2x2Test, false)
488ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling2dSize2x2Stride2x2Uint8,
489 SimpleMaxPooling2dSize2x2Stride2x2Uint8Test, false)
490ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling2dSize2x2Stride2x2Int16,
491 SimpleMaxPooling2dSize2x2Stride2x2Int16Test, false)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]492
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]493ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling2dSize3x3Stride2x4, SimpleMaxPooling2dSize3x3Stride2x4Test, false)
494ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling2dSize3x3Stride2x4Uint8,
495 SimpleMaxPooling2dSize3x3Stride2x4Uint8Test, false)
496ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling2dSize3x3Stride2x4Int16,
497 SimpleMaxPooling2dSize3x3Stride2x4Int16Test, false)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]498
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]499ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling2d, SimpleMaxPooling2dTest, DataLayout::NCHW)
500ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling2dNhwc, SimpleMaxPooling2dTest, DataLayout::NHWC)
501ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling2dUint8, SimpleMaxPooling2dUint8Test, DataLayout::NCHW)
502ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling2dInt16, SimpleMaxPooling2dInt16Test, DataLayout::NCHW)
503ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling2dUint8Nhwc, SimpleMaxPooling2dUint8Test, DataLayout::NHWC)
504ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling2dInt16Nhwc, SimpleMaxPooling2dInt16Test, DataLayout::NHWC)
James Conroy45a9b772018-10-31 11:47:53 +0000[diff] [blame]505
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]506ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleMaxPooling2d, IgnorePaddingSimpleMaxPooling2dTest)
507ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleMaxPooling2dUint8, IgnorePaddingSimpleMaxPooling2dUint8Test)
508ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleMaxPooling2dInt16, IgnorePaddingSimpleMaxPooling2dInt16Test)
509ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingMaxPooling2dSize3, IgnorePaddingMaxPooling2dSize3Test)
510ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingMaxPooling2dSize3Uint8, IgnorePaddingMaxPooling2dSize3Uint8Test)
511ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingMaxPooling2dSize3Int16, IgnorePaddingMaxPooling2dSize3Int16Test)
Teresa Charlin0434df62019-06-06 13:40:35 +0100[diff] [blame]512
513//AveragePooling
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]514ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleAveragePooling2d, SimpleAveragePooling2dTest, DataLayout::NCHW)
515ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleAveragePooling2dNhwc, SimpleAveragePooling2dTest, DataLayout::NHWC)
516ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleAveragePooling2dUint8, SimpleAveragePooling2dUint8Test, DataLayout::NCHW)
517ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleAveragePooling2dInt16, SimpleAveragePooling2dInt16Test, DataLayout::NCHW)
518ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleAveragePooling2dUint8Nhwc, SimpleAveragePooling2dUint8Test, DataLayout::NHWC)
519ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleAveragePooling2dInt16Nhwc, SimpleAveragePooling2dInt16Test, DataLayout::NHWC)
Teresa Charlin0434df62019-06-06 13:40:35 +0100[diff] [blame]520
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]521ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleAveragePooling2d, IgnorePaddingSimpleAveragePooling2dTest)
522ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleAveragePooling2dUint8, IgnorePaddingSimpleAveragePooling2dUint8Test)
523ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleAveragePooling2dInt16, IgnorePaddingSimpleAveragePooling2dInt16Test)
524ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleAveragePooling2dNoPadding,
525 IgnorePaddingSimpleAveragePooling2dNoPaddingTest)
526ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleAveragePooling2dNoPaddingUint8,
527 IgnorePaddingSimpleAveragePooling2dNoPaddingUint8Test)
528ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleAveragePooling2dNoPaddingInt16,
529 IgnorePaddingSimpleAveragePooling2dNoPaddingInt16Test)
530ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingAveragePooling2dSize3, IgnorePaddingAveragePooling2dSize3Test)
531ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingAveragePooling2dSize3Uint8, IgnorePaddingAveragePooling2dSize3Uint8Test)
532ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingAveragePooling2dSize3Int16, IgnorePaddingAveragePooling2dSize3Int16Test)
James Conroy45a9b772018-10-31 11:47:53 +0000[diff] [blame]533
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]534ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingAveragePooling2dSize3x2Stride2x2,
535 IgnorePaddingAveragePooling2dSize3x2Stride2x2Test, false)
536ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingAveragePooling2dSize3x2Stride2x2NoPadding,
537 IgnorePaddingAveragePooling2dSize3x2Stride2x2Test, true)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]538
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]539ARMNN_AUTO_TEST_CASE_WITH_THF(LargeTensorsAveragePooling2d, LargeTensorsAveragePooling2dTest)
540ARMNN_AUTO_TEST_CASE_WITH_THF(LargeTensorsAveragePooling2dUint8, LargeTensorsAveragePooling2dUint8Test)
541ARMNN_AUTO_TEST_CASE_WITH_THF(LargeTensorsAveragePooling2dInt16, LargeTensorsAveragePooling2dInt16Test)
Teresa Charlin0434df62019-06-06 13:40:35 +0100[diff] [blame]542
543//L2Pooling
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]544ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleL2Pooling2d, IgnorePaddingSimpleL2Pooling2dTest)
545ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleL2Pooling2dUint8, IgnorePaddingSimpleL2Pooling2dUint8Test)
546ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleL2Pooling2dInt16, IgnorePaddingSimpleL2Pooling2dInt16Test)
Teresa Charlin0434df62019-06-06 13:40:35 +0100[diff] [blame]547
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]548ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingL2Pooling2dSize3, IgnorePaddingL2Pooling2dSize3Test)
549ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingL2Pooling2dSize3Uint8, IgnorePaddingL2Pooling2dSize3Uint8Test)
550ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingL2Pooling2dSize3Int16, IgnorePaddingL2Pooling2dSize3Int16Test)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]551
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]552ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleL2Pooling2d, SimpleL2Pooling2dTest, DataLayout::NCHW)
553ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleL2Pooling2dNhwc, SimpleL2Pooling2dTest, DataLayout::NHWC)
554ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleL2Pooling2dUint8, SimpleL2Pooling2dUint8Test, DataLayout::NCHW)
555ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleL2Pooling2dInt16, SimpleL2Pooling2dInt16Test, DataLayout::NCHW)
556ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleL2Pooling2dNhwcUint8, SimpleL2Pooling2dUint8Test, DataLayout::NHWC)
557ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleL2Pooling2dNhwcInt16, SimpleL2Pooling2dInt16Test, DataLayout::NHWC)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]558
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]559ARMNN_AUTO_TEST_CASE_WITH_THF(L2Pooling2dSize7, L2Pooling2dSize7Test)
560ARMNN_AUTO_TEST_CASE_WITH_THF(L2Pooling2dSize7Uint8, L2Pooling2dSize7Uint8Test)
561ARMNN_AUTO_TEST_CASE_WITH_THF(L2Pooling2dSize7Int16, L2Pooling2dSize7Int16Test)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]562
Teresa Charlin0434df62019-06-06 13:40:35 +0100[diff] [blame]563//NonSquarePooling
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]564ARMNN_AUTO_TEST_CASE_WITH_THF(AsymmNonSquarePooling2d, AsymmetricNonSquarePooling2dTest)
565ARMNN_AUTO_TEST_CASE_WITH_THF(AsymmNonSquarePooling2dUint8, AsymmetricNonSquarePooling2dUint8Test)
566ARMNN_AUTO_TEST_CASE_WITH_THF(AsymmNonSquarePooling2dInt16, AsymmetricNonSquarePooling2dInt16Test)
Tamás Nyíri7b885b32021-10-26 14:47:57 +0100[diff] [blame]567// Pooling 2D ]
568
569// [ Pooling 3D
570//MaxPooling
571ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling3dSize2x2x2Stride1x1x1, SimpleMaxPooling3dSize2x2x2Stride1x1x1Test)
572ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling3dSize2x2x2Stride1x1x1Uint8,
573 SimpleMaxPooling3dSize2x2x2Stride1x1x1Uint8Test)
574ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling3dSize2x2x2Stride1x1x1Int16,
575 SimpleMaxPooling3dSize2x2x2Stride1x1x1Int16Test)
576
577ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling3d, SimpleMaxPooling3dTest, DataLayout::NDHWC)
578ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling3dNCDHW, SimpleMaxPooling3dTest, DataLayout::NCDHW)
579ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling3dUint8, SimpleMaxPooling3dUint8Test, DataLayout::NDHWC)
580ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling3dInt16, SimpleMaxPooling3dInt16Test, DataLayout::NDHWC)
581ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling3dUint8NCDHW, SimpleMaxPooling3dUint8Test, DataLayout::NCDHW)
582ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaxPooling3dInt16NCDHW, SimpleMaxPooling3dInt16Test, DataLayout::NCDHW)
583
584ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleMaxPooling3d, IgnorePaddingSimpleMaxPooling3dTest)
585ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleMaxPooling3dUint8, IgnorePaddingSimpleMaxPooling3dUint8Test)
586ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleMaxPooling3dInt16, IgnorePaddingSimpleMaxPooling3dInt16Test)
587
588//AveragePooling
589ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleAveragePooling3d, SimpleAveragePooling3dTest, DataLayout::NDHWC)
590ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleAveragePooling3dNCDHW, SimpleAveragePooling3dTest, DataLayout::NCDHW)
591ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleAveragePooling3dUint8, SimpleAveragePooling3dUint8Test, DataLayout::NDHWC)
592ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleAveragePooling3dInt16, SimpleAveragePooling3dInt16Test, DataLayout::NDHWC)
593ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleAveragePooling3dUint8NCDHW, SimpleAveragePooling3dUint8Test, DataLayout::NCDHW)
594ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleAveragePooling3dInt16NCDHW, SimpleAveragePooling3dInt16Test, DataLayout::NCDHW)
595
Colm Donelan19bde172021-12-17 10:24:54 +0000[diff] [blame]596// Due to the size of the input each of these test cases takes about 3 minutes
597// to execute in an x86 environment. We'll disable but not remove them for the
598// moment.
599//ARMNN_AUTO_TEST_CASE_WITH_THF(LargeTensorsAveragePooling3d, LargeTensorsAveragePooling3dTest)
600//ARMNN_AUTO_TEST_CASE_WITH_THF(LargeTensorsAveragePooling3dUint8, LargeTensorsAveragePooling3dUint8Test)
601//ARMNN_AUTO_TEST_CASE_WITH_THF(LargeTensorsAveragePooling3dInt16, LargeTensorsAveragePooling3dInt16Test)
Tamás Nyíri7b885b32021-10-26 14:47:57 +0100[diff] [blame]602
603ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleAveragePooling3d, IgnorePaddingSimpleAveragePooling3dTest)
604ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleAveragePooling3dUint8, IgnorePaddingSimpleAveragePooling3dUint8Test)
605ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleAveragePooling3dInt16, IgnorePaddingSimpleAveragePooling3dInt16Test)
606
607//L2Pooling
608ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleL2Pooling3d, SimpleL2Pooling3dTest, DataLayout::NDHWC)
609ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleL2Pooling3dNCDHW, SimpleL2Pooling3dTest, DataLayout::NCDHW)
610ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleL2Pooling3dUint8, SimpleL2Pooling3dUint8Test, DataLayout::NDHWC)
611ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleL2Pooling3dInt16, SimpleL2Pooling3dInt16Test, DataLayout::NDHWC)
612ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleL2Pooling3dUint8NCDHW, SimpleL2Pooling3dUint8Test, DataLayout::NCDHW)
613ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleL2Pooling3dInt16NCDHW, SimpleL2Pooling3dInt16Test, DataLayout::NCDHW)
614
615ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleL2Pooling3d, IgnorePaddingSimpleL2Pooling3dTest)
616ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleL2Pooling3dUint8, IgnorePaddingSimpleL2Pooling3dUint8Test)
617ARMNN_AUTO_TEST_CASE_WITH_THF(IgnorePaddingSimpleL2Pooling3dInt16, IgnorePaddingSimpleL2Pooling3dInt16Test)
618
619//NonSquarePooling
620ARMNN_AUTO_TEST_CASE_WITH_THF(AsymmNonSquareMaxPooling3d, AsymmetricNonSquareMaxPooling3dTest)
621ARMNN_AUTO_TEST_CASE_WITH_THF(AsymmNonSquareMaxPooling3dUint8, AsymmetricNonSquareMaxPooling3dUint8Test)
622ARMNN_AUTO_TEST_CASE_WITH_THF(AsymmNonSquareMaxPooling3dInt16, AsymmetricNonSquareMaxPooling3dInt16Test)
623
624ARMNN_AUTO_TEST_CASE_WITH_THF(AsymmNonSquareAveragePooling3d, AsymmetricNonSquareAveragePooling3dTest)
625ARMNN_AUTO_TEST_CASE_WITH_THF(AsymmNonSquareAveragePooling3dUint8, AsymmetricNonSquareAveragePooling3dUint8Test)
626ARMNN_AUTO_TEST_CASE_WITH_THF(AsymmNonSquareAveragePooling3dInt16, AsymmetricNonSquareAveragePooling3dInt16Test)
627
628ARMNN_AUTO_TEST_CASE_WITH_THF(AsymmNonSquareL2Pooling3d, AsymmetricNonSquareL2Pooling3dTest)
629ARMNN_AUTO_TEST_CASE_WITH_THF(AsymmNonSquareL2Pooling3dUint8, AsymmetricNonSquareL2Pooling3dUint8Test)
630ARMNN_AUTO_TEST_CASE_WITH_THF(AsymmNonSquareL2Pooling3dInt16, AsymmetricNonSquareL2Pooling3dInt16Test)
631// Pooling 3D ]
Teresa Charlin0434df62019-06-06 13:40:35 +0100[diff] [blame]632
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]633
Teresa Charlin18515e22019-04-24 10:17:46 +0100[diff] [blame]634// Linear Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]635ARMNN_AUTO_TEST_CASE_WITH_THF(ConstantLinearActivation, ConstantLinearActivationTest)
636ARMNN_AUTO_TEST_CASE_WITH_THF(ConstantLinearActivationUint8, ConstantLinearActivationUint8Test)
637ARMNN_AUTO_TEST_CASE_WITH_THF(ConstantLinearActivationInt16, ConstantLinearActivationInt16Test)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]638
Kevin May09ca49c2019-10-09 12:37:34 +0100[diff] [blame]639// InstanceNormalization
Finn Williamsc43de6a2020-08-27 11:13:25 +0100[diff] [blame]640ARMNN_AUTO_TEST_CASE_WITH_THF(InstanceNormFloat32Nchw, InstanceNormFloat32Test, DataLayout::NCHW);
641ARMNN_AUTO_TEST_CASE_WITH_THF(InstanceNormFloat16Nchw, InstanceNormFloat16Test, DataLayout::NCHW);
Kevin May09ca49c2019-10-09 12:37:34 +0100[diff] [blame]642
Finn Williamsc43de6a2020-08-27 11:13:25 +0100[diff] [blame]643ARMNN_AUTO_TEST_CASE_WITH_THF(InstanceNormFloat32Nhwc, InstanceNormFloat32Test, DataLayout::NHWC);
644ARMNN_AUTO_TEST_CASE_WITH_THF(InstanceNormFloat16Nhwc, InstanceNormFloat16Test, DataLayout::NHWC);
Kevin May09ca49c2019-10-09 12:37:34 +0100[diff] [blame]645
Finn Williamsc43de6a2020-08-27 11:13:25 +0100[diff] [blame]646ARMNN_AUTO_TEST_CASE_WITH_THF(InstanceNormFloat32Nchw2, InstanceNormFloat32Test2, DataLayout::NCHW);
647ARMNN_AUTO_TEST_CASE_WITH_THF(InstanceNormFloat16Nchw2, InstanceNormFloat16Test2, DataLayout::NCHW);
Kevin May09ca49c2019-10-09 12:37:34 +0100[diff] [blame]648
Finn Williamsc43de6a2020-08-27 11:13:25 +0100[diff] [blame]649ARMNN_AUTO_TEST_CASE_WITH_THF(InstanceNormFloat32Nhwc2, InstanceNormFloat32Test2, DataLayout::NHWC);
650ARMNN_AUTO_TEST_CASE_WITH_THF(InstanceNormFloat16Nhwc2, InstanceNormFloat16Test2, DataLayout::NHWC);
Kevin May09ca49c2019-10-09 12:37:34 +0100[diff] [blame]651
Matteo Martincigh8e6f92d2018-10-18 08:45:39 +0100[diff] [blame]652// Normalization
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]653ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleNormalizationAcross, SimpleNormalizationAcrossTest)
654ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleNormalizationWithin, SimpleNormalizationWithinTest)
655ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleNormalizationAcrossNhwc, SimpleNormalizationAcrossNhwcTest)
Sadik Armagan483c8112021-06-01 09:24:52 +0100[diff] [blame]656ARMNN_AUTO_TEST_CASE_WITH_THF(AcrossChannelNormalization, AcrossChannelNormalizationTest)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]657
Narumol Prangnawarat65d30962019-03-14 11:55:03 +0000[diff] [blame]658// Softmax
Sadik Armagan56785c72020-08-27 12:57:20 +0100[diff] [blame]659ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSoftmaxBeta1, SimpleSoftmaxTest, 1.0f)
660ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSoftmaxBeta2, SimpleSoftmaxTest, 2.0f)
661ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSoftmaxBeta1Uint8, SimpleSoftmaxUint8Test, 1.0f)
662ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSoftmaxBeta2Uint8, SimpleSoftmaxUint8Test, 2.0f)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]663
Sadik Armagan56785c72020-08-27 12:57:20 +0100[diff] [blame]664ARMNN_AUTO_TEST_CASE_WITH_THF(Simple3dSoftmax, Simple3dSoftmaxTest, 1.0f)
665ARMNN_AUTO_TEST_CASE_WITH_THF(Simple3dSoftmaxUint8, Simple3dSoftmaxUint8Test, 1.0f)
Narumol Prangnawarat65d30962019-03-14 11:55:03 +0000[diff] [blame]666
Sadik Armagan56785c72020-08-27 12:57:20 +0100[diff] [blame]667ARMNN_AUTO_TEST_CASE_WITH_THF(Simple4dSoftmax, Simple4dSoftmaxTest, 1.0f)
668ARMNN_AUTO_TEST_CASE_WITH_THF(Simple4dSoftmaxUint8, Simple4dSoftmaxUint8Test, 1.0f)
Narumol Prangnawarat65d30962019-03-14 11:55:03 +0000[diff] [blame]669
Sadik Armagan56785c72020-08-27 12:57:20 +0100[diff] [blame]670ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSoftmaxFloat16, SimpleSoftmaxFloat16Test, 1.0f)
671ARMNN_AUTO_TEST_CASE_WITH_THF(Simple3dSoftmaxFloat16, Simple3dSoftmaxFloat16Test, 1.0f)
672ARMNN_AUTO_TEST_CASE_WITH_THF(Simple4dSoftmaxFloat16, Simple4dSoftmaxFloat16Test, 1.0f)
Matthew Jackson9bff1442019-09-12 09:08:23 +0100[diff] [blame]673
Sadik Armagan56785c72020-08-27 12:57:20 +0100[diff] [blame]674ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSoftmaxUint16, SimpleSoftmaxUint16Test, 1.0f)
675ARMNN_AUTO_TEST_CASE_WITH_THF(Simple3dSoftmaxUint16, Simple3dSoftmaxUint16Test, 1.0f)
676ARMNN_AUTO_TEST_CASE_WITH_THF(Simple4dSoftmaxUint16, Simple4dSoftmaxUint16Test, 1.0f)
nikraj01248683f2019-05-29 16:46:50 +0100[diff] [blame]677
Sadik Armagan56785c72020-08-27 12:57:20 +0100[diff] [blame]678ARMNN_AUTO_TEST_CASE_WITH_THF(Simple2dAxis0Softmax, SimpleAxisSoftmaxTest, 1.0f, 0)
679ARMNN_AUTO_TEST_CASE_WITH_THF(Simple2dAxis1Softmax, SimpleAxisSoftmaxTest, 1.0f, 1)
Francis Murtagh07f21212019-07-23 09:50:50 +0100[diff] [blame]680
Sadik Armagan56785c72020-08-27 12:57:20 +0100[diff] [blame]681ARMNN_AUTO_TEST_CASE_WITH_THF(Simple2dAxis0NegSoftmax, SimpleAxisSoftmaxTest, 1.0f, -2)
682ARMNN_AUTO_TEST_CASE_WITH_THF(Simple2dAxis1NegSoftmax, SimpleAxisSoftmaxTest, 1.0f, -1)
Francis Murtagh07f21212019-07-23 09:50:50 +0100[diff] [blame]683
Sadik Armagan56785c72020-08-27 12:57:20 +0100[diff] [blame]684ARMNN_AUTO_TEST_CASE_WITH_THF(Simple3dAxis0Softmax, Simple3dAxisSoftmaxTest, 1.0f, 0)
685ARMNN_AUTO_TEST_CASE_WITH_THF(Simple3dAxis1Softmax, Simple3dAxisSoftmaxTest, 1.0f, 1)
686ARMNN_AUTO_TEST_CASE_WITH_THF(Simple3dAxis2Softmax, Simple3dAxisSoftmaxTest, 1.0f, 2)
Francis Murtagh07f21212019-07-23 09:50:50 +0100[diff] [blame]687
Sadik Armagan56785c72020-08-27 12:57:20 +0100[diff] [blame]688ARMNN_AUTO_TEST_CASE_WITH_THF(Simple3dAxis0NegSoftmax, Simple3dAxisSoftmaxTest, 1.0f, -3)
689ARMNN_AUTO_TEST_CASE_WITH_THF(Simple3dAxis1NegSoftmax, Simple3dAxisSoftmaxTest, 1.0f, -2)
690ARMNN_AUTO_TEST_CASE_WITH_THF(Simple3dAxis2NegSoftmax, Simple3dAxisSoftmaxTest, 1.0f, -1)
Francis Murtagh07f21212019-07-23 09:50:50 +0100[diff] [blame]691
Sadik Armagan56785c72020-08-27 12:57:20 +0100[diff] [blame]692ARMNN_AUTO_TEST_CASE_WITH_THF(Simple4dAxis0Softmax, Simple4dAxisSoftmaxTest, 1.0f, 0)
693ARMNN_AUTO_TEST_CASE_WITH_THF(Simple4dAxis1Softmax, Simple4dAxisSoftmaxTest, 1.0f, 1)
694ARMNN_AUTO_TEST_CASE_WITH_THF(Simple4dAxis2Softmax, Simple4dAxisSoftmaxTest, 1.0f, 2)
695ARMNN_AUTO_TEST_CASE_WITH_THF(Simple4dAxis3Softmax, Simple4dAxisSoftmaxTest, 1.0f, 3)
Francis Murtagh07f21212019-07-23 09:50:50 +0100[diff] [blame]696
Sadik Armagan56785c72020-08-27 12:57:20 +0100[diff] [blame]697ARMNN_AUTO_TEST_CASE_WITH_THF(Simple4dAxis0NegSoftmax, Simple4dAxisSoftmaxTest, 1.0f, -4)
698ARMNN_AUTO_TEST_CASE_WITH_THF(Simple4dAxis1NegSoftmax, Simple4dAxisSoftmaxTest, 1.0f, -3)
699ARMNN_AUTO_TEST_CASE_WITH_THF(Simple4dAxis2NegSoftmax, Simple4dAxisSoftmaxTest, 1.0f, -2)
700ARMNN_AUTO_TEST_CASE_WITH_THF(Simple4dAxis3NegSoftmax, Simple4dAxisSoftmaxTest, 1.0f, -1)
Francis Murtagh07f21212019-07-23 09:50:50 +0100[diff] [blame]701
Teresa Charlin18515e22019-04-24 10:17:46 +0100[diff] [blame]702// Sigmoid Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]703ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSigmoid, SimpleSigmoidTest)
704ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSigmoidUint8, SimpleSigmoidUint8Test)
705ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSigmoidInt16, SimpleSigmoidInt16Test)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]706
Teresa Charlin18515e22019-04-24 10:17:46 +0100[diff] [blame]707// BoundedReLU Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]708ARMNN_AUTO_TEST_CASE_WITH_THF(ReLu1, BoundedReLuUpperAndLowerBoundTest)
709ARMNN_AUTO_TEST_CASE_WITH_THF(ReLu6, BoundedReLuUpperBoundOnlyTest)
710ARMNN_AUTO_TEST_CASE_WITH_THF(ReLu1Uint8, BoundedReLuUint8UpperAndLowerBoundTest)
711ARMNN_AUTO_TEST_CASE_WITH_THF(ReLu6Uint8, BoundedReLuUint8UpperBoundOnlyTest)
712ARMNN_AUTO_TEST_CASE_WITH_THF(BoundedReLuInt16, BoundedReLuInt16Test)
Teresa Charlin18515e22019-04-24 10:17:46 +0100[diff] [blame]713
714// ReLU Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]715ARMNN_AUTO_TEST_CASE_WITH_THF(ReLu, ReLuTest)
716ARMNN_AUTO_TEST_CASE_WITH_THF(ReLuUint8, ReLuUint8Test)
717ARMNN_AUTO_TEST_CASE_WITH_THF(ReLuInt16, ReLuInt16Test)
Teresa Charlin18515e22019-04-24 10:17:46 +0100[diff] [blame]718
719// SoftReLU Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]720ARMNN_AUTO_TEST_CASE_WITH_THF(SoftReLu, SoftReLuTest)
721ARMNN_AUTO_TEST_CASE_WITH_THF(SoftReLuUint8, SoftReLuUint8Test)
722ARMNN_AUTO_TEST_CASE_WITH_THF(SoftReLuInt16, SoftReLuInt16Test)
Teresa Charlin18515e22019-04-24 10:17:46 +0100[diff] [blame]723
konsof017f6db402019-06-07 15:15:58 +0100[diff] [blame]724
Teresa Charlin18515e22019-04-24 10:17:46 +0100[diff] [blame]725// LeakyReLU Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]726ARMNN_AUTO_TEST_CASE_WITH_THF(LeakyReLu, LeakyReLuTest)
727ARMNN_AUTO_TEST_CASE_WITH_THF(LeakyReLuUint8, LeakyReLuUint8Test)
728ARMNN_AUTO_TEST_CASE_WITH_THF(LeakyReLuInt16, LeakyReLuInt16Test)
Teresa Charlin18515e22019-04-24 10:17:46 +0100[diff] [blame]729
730// Abs Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]731ARMNN_AUTO_TEST_CASE_WITH_THF(Abs, AbsTest)
732ARMNN_AUTO_TEST_CASE_WITH_THF(AbsUint8, AbsUint8Test)
733ARMNN_AUTO_TEST_CASE_WITH_THF(AbsInt16, AbsInt16Test)
Teresa Charlin18515e22019-04-24 10:17:46 +0100[diff] [blame]734
735// Sqrt Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]736ARMNN_AUTO_TEST_CASE_WITH_THF(Sqrt, SqrtTest)
737ARMNN_AUTO_TEST_CASE_WITH_THF(SqrtNN, SqrtNNTest)
738ARMNN_AUTO_TEST_CASE_WITH_THF(SqrtUint8, SqrtUint8Test)
739ARMNN_AUTO_TEST_CASE_WITH_THF(SqrtInt16, SqrtInt16Test)
Teresa Charlin18515e22019-04-24 10:17:46 +0100[diff] [blame]740
741// Square Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]742ARMNN_AUTO_TEST_CASE_WITH_THF(Square, SquareTest)
743ARMNN_AUTO_TEST_CASE_WITH_THF(SquareUint8, SquareUint8Test)
744ARMNN_AUTO_TEST_CASE_WITH_THF(SquareInt16, SquareInt16Test)
Teresa Charlin18515e22019-04-24 10:17:46 +0100[diff] [blame]745
746// Tanh Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]747ARMNN_AUTO_TEST_CASE_WITH_THF(Tanh, TanhTest)
748ARMNN_AUTO_TEST_CASE_WITH_THF(TanhUint8, TanhUint8Test)
749ARMNN_AUTO_TEST_CASE_WITH_THF(TanhInt16, TanhInt16Test)
Teresa Charlin18515e22019-04-24 10:17:46 +0100[diff] [blame]750
David Monahan3b3c3812020-02-25 09:03:29 +0000[diff] [blame]751// Elu Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]752ARMNN_AUTO_TEST_CASE_WITH_THF(Elu, EluTest)
753ARMNN_AUTO_TEST_CASE_WITH_THF(EluUint8, EluUint8Test)
754ARMNN_AUTO_TEST_CASE_WITH_THF(EluInt16, EluInt16Test)
Colm Donelan03fbeaf2020-02-26 15:39:23 +0000[diff] [blame]755// HardSwish Activation
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]756ARMNN_AUTO_TEST_CASE_WITH_THF(HardSwish, HardSwishTest)
757ARMNN_AUTO_TEST_CASE_WITH_THF(HardSwishUint8, HardSwishUint8Test)
758ARMNN_AUTO_TEST_CASE_WITH_THF(HardSwishInt16, HardSwishInt16Test)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]759
Francis Murtagh46c09d02019-05-28 08:15:28 +0100[diff] [blame]760// Fully Connected
Finn Williams7faf9a82020-08-27 10:37:36 +0100[diff] [blame]761ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleFullyConnected, FullyConnectedFloat32Test, false, false)
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000[diff] [blame]762ARMNN_AUTO_TEST_CASE_WITH_THF(FullyConnectedUint8, FullyConnectedTest<DataType::QAsymmU8>, false, true)
763ARMNN_AUTO_TEST_CASE_WITH_THF(FullyConnectedQSymm16, FullyConnectedTest<DataType::QSymmS16>, false, true)
Finn Williams7faf9a82020-08-27 10:37:36 +0100[diff] [blame]764ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleFullyConnectedWithBias, FullyConnectedFloat32Test, true, false)
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000[diff] [blame]765ARMNN_AUTO_TEST_CASE_WITH_THF(FullyConnectedBiasedUint8, FullyConnectedTest<DataType::QAsymmU8>, true, true)
766ARMNN_AUTO_TEST_CASE_WITH_THF(FullyConnectedBiasedQSymm16, FullyConnectedTest<DataType::QSymmS16>, true, true)
Finn Williams7faf9a82020-08-27 10:37:36 +0100[diff] [blame]767ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleFullyConnectedWithTranspose, FullyConnectedFloat32Test, false, true)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]768
Finn Williams7faf9a82020-08-27 10:37:36 +0100[diff] [blame]769ARMNN_AUTO_TEST_CASE_WITH_THF(FullyConnectedLarge, FullyConnectedLargeTest, false)
770ARMNN_AUTO_TEST_CASE_WITH_THF(FullyConnectedLargeTransposed, FullyConnectedLargeTest, true)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]771
Sadik Armaganf0a6dec2021-03-25 07:46:55 +0000[diff] [blame]772
773ARMNN_AUTO_TEST_CASE_WITH_THF(FullyConnectedWeightsAsInputsUint8,
774 FullyConnectedTest<DataType::QAsymmU8>,
775 false,
776 false)
777
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]778// Splitter
Sadik Armagan5e719842020-08-28 12:17:33 +0100[diff] [blame]779ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSplitterFloat32, SplitterFloat32Test)
780ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSplitterFloat16, SplitterFloat16Test)
781ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSplitterUint8, SplitterUint8Test)
782ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSplitterInt16, SplitterInt16Test)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]783
Sadik Armagan5e719842020-08-28 12:17:33 +0100[diff] [blame]784ARMNN_AUTO_TEST_CASE_WITH_THF(CopyViaSplitterFloat32, CopyViaSplitterFloat32Test)
785ARMNN_AUTO_TEST_CASE_WITH_THF(CopyViaSplitterFloat16, CopyViaSplitterFloat16Test)
786ARMNN_AUTO_TEST_CASE_WITH_THF(CopyViaSplitterUint8, CopyViaSplitterUint8Test)
787ARMNN_AUTO_TEST_CASE_WITH_THF(CopyViaSplitterInt16, CopyViaSplitterInt16Test)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]788
Jim Flynn4ed6c832019-05-20 11:02:46 +0100[diff] [blame]789// Concat
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]790ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConcat, ConcatTest)
791ARMNN_AUTO_TEST_CASE_WITH_THF(ConcatBFloat16, ConcatBFloat16Test)
792ARMNN_AUTO_TEST_CASE_WITH_THF(ConcatFloat16, ConcatFloat16Test)
793ARMNN_AUTO_TEST_CASE_WITH_THF(ConcatUint8, ConcatUint8Test)
794ARMNN_AUTO_TEST_CASE_WITH_THF(ConcatUint8DifferentQParams, ConcatUint8DifferentQParamsTest)
795ARMNN_AUTO_TEST_CASE_WITH_THF(ConcatUint16, ConcatUint16Test)
796ARMNN_AUTO_TEST_CASE_WITH_THF(ConcatUint8DifferentInputOutputQParam,
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]797 ConcatDifferentInputOutputQParamTest<DataType::QAsymmU8>, true)
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]798ARMNN_AUTO_TEST_CASE_WITH_THF(ConcatInt16DifferentInputOutputQParam,
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]799 ConcatDifferentInputOutputQParamTest<DataType::QSymmS16>, true)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]800
801// Add
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]802ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleAdd, AdditionTest)
803ARMNN_AUTO_TEST_CASE_WITH_THF(Add5d, Addition5dTest)
804ARMNN_AUTO_TEST_CASE_WITH_THF(AddBroadcast1Element, AdditionBroadcast1ElementTest)
805ARMNN_AUTO_TEST_CASE_WITH_THF(AddBroadcast, AdditionBroadcastTest)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]806
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]807ARMNN_AUTO_TEST_CASE_WITH_THF(AdditionUint8, AdditionUint8Test)
808ARMNN_AUTO_TEST_CASE_WITH_THF(AddBroadcastUint8, AdditionBroadcastUint8Test)
809ARMNN_AUTO_TEST_CASE_WITH_THF(AddBroadcast1ElementUint8, AdditionBroadcast1ElementUint8Test)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]810
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]811ARMNN_AUTO_TEST_CASE_WITH_THF(AdditionInt16, AdditionInt16Test)
812ARMNN_AUTO_TEST_CASE_WITH_THF(AddBroadcastInt16, AdditionBroadcastInt16Test)
813ARMNN_AUTO_TEST_CASE_WITH_THF(AddBroadcast1ElementInt16, AdditionBroadcast1ElementInt16Test)
Sadik Armagan2999a022019-04-09 14:20:12 +0100[diff] [blame]814
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]815ARMNN_AUTO_TEST_CASE_WITH_THF(AdditionInt32, AdditionInt32Test)
816ARMNN_AUTO_TEST_CASE_WITH_THF(AddBroadcastInt32, AdditionBroadcastInt32Test)
817ARMNN_AUTO_TEST_CASE_WITH_THF(AddBroadcast1ElementInt32, AdditionBroadcast1ElementInt32Test)
Teresa Charlinecb6b8e2020-05-22 18:08:23 +0100[diff] [blame]818
David Beckf195f032018-09-06 16:46:34 +0100[diff] [blame]819// Sub
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]820ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSub, SubtractionTest)
821ARMNN_AUTO_TEST_CASE_WITH_THF(SubBroadcast1Element, SubtractionBroadcast1ElementTest)
822ARMNN_AUTO_TEST_CASE_WITH_THF(SubBroadcast, SubtractionBroadcastTest)
David Beckf195f032018-09-06 16:46:34 +0100[diff] [blame]823
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]824ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSubFloat16, SubtractionTest)
825ARMNN_AUTO_TEST_CASE_WITH_THF(SubBroadcast1ElementFloat16, SubtractionBroadcast1ElementTest)
826ARMNN_AUTO_TEST_CASE_WITH_THF(SubBroadcastFloat16, SubtractionBroadcastTest)
Matthew Jackson9bff1442019-09-12 09:08:23 +0100[diff] [blame]827
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]828ARMNN_AUTO_TEST_CASE_WITH_THF(SubtractionUint8, SubtractionUint8Test)
829ARMNN_AUTO_TEST_CASE_WITH_THF(SubBroadcastUint8, SubtractionBroadcastUint8Test)
830ARMNN_AUTO_TEST_CASE_WITH_THF(SubBroadcast1ElementUint8, SubtractionBroadcast1ElementUint8Test)
David Beckf195f032018-09-06 16:46:34 +0100[diff] [blame]831
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]832ARMNN_AUTO_TEST_CASE_WITH_THF(SubtractionInt16, SubtractionInt16Test)
833ARMNN_AUTO_TEST_CASE_WITH_THF(SubBroadcastInt16, SubtractionBroadcastInt16Test)
834ARMNN_AUTO_TEST_CASE_WITH_THF(SubBroadcast1ElementInt16, SubtractionBroadcast1ElementInt16Test)
Sadik Armagan2999a022019-04-09 14:20:12 +0100[diff] [blame]835
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]836ARMNN_AUTO_TEST_CASE_WITH_THF(SubtractionInt32, SubtractionInt32Test)
837ARMNN_AUTO_TEST_CASE_WITH_THF(SubBroadcastInt32, SubtractionBroadcastInt32Test)
838ARMNN_AUTO_TEST_CASE_WITH_THF(SubBroadcast1ElementInt32, SubtractionBroadcast1ElementInt32Test)
Teresa Charlinecb6b8e2020-05-22 18:08:23 +0100[diff] [blame]839
Francis Murtaghe7a86a42018-08-29 12:42:10 +0100[diff] [blame]840// Div
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]841ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleDivision, DivisionTest)
842ARMNN_AUTO_TEST_CASE_WITH_THF(DivisionByZero, DivisionByZeroTest)
843ARMNN_AUTO_TEST_CASE_WITH_THF(DivisionBroadcast1Element, DivisionBroadcast1ElementTest)
844ARMNN_AUTO_TEST_CASE_WITH_THF(DivisionBroadcast1DVector, DivisionBroadcast1DVectorTest)
Matthew Jackson9bff1442019-09-12 09:08:23 +0100[diff] [blame]845
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]846ARMNN_AUTO_TEST_CASE_WITH_THF(DivisionFloat16, DivisionFloat16Test)
847ARMNN_AUTO_TEST_CASE_WITH_THF(DivisionFloat16Broadcast1Element, DivisionBroadcast1ElementFloat16Test)
848ARMNN_AUTO_TEST_CASE_WITH_THF(DivisionFloat16Broadcast1DVector, DivisionBroadcast1DVectorFloat16Test)
Matthew Jackson9bff1442019-09-12 09:08:23 +0100[diff] [blame]849
David Beck5cd01f32018-09-12 16:00:08 +0100[diff] [blame]850// NOTE: division by zero for quantized div needs more attention
851// see IVGCVSW-1849
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]852ARMNN_AUTO_TEST_CASE_WITH_THF(DivisionUint8, DivisionUint8Test)
853ARMNN_AUTO_TEST_CASE_WITH_THF(DivisionUint8Broadcast1Element, DivisionBroadcast1ElementUint8Test)
854ARMNN_AUTO_TEST_CASE_WITH_THF(DivisionUint8Broadcast1DVector, DivisionBroadcast1DVectorUint8Test)
Francis Murtaghe7a86a42018-08-29 12:42:10 +0100[diff] [blame]855
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]856ARMNN_AUTO_TEST_CASE_WITH_THF(DivisionInt16, DivisionInt16Test)
857ARMNN_AUTO_TEST_CASE_WITH_THF(DivisionInt16Broadcast1Element, DivisionBroadcast1ElementInt16Test)
858ARMNN_AUTO_TEST_CASE_WITH_THF(DivisionInt16Broadcast1DVector, DivisionBroadcast1DVectorInt16Test)
Sadik Armagan2999a022019-04-09 14:20:12 +0100[diff] [blame]859
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]860ARMNN_AUTO_TEST_CASE_WITH_THF(DivisionInt32, DivisionInt32Test)
861ARMNN_AUTO_TEST_CASE_WITH_THF(DivisionInt32Broadcast1Element, DivisionBroadcast1ElementInt32Test)
862ARMNN_AUTO_TEST_CASE_WITH_THF(DivisionInt32Broadcast1DVector, DivisionBroadcast1DVectorInt32Test)
Teresa Charlinecb6b8e2020-05-22 18:08:23 +0100[diff] [blame]863
FrancisMurtagh30cdfca2018-12-18 12:57:35 +0000[diff] [blame]864// Equal
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]865ARMNN_AUTO_TEST_CASE_WITH_THF(EqualSimple, EqualSimpleTest)
866ARMNN_AUTO_TEST_CASE_WITH_THF(EqualBroadcast1Element, EqualBroadcast1ElementTest)
867ARMNN_AUTO_TEST_CASE_WITH_THF(EqualBroadcast1dVector, EqualBroadcast1dVectorTest)
Aron Virginas-Tar3bc00ec2019-10-18 15:42:58 +0100[diff] [blame]868
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]869ARMNN_AUTO_TEST_CASE_WITH_THF(EqualSimpleFloat16, EqualSimpleFloat16Test)
870ARMNN_AUTO_TEST_CASE_WITH_THF(EqualBroadcast1ElementFloat16, EqualBroadcast1ElementFloat16Test)
871ARMNN_AUTO_TEST_CASE_WITH_THF(EqualBroadcast1dVectorFloat16, EqualBroadcast1dVectorFloat16Test)
Aron Virginas-Tar3bc00ec2019-10-18 15:42:58 +0100[diff] [blame]872
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]873ARMNN_AUTO_TEST_CASE_WITH_THF(EqualSimpleUint8, EqualSimpleUint8Test)
874ARMNN_AUTO_TEST_CASE_WITH_THF(EqualBroadcast1ElementUint8, EqualBroadcast1ElementUint8Test)
875ARMNN_AUTO_TEST_CASE_WITH_THF(EqualBroadcast1dVectorUint8, EqualBroadcast1dVectorUint8Test)
FrancisMurtagh30cdfca2018-12-18 12:57:35 +0000[diff] [blame]876
FrancisMurtagh878f0232018-12-19 10:56:15 +0000[diff] [blame]877// Greater
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]878ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterSimple, GreaterSimpleTest)
879ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterBroadcast1Element, GreaterBroadcast1ElementTest)
880ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterBroadcast1dVector, GreaterBroadcast1dVectorTest)
Aron Virginas-Tar3bc00ec2019-10-18 15:42:58 +0100[diff] [blame]881
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]882ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterSimpleFloat16, GreaterSimpleFloat16Test)
883ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterBroadcast1ElementFloat16, GreaterBroadcast1ElementFloat16Test)
884ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterBroadcast1dVectorFloat16, GreaterBroadcast1dVectorFloat16Test)
Aron Virginas-Tar3bc00ec2019-10-18 15:42:58 +0100[diff] [blame]885
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]886ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterSimpleUint8, GreaterSimpleUint8Test)
887ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterBroadcast1ElementUint8, GreaterBroadcast1ElementUint8Test)
888ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterBroadcast1dVectorUint8, GreaterBroadcast1dVectorUint8Test)
Aron Virginas-Tar3bc00ec2019-10-18 15:42:58 +0100[diff] [blame]889
890// GreaterOrEqual
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]891ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterOrEqualSimple, GreaterOrEqualSimpleTest)
892ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterOrEqualBroadcast1Element, GreaterOrEqualBroadcast1ElementTest)
893ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterOrEqualBroadcast1dVector, GreaterOrEqualBroadcast1dVectorTest)
Aron Virginas-Tar3bc00ec2019-10-18 15:42:58 +0100[diff] [blame]894
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]895ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterOrEqualSimpleFloat16, GreaterOrEqualSimpleFloat16Test)
896ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterOrEqualBroadcast1ElementFloat16, GreaterOrEqualBroadcast1ElementFloat16Test)
897ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterOrEqualBroadcast1dVectorFloat16, GreaterOrEqualBroadcast1dVectorFloat16Test)
Aron Virginas-Tar3bc00ec2019-10-18 15:42:58 +0100[diff] [blame]898
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]899ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterOrEqualSimpleUint8, GreaterOrEqualSimpleUint8Test)
900ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterOrEqualBroadcast1ElementUint8, GreaterOrEqualBroadcast1ElementUint8Test)
901ARMNN_AUTO_TEST_CASE_WITH_THF(GreaterOrEqualBroadcast1dVectorUint8, GreaterOrEqualBroadcast1dVectorUint8Test)
Aron Virginas-Tar3bc00ec2019-10-18 15:42:58 +0100[diff] [blame]902
903// Less
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]904ARMNN_AUTO_TEST_CASE_WITH_THF(LessSimple, LessSimpleTest)
905ARMNN_AUTO_TEST_CASE_WITH_THF(LessBroadcast1Element, LessBroadcast1ElementTest)
906ARMNN_AUTO_TEST_CASE_WITH_THF(LessBroadcast1dVector, LessBroadcast1dVectorTest)
Aron Virginas-Tar3bc00ec2019-10-18 15:42:58 +0100[diff] [blame]907
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]908ARMNN_AUTO_TEST_CASE_WITH_THF(LessSimpleFloat16, LessSimpleFloat16Test)
909ARMNN_AUTO_TEST_CASE_WITH_THF(LessBroadcast1ElementFloat16, LessBroadcast1ElementFloat16Test)
910ARMNN_AUTO_TEST_CASE_WITH_THF(LessBroadcast1dVectorFloat16, LessBroadcast1dVectorFloat16Test)
Aron Virginas-Tar3bc00ec2019-10-18 15:42:58 +0100[diff] [blame]911
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]912ARMNN_AUTO_TEST_CASE_WITH_THF(LessSimpleUint8, LessSimpleUint8Test)
913ARMNN_AUTO_TEST_CASE_WITH_THF(LessBroadcast1ElementUint8, LessBroadcast1ElementUint8Test)
914ARMNN_AUTO_TEST_CASE_WITH_THF(LessBroadcast1dVectorUint8, LessBroadcast1dVectorUint8Test)
Aron Virginas-Tar3bc00ec2019-10-18 15:42:58 +0100[diff] [blame]915
Teresa Charlin2b030d92020-03-27 16:40:56 +0000[diff] [blame]916// LessOrEqual
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]917ARMNN_AUTO_TEST_CASE_WITH_THF(LessOrEqualSimple, LessOrEqualSimpleTest)
918ARMNN_AUTO_TEST_CASE_WITH_THF(LessOrEqualBroadcast1Element, LessOrEqualBroadcast1ElementTest)
919ARMNN_AUTO_TEST_CASE_WITH_THF(LessOrEqualBroadcast1dVector, LessOrEqualBroadcast1dVectorTest)
Aron Virginas-Tar3bc00ec2019-10-18 15:42:58 +0100[diff] [blame]920
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]921ARMNN_AUTO_TEST_CASE_WITH_THF(LessOrEqualSimpleFloat16, LessOrEqualSimpleFloat16Test)
922ARMNN_AUTO_TEST_CASE_WITH_THF(LessOrEqualBroadcast1ElementFloat16, LessOrEqualBroadcast1ElementFloat16Test)
923ARMNN_AUTO_TEST_CASE_WITH_THF(LessOrEqualBroadcast1dVectorFloat16, LessOrEqualBroadcast1dVectorFloat16Test)
Aron Virginas-Tar3bc00ec2019-10-18 15:42:58 +0100[diff] [blame]924
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]925ARMNN_AUTO_TEST_CASE_WITH_THF(LessOrEqualSimpleUint8, LessOrEqualSimpleUint8Test)
926ARMNN_AUTO_TEST_CASE_WITH_THF(LessOrEqualBroadcast1ElementUint8, LessOrEqualBroadcast1ElementUint8Test)
927ARMNN_AUTO_TEST_CASE_WITH_THF(LessOrEqualBroadcast1dVectorUint8, LessOrEqualBroadcast1dVectorUint8Test)
Aron Virginas-Tar3bc00ec2019-10-18 15:42:58 +0100[diff] [blame]928
929// NotEqual
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]930ARMNN_AUTO_TEST_CASE_WITH_THF(NotEqualSimple, NotEqualSimpleTest)
931ARMNN_AUTO_TEST_CASE_WITH_THF(NotEqualBroadcast1Element, NotEqualBroadcast1ElementTest)
932ARMNN_AUTO_TEST_CASE_WITH_THF(NotEqualBroadcast1dVector, NotEqualBroadcast1dVectorTest)
Aron Virginas-Tar3bc00ec2019-10-18 15:42:58 +0100[diff] [blame]933
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]934ARMNN_AUTO_TEST_CASE_WITH_THF(NotEqualSimpleFloat16, NotEqualSimpleFloat16Test)
935ARMNN_AUTO_TEST_CASE_WITH_THF(NotEqualBroadcast1ElementFloat16, NotEqualBroadcast1ElementFloat16Test)
936ARMNN_AUTO_TEST_CASE_WITH_THF(NotEqualBroadcast1dVectorFloat16, NotEqualBroadcast1dVectorFloat16Test)
Aron Virginas-Tar3bc00ec2019-10-18 15:42:58 +0100[diff] [blame]937
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]938ARMNN_AUTO_TEST_CASE_WITH_THF(NotEqualSimpleUint8, NotEqualSimpleUint8Test)
939ARMNN_AUTO_TEST_CASE_WITH_THF(NotEqualBroadcast1ElementUint8, NotEqualBroadcast1ElementUint8Test)
940ARMNN_AUTO_TEST_CASE_WITH_THF(NotEqualBroadcast1dVectorUint8, NotEqualBroadcast1dVectorUint8Test)
FrancisMurtagh878f0232018-12-19 10:56:15 +0000[diff] [blame]941
Éanna Ó Catháin20e58802018-12-04 10:29:06 +0000[diff] [blame]942// Max
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]943ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMaximum, MaximumSimpleTest)
944ARMNN_AUTO_TEST_CASE_WITH_THF(MaximumBroadcast1Element, MaximumBroadcast1ElementTest)
945ARMNN_AUTO_TEST_CASE_WITH_THF(MaximumBroadcast1DVector, MaximumBroadcast1DVectorTest)
946ARMNN_AUTO_TEST_CASE_WITH_THF(MaximumFloat16, MaximumFloat16Test)
947ARMNN_AUTO_TEST_CASE_WITH_THF(MaximumBroadcast1ElementFloat16, MaximumBroadcast1ElementFloat16Test)
948ARMNN_AUTO_TEST_CASE_WITH_THF(MaximumBroadcast1DVectorFloat16, MaximumBroadcast1DVectorFloat16Test)
949ARMNN_AUTO_TEST_CASE_WITH_THF(MaximumUint8, MaximumUint8Test)
950ARMNN_AUTO_TEST_CASE_WITH_THF(MaximumBroadcast1ElementUint8, MaximumBroadcast1ElementUint8Test)
951ARMNN_AUTO_TEST_CASE_WITH_THF(MaximumBroadcast1DVectorUint8, MaximumBroadcast1DVectorUint8Test)
952ARMNN_AUTO_TEST_CASE_WITH_THF(MaximumInt16, MaximumInt16Test)
953ARMNN_AUTO_TEST_CASE_WITH_THF(MaximumBroadcast1ElementInt16, MaximumBroadcast1ElementInt16Test)
954ARMNN_AUTO_TEST_CASE_WITH_THF(MaximumBroadcast1DVectorInt16, MaximumBroadcast1DVectorInt16Test)
955ARMNN_AUTO_TEST_CASE_WITH_THF(MaximumInt32, MaximumInt32Test)
956ARMNN_AUTO_TEST_CASE_WITH_THF(MaximumBroadcast1ElementInt32, MaximumBroadcast1ElementInt32Test)
957ARMNN_AUTO_TEST_CASE_WITH_THF(MaximumBroadcast1DVectorInt32, MaximumBroadcast1DVectorInt32Test)
Éanna Ó Catháinde705582018-12-03 13:04:22 +0000[diff] [blame]958
Éanna Ó Catháin20e58802018-12-04 10:29:06 +0000[diff] [blame]959// Min
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]960ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMinimum1, MinimumBroadcast1ElementTest1)
961ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMinimum2, MinimumBroadcast1ElementTest2)
962ARMNN_AUTO_TEST_CASE_WITH_THF(Minimum1DVectorUint8, MinimumBroadcast1DVectorUint8Test)
963ARMNN_AUTO_TEST_CASE_WITH_THF(MinimumFloat16, MinimumFloat16Test)
964ARMNN_AUTO_TEST_CASE_WITH_THF(MinimumBroadcast1ElementFloat16, MinimumBroadcast1ElementFloat16Test)
965ARMNN_AUTO_TEST_CASE_WITH_THF(MinimumBroadcast1DVectorFloat16, MinimumBroadcast1DVectorFloat16Test)
966ARMNN_AUTO_TEST_CASE_WITH_THF(MinimumInt16, MinimumInt16Test)
967ARMNN_AUTO_TEST_CASE_WITH_THF(MinimumBroadcast1ElementInt16, MinimumBroadcast1ElementInt16Test)
968ARMNN_AUTO_TEST_CASE_WITH_THF(MinimumBroadcast1DVectorInt16, MinimumBroadcast1DVectorInt16Test)
969ARMNN_AUTO_TEST_CASE_WITH_THF(MinimumInt32, MinimumInt32Test)
970ARMNN_AUTO_TEST_CASE_WITH_THF(MinimumBroadcast1ElementInt32, MinimumBroadcast1ElementInt32Test)
971ARMNN_AUTO_TEST_CASE_WITH_THF(MinimumBroadcast1DVectorInt32, MinimumBroadcast1DVectorInt32Test)
Éanna Ó Catháin20e58802018-12-04 10:29:06 +0000[diff] [blame]972
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]973// Mul
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]974ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleMultiplication, MultiplicationTest)
975ARMNN_AUTO_TEST_CASE_WITH_THF(MultiplicationBroadcast1Element, MultiplicationBroadcast1ElementTest)
976ARMNN_AUTO_TEST_CASE_WITH_THF(MultiplicationBroadcast1DVector, MultiplicationBroadcast1DVectorTest)
977ARMNN_AUTO_TEST_CASE_WITH_THF(MultiplicationUint8, MultiplicationUint8Test)
978ARMNN_AUTO_TEST_CASE_WITH_THF(MultiplicationBroadcast1ElementUint8, MultiplicationBroadcast1ElementUint8Test)
979ARMNN_AUTO_TEST_CASE_WITH_THF(MultiplicationBroadcast1DVectorUint8, MultiplicationBroadcast1DVectorUint8Test)
980ARMNN_AUTO_TEST_CASE_WITH_THF(MultiplicationInt16, MultiplicationInt16Test)
981ARMNN_AUTO_TEST_CASE_WITH_THF(MultiplicationBroadcast1ElementInt16, MultiplicationBroadcast1ElementInt16Test)
982ARMNN_AUTO_TEST_CASE_WITH_THF(MultiplicationBroadcast1DVectorInt16, MultiplicationBroadcast1DVectorInt16Test)
983ARMNN_AUTO_TEST_CASE_WITH_THF(MultiplicationInt32, MultiplicationInt32Test)
984ARMNN_AUTO_TEST_CASE_WITH_THF(MultiplicationBroadcast1ElementInt32, MultiplicationBroadcast1ElementInt32Test)
985ARMNN_AUTO_TEST_CASE_WITH_THF(MultiplicationBroadcast1DVectorInt32, MultiplicationBroadcast1DVectorInt32Test)
986ARMNN_AUTO_TEST_CASE_WITH_THF(Multiplication5d, Multiplication5dTest)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]987
988// Batch Norm
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]989ARMNN_AUTO_TEST_CASE_WITH_THF(BatchNormFloat32, BatchNormFloat32Test)
990ARMNN_AUTO_TEST_CASE_WITH_THF(BatchNormFloat32Nhwc, BatchNormFloat32NhwcTest)
991ARMNN_AUTO_TEST_CASE_WITH_THF(BatchNormFloat16, BatchNormFloat16Test)
992ARMNN_AUTO_TEST_CASE_WITH_THF(BatchNormFloat16Nhwc, BatchNormFloat16NhwcTest)
993ARMNN_AUTO_TEST_CASE_WITH_THF(BatchNormUint8, BatchNormUint8Test)
994ARMNN_AUTO_TEST_CASE_WITH_THF(BatchNormUint8Nhwc, BatchNormUint8NhwcTest)
995ARMNN_AUTO_TEST_CASE_WITH_THF(BatchNormInt16, BatchNormInt16Test)
996ARMNN_AUTO_TEST_CASE_WITH_THF(BatchNormInt16Nhwc, BatchNormInt16NhwcTest)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]997
Finn Williams2605b232020-06-10 15:53:46 +0100[diff] [blame]998// Rank
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]999ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize1Float16, RankDimSize1Test<DataType::Float16>)
1000ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize1Float32, RankDimSize1Test<DataType::Float32>)
1001ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize1QAsymmU8, RankDimSize1Test<DataType::QAsymmU8>)
1002ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize1Signed32, RankDimSize1Test<DataType::Signed32>)
1003ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize1QSymmS16, RankDimSize1Test<DataType::QSymmS16>)
1004ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize1QSymmS8, RankDimSize1Test<DataType::QSymmS8>)
1005ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize1QAsymmS8, RankDimSize1Test<DataType::QAsymmS8>)
1006ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize1BFloat16, RankDimSize1Test<DataType::BFloat16>)
Finn Williams2605b232020-06-10 15:53:46 +0100[diff] [blame]1007
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]1008ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize2Float16, RankDimSize2Test<DataType::Float16>)
1009ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize2Float32, RankDimSize2Test<DataType::Float32>)
1010ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize2QAsymmU8, RankDimSize2Test<DataType::QAsymmU8>)
1011ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize2Signed32, RankDimSize2Test<DataType::Signed32>)
1012ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize2QSymmS16, RankDimSize2Test<DataType::QSymmS16>)
1013ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize2QSymmS8, RankDimSize2Test<DataType::QSymmS8>)
1014ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize2QAsymmS8, RankDimSize2Test<DataType::QAsymmS8>)
1015ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize2BFloat16, RankDimSize2Test<DataType::BFloat16>)
Finn Williams2605b232020-06-10 15:53:46 +0100[diff] [blame]1016
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]1017ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize3Float16, RankDimSize3Test<DataType::Float16>)
1018ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize3Float32, RankDimSize3Test<DataType::Float32>)
1019ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize3QAsymmU8, RankDimSize3Test<DataType::QAsymmU8>)
1020ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize3Signed32, RankDimSize3Test<DataType::Signed32>)
1021ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize3QSymmS16, RankDimSize3Test<DataType::QSymmS16>)
1022ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize3QSymmS8, RankDimSize3Test<DataType::QSymmS8>)
1023ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize3QAsymmS8, RankDimSize3Test<DataType::QAsymmS8>)
1024ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize3BFloat16, RankDimSize3Test<DataType::BFloat16>)
Finn Williams2605b232020-06-10 15:53:46 +0100[diff] [blame]1025
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]1026ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize4Float16, RankDimSize4Test<DataType::Float16>)
1027ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize4Float32, RankDimSize4Test<DataType::Float32>)
1028ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize4QAsymmU8, RankDimSize4Test<DataType::QAsymmU8>)
1029ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize4Signed32, RankDimSize4Test<DataType::Signed32>)
1030ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize4QSymmS16, RankDimSize4Test<DataType::QSymmS16>)
1031ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize4QSymmS8, RankDimSize4Test<DataType::QSymmS8>)
1032ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize4QAsymmS8, RankDimSize4Test<DataType::QAsymmS8>)
1033ARMNN_AUTO_TEST_CASE_WITH_THF(RankDimSize4BFloat16, RankDimSize4Test<DataType::BFloat16>)
Finn Williams2605b232020-06-10 15:53:46 +0100[diff] [blame]1034
James Conroy59540822018-10-11 12:39:05 +0100[diff] [blame]1035// Resize Bilinear - NCHW
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]1036ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeBilinear,
1037 SimpleResizeBilinearTest<DataType::Float32>,
1038 DataLayout::NCHW)
1039ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeBilinearFloat16,
1040 SimpleResizeBilinearTest<DataType::Float16>,
1041 DataLayout::NCHW)
1042ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeBilinearInt8,
1043 SimpleResizeBilinearTest<DataType::QAsymmS8>,
1044 DataLayout::NCHW)
1045ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeBilinearUint8,
1046 SimpleResizeBilinearTest<DataType::QAsymmU8>,
1047 DataLayout::NCHW)
1048ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeBilinearUint16,
1049 SimpleResizeBilinearTest<DataType::QSymmS16>,
1050 DataLayout::NCHW)
1051ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearNop,
1052 ResizeBilinearNopTest<DataType::Float32>,
1053 DataLayout::NCHW)
1054ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearNopFloat16,
1055 ResizeBilinearNopTest<DataType::Float16>,
1056 DataLayout::NCHW)
1057ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearNopInt8,
1058 ResizeBilinearNopTest<DataType::QAsymmS8>,
1059 DataLayout::NCHW)
1060ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearNopUint8,
1061 ResizeBilinearNopTest<DataType::QAsymmU8>,
1062 DataLayout::NCHW)
1063ARMNN_AUTO_TEST_CASE_WITH_THF(esizeBilinearNopUint16,
1064 SimpleResizeBilinearTest<DataType::QSymmS16>,
1065 DataLayout::NCHW)
1066ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearSqMin,
1067 ResizeBilinearSqMinTest<DataType::Float32>,
1068 DataLayout::NCHW)
1069ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearSqMinFloat16,
1070 ResizeBilinearSqMinTest<DataType::Float16>,
1071 DataLayout::NCHW)
1072ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearSqMinInt8,
1073 ResizeBilinearSqMinTest<DataType::QAsymmS8>,
1074 DataLayout::NCHW)
1075ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearSqMinUint8,
1076 ResizeBilinearSqMinTest<DataType::QAsymmU8>,
1077 DataLayout::NCHW)
1078ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearSqMinUint16,
1079 SimpleResizeBilinearTest<DataType::QSymmS16>,
1080 DataLayout::NCHW)
1081ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMin,
1082 ResizeBilinearMinTest<DataType::Float32>,
1083 DataLayout::NCHW)
1084ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMinFloat16,
1085 ResizeBilinearMinTest<DataType::Float16>,
1086 DataLayout::NCHW)
1087ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMinInt8,
1088 ResizeBilinearMinTest<DataType::QAsymmS8>,
1089 DataLayout::NCHW)
1090ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMinUint8,
1091 ResizeBilinearMinTest<DataType::QAsymmU8>,
1092 DataLayout::NCHW)
1093ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMinUint16,
1094 SimpleResizeBilinearTest<DataType::QSymmS16>,
1095 DataLayout::NCHW)
1096ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMag,
1097 ResizeBilinearMagTest<DataType::Float32>,
1098 DataLayout::NCHW)
1099ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMagFloat16,
1100 ResizeBilinearMagTest<DataType::Float16>,
1101 DataLayout::NCHW)
1102ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMagInt8,
1103 ResizeBilinearMagTest<DataType::QAsymmS8>,
1104 DataLayout::NCHW)
1105ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMagUint8,
1106 ResizeBilinearMagTest<DataType::QAsymmU8>,
1107 DataLayout::NCHW)
1108ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMagUint16,
1109 SimpleResizeBilinearTest<DataType::QSymmS16>,
1110 DataLayout::NCHW)
1111ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeBilinear,
1112 HalfPixelCentersResizeBilinearTest<DataType::Float32>,
1113 DataLayout::NCHW)
1114ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeBilinearFloat16,
1115 HalfPixelCentersResizeBilinearTest<DataType::Float16>,
1116 DataLayout::NCHW)
1117ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeBilinearInt8,
1118 HalfPixelCentersResizeBilinearTest<DataType::QAsymmS8>,
1119 DataLayout::NCHW)
1120ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeBilinearUint8,
1121 HalfPixelCentersResizeBilinearTest<DataType::QAsymmU8>,
1122 DataLayout::NCHW)
1123ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeBilinearUint16,
1124 HalfPixelCentersResizeBilinearTest<DataType::QSymmS16>,
1125 DataLayout::NCHW)
1126ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeBilinear,
1127 AlignCornersResizeBilinearTest<DataType::Float32>,
1128 DataLayout::NCHW)
1129ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeBilinearFloat16,
1130 AlignCornersResizeBilinearTest<DataType::Float16>,
1131 DataLayout::NCHW)
1132ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeBilinearInt8,
1133 AlignCornersResizeBilinearTest<DataType::QAsymmS8>,
1134 DataLayout::NCHW)
1135ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeBilinearUint8,
1136 AlignCornersResizeBilinearTest<DataType::QAsymmU8>,
1137 DataLayout::NCHW)
1138ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeBilinearUint16,
1139 AlignCornersResizeBilinearTest<DataType::QSymmS16>,
1140 DataLayout::NCHW)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1141
James Conroy59540822018-10-11 12:39:05 +0100[diff] [blame]1142// Resize Bilinear - NHWC
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]1143ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearNopNhwc,
1144 ResizeBilinearNopTest<DataType::Float32>,
1145 DataLayout::NHWC)
1146ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearNopNhwcFloat16,
1147 ResizeBilinearNopTest<DataType::Float16>,
1148 DataLayout::NHWC)
1149ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearNopInt8Nhwc,
1150 ResizeBilinearNopTest<DataType::QAsymmS8>,
1151 DataLayout::NHWC)
1152ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearNopUint8Nhwc,
1153 ResizeBilinearNopTest<DataType::QAsymmU8>,
1154 DataLayout::NHWC)
1155ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearNopUint16Nhwc,
1156 ResizeBilinearNopTest<DataType::QSymmS16>,
1157 DataLayout::NHWC)
1158ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeBilinearNhwc,
1159 SimpleResizeBilinearTest<DataType::Float32>,
1160 DataLayout::NHWC)
1161ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeBilinearNhwcFloat16,
1162 SimpleResizeBilinearTest<DataType::Float16>,
1163 DataLayout::NHWC)
1164ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeBilinearInt8Nhwc,
1165 SimpleResizeBilinearTest<DataType::QAsymmS8>,
1166 DataLayout::NHWC)
1167ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeBilinearUint8Nhwc,
1168 SimpleResizeBilinearTest<DataType::QAsymmU8>,
1169 DataLayout::NHWC)
1170ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeBilinearUint16Nhwc,
1171 ResizeBilinearNopTest<DataType::QSymmS16>,
1172 DataLayout::NHWC)
1173ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearSqMinNhwc,
1174 ResizeBilinearSqMinTest<DataType::Float32>,
1175 DataLayout::NHWC)
1176ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearSqMinNhwcFloat16,
1177 ResizeBilinearSqMinTest<DataType::Float16>,
1178 DataLayout::NHWC)
1179ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearSqMinInt8Nhwc,
1180 ResizeBilinearSqMinTest<DataType::QAsymmS8>,
1181 DataLayout::NHWC)
1182ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearSqMinUint8Nhwc,
1183 ResizeBilinearSqMinTest<DataType::QAsymmU8>,
1184 DataLayout::NHWC)
1185ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearSqMinUint16Nhwc,
1186 ResizeBilinearNopTest<DataType::QSymmS16>,
1187 DataLayout::NHWC)
1188ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMinNhwc,
1189 ResizeBilinearMinTest<DataType::Float32>,
1190 DataLayout::NHWC)
1191ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMinNhwcFloat16,
1192 ResizeBilinearMinTest<DataType::Float16>,
1193 DataLayout::NHWC)
1194ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMinInt8Nhwc,
1195 ResizeBilinearMinTest<DataType::QAsymmS8>,
1196 DataLayout::NHWC)
1197ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMinUint8Nhwc,
1198 ResizeBilinearMinTest<DataType::QAsymmU8>,
1199 DataLayout::NHWC)
1200ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMinUint16Nhwc,
1201 ResizeBilinearNopTest<DataType::QSymmS16>,
1202 DataLayout::NHWC)
1203ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMagNhwc,
1204 ResizeBilinearMagTest<DataType::Float32>,
1205 DataLayout::NHWC)
1206ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMagNhwcFloat16,
1207 ResizeBilinearMagTest<DataType::Float16>,
1208 DataLayout::NHWC)
1209ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMagInt8Nhwc,
1210 ResizeBilinearMagTest<DataType::QAsymmS8>,
1211 DataLayout::NHWC)
1212ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMagUint8Nhwc,
1213 ResizeBilinearMagTest<DataType::QAsymmU8>,
1214 DataLayout::NHWC)
1215ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeBilinearMagUint16Nhwc,
1216 ResizeBilinearNopTest<DataType::QSymmS16>,
1217 DataLayout::NHWC)
1218ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeBilinearNhwc,
1219 HalfPixelCentersResizeBilinearTest<DataType::Float32>,
1220 DataLayout::NCHW)
1221ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeBilinearFloat16Nhwc,
1222 HalfPixelCentersResizeBilinearTest<DataType::Float16>,
1223 DataLayout::NCHW)
1224ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeBilinearInt8Nhwc,
1225 HalfPixelCentersResizeBilinearTest<DataType::QAsymmS8>,
1226 DataLayout::NCHW)
1227ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeBilinearUint8Nhwc,
1228 HalfPixelCentersResizeBilinearTest<DataType::QAsymmU8>,
1229 DataLayout::NCHW)
1230ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeBilinearUint16Nhwc,
1231 HalfPixelCentersResizeBilinearTest<DataType::QSymmS16>,
1232 DataLayout::NCHW)
1233ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeBilinearNhwc,
1234 AlignCornersResizeBilinearTest<DataType::Float32>,
1235 DataLayout::NCHW)
1236ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeBilinearFloat16Nhwc,
1237 AlignCornersResizeBilinearTest<DataType::Float16>,
1238 DataLayout::NCHW)
1239ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeBilinearInt8Nhwc,
1240 AlignCornersResizeBilinearTest<DataType::QAsymmS8>,
1241 DataLayout::NCHW)
1242ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeBilinearUint8Nhwc,
1243 AlignCornersResizeBilinearTest<DataType::QAsymmU8>,
1244 DataLayout::NCHW)
1245ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeBilinearUint16Nhwc,
1246 AlignCornersResizeBilinearTest<DataType::QSymmS16>,
1247 DataLayout::NCHW)
James Conroy59540822018-10-11 12:39:05 +0100[diff] [blame]1248
Teresa Charlin970f43b2019-07-01 13:51:07 +0100[diff] [blame]1249// Resize NearestNeighbor - NCHW
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]1250ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeNearestNeighbor,
1251 SimpleResizeNearestNeighborTest<DataType::Float32>,
1252 DataLayout::NCHW)
1253ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeNearestNeighborInt8,
1254 SimpleResizeNearestNeighborTest<DataType::QAsymmS8>,
1255 DataLayout::NCHW)
1256ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeNearestNeighborUint8,
1257 SimpleResizeNearestNeighborTest<DataType::QAsymmU8>,
1258 DataLayout::NCHW)
1259ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeNearestNeighborUint16,
1260 SimpleResizeNearestNeighborTest<DataType::QSymmS16>,
1261 DataLayout::NCHW)
1262ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborNop,
1263 ResizeNearestNeighborNopTest<DataType::Float32>,
1264 DataLayout::NCHW)
1265ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborNopInt8,
1266 ResizeNearestNeighborNopTest<DataType::QAsymmS8>,
1267 DataLayout::NCHW)
1268ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborNopUint8,
1269 ResizeNearestNeighborNopTest<DataType::QAsymmU8>,
1270 DataLayout::NCHW)
1271ARMNN_AUTO_TEST_CASE_WITH_THF(esizeNearestNeighborNopUint16,
1272 SimpleResizeNearestNeighborTest<DataType::QSymmS16>,
1273 DataLayout::NCHW)
1274ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborSqMin,
1275 ResizeNearestNeighborSqMinTest<DataType::Float32>,
1276 DataLayout::NCHW)
1277ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborSqMinInt8,
1278 ResizeNearestNeighborSqMinTest<DataType::QAsymmS8>,
1279 DataLayout::NCHW)
1280ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborSqMinUint8,
1281 ResizeNearestNeighborSqMinTest<DataType::QAsymmU8>,
1282 DataLayout::NCHW)
1283ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborSqMinUint16,
1284 SimpleResizeNearestNeighborTest<DataType::QSymmS16>,
1285 DataLayout::NCHW)
1286ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMin,
1287 ResizeNearestNeighborMinTest<DataType::Float32>,
1288 DataLayout::NCHW)
1289ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMinInt8,
1290 ResizeNearestNeighborMinTest<DataType::QAsymmS8>,
1291 DataLayout::NCHW)
1292ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMinUint8,
1293 ResizeNearestNeighborMinTest<DataType::QAsymmU8>,
1294 DataLayout::NCHW)
1295ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMinUint16,
1296 SimpleResizeNearestNeighborTest<DataType::QSymmS16>,
1297 DataLayout::NCHW)
1298ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMag,
1299 ResizeNearestNeighborMagTest<DataType::Float32>,
1300 DataLayout::NCHW, 0.10f, 50, 0.11f, 20)
1301ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMagInt8,
1302 ResizeNearestNeighborMagTest<DataType::QAsymmS8>,
1303 DataLayout::NCHW, 0.10f, 50, 0.11f, 20)
1304ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMagUint8,
1305 ResizeNearestNeighborMagTest<DataType::QAsymmU8>,
1306 DataLayout::NCHW, 0.10f, 50, 0.11f, 20)
1307ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMagUint16,
1308 SimpleResizeNearestNeighborTest<DataType::QSymmS16>,
1309 DataLayout::NCHW)
1310ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeNearestNeighbour,
1311 HalfPixelCentersResizeNearestNeighbourTest<DataType::Float32>,
1312 DataLayout::NCHW)
1313ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeNearestNeighbourFloat16,
1314 HalfPixelCentersResizeNearestNeighbourTest<DataType::Float16>,
1315 DataLayout::NCHW)
1316ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeNearestNeighbourInt8,
1317 HalfPixelCentersResizeNearestNeighbourTest<DataType::QAsymmS8>,
1318 DataLayout::NCHW)
1319ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeNearestNeighbourUint8,
1320 HalfPixelCentersResizeNearestNeighbourTest<DataType::QAsymmU8>,
1321 DataLayout::NCHW)
1322ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeNearestNeighbourUint16,
1323 HalfPixelCentersResizeNearestNeighbourTest<DataType::QSymmS16>,
1324 DataLayout::NCHW)
1325ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeNearestNeighbour,
1326 AlignCornersResizeNearestNeighbourTest<DataType::Float32>,
1327 DataLayout::NCHW)
1328ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeNearestNeighbourFloat16,
1329 AlignCornersResizeNearestNeighbourTest<DataType::Float16>,
1330 DataLayout::NCHW)
1331ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeNearestNeighbourInt8,
1332 AlignCornersResizeNearestNeighbourTest<DataType::QAsymmS8>,
1333 DataLayout::NCHW)
1334ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeNearestNeighbourUint8,
1335 AlignCornersResizeNearestNeighbourTest<DataType::QAsymmU8>,
1336 DataLayout::NCHW)
1337ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeNearestNeighbourUint16,
1338 AlignCornersResizeNearestNeighbourTest<DataType::QSymmS16>,
1339 DataLayout::NCHW)
Teresa Charlin970f43b2019-07-01 13:51:07 +0100[diff] [blame]1340
1341// Resize NearestNeighbor - NHWC
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]1342ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborNopNhwc,
1343 ResizeNearestNeighborNopTest<DataType::Float32>,
1344 DataLayout::NHWC)
1345ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborNopInt8Nhwc,
1346 ResizeNearestNeighborNopTest<DataType::QAsymmS8>,
1347 DataLayout::NHWC)
1348ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborNopUint8Nhwc,
1349 ResizeNearestNeighborNopTest<DataType::QAsymmU8>,
1350 DataLayout::NHWC)
1351ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborNopUint16Nhwc,
1352 ResizeNearestNeighborNopTest<DataType::QSymmS16>,
1353 DataLayout::NHWC)
1354ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeNearestNeighborNhwc,
1355 SimpleResizeNearestNeighborTest<DataType::Float32>,
1356 DataLayout::NHWC)
1357ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeNearestNeighborInt8Nhwc,
1358 SimpleResizeNearestNeighborTest<DataType::QAsymmS8>,
1359 DataLayout::NHWC)
1360ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeNearestNeighborUint8Nhwc,
1361 SimpleResizeNearestNeighborTest<DataType::QAsymmU8>,
1362 DataLayout::NHWC)
1363ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleResizeNearestNeighborUint16Nhwc,
1364 ResizeNearestNeighborNopTest<DataType::QSymmS16>,
1365 DataLayout::NHWC)
1366ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborSqMinNhwc,
1367 ResizeNearestNeighborSqMinTest<DataType::Float32>,
1368 DataLayout::NHWC)
1369ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborSqMinInt8Nhwc,
1370 ResizeNearestNeighborSqMinTest<DataType::QAsymmS8>,
1371 DataLayout::NHWC)
1372ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborSqMinUint8Nhwc,
1373 ResizeNearestNeighborSqMinTest<DataType::QAsymmU8>,
1374 DataLayout::NHWC)
1375ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborSqMinUint16Nhwc,
1376 ResizeNearestNeighborNopTest<DataType::QSymmS16>,
1377 DataLayout::NHWC)
1378ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMinNhwc,
1379 ResizeNearestNeighborMinTest<DataType::Float32>,
1380 DataLayout::NHWC)
1381ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMinInt8Nhwc,
1382 ResizeNearestNeighborMinTest<DataType::QAsymmS8>,
1383 DataLayout::NHWC)
1384ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMinUint8Nhwc,
1385 ResizeNearestNeighborMinTest<DataType::QAsymmU8>,
1386 DataLayout::NHWC)
1387ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMinUint16Nhwc,
1388 ResizeNearestNeighborNopTest<DataType::QSymmS16>,
1389 DataLayout::NHWC)
1390ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMagNhwc,
1391 ResizeNearestNeighborMagTest<DataType::Float32>,
1392 DataLayout::NHWC, 0.10f, 50, 0.11f, 20)
1393ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMagInt8Nhwc,
1394 ResizeNearestNeighborMagTest<DataType::QAsymmS8>,
1395 DataLayout::NHWC, 0.10f, 50, 0.11f, 20)
1396ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMagUint8Nhwc,
1397 ResizeNearestNeighborMagTest<DataType::QAsymmU8>,
1398 DataLayout::NHWC, 0.10f, 50, 0.11f, 20)
1399ARMNN_AUTO_TEST_CASE_WITH_THF(ResizeNearestNeighborMagUint16Nhwc,
1400 ResizeNearestNeighborNopTest<DataType::QSymmS16>,
1401 DataLayout::NHWC)
1402ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeNearestNeighbourNchw,
1403 HalfPixelCentersResizeNearestNeighbourTest<DataType::Float32>,
1404 DataLayout::NCHW)
1405ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeNearestNeighbourFloat16Nchw,
1406 HalfPixelCentersResizeNearestNeighbourTest<DataType::Float16>,
1407 DataLayout::NCHW)
1408ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeNearestNeighbourInt8Nchw,
1409 HalfPixelCentersResizeNearestNeighbourTest<DataType::QAsymmS8>,
1410 DataLayout::NCHW)
1411ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeNearestNeighbourUint8Nchw,
1412 HalfPixelCentersResizeNearestNeighbourTest<DataType::QAsymmU8>,
1413 DataLayout::NCHW)
1414ARMNN_AUTO_TEST_CASE_WITH_THF(HalfPixelCentersResizeNearestNeighbourUint16Nchw,
1415 HalfPixelCentersResizeNearestNeighbourTest<DataType::QSymmS16>,
1416 DataLayout::NCHW)
1417ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeNearestNeighbourNchw,
1418 AlignCornersResizeNearestNeighbourTest<DataType::Float32>,
1419 DataLayout::NCHW)
1420ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeNearestNeighbourFloat16Nchw,
1421 AlignCornersResizeNearestNeighbourTest<DataType::Float16>,
1422 DataLayout::NCHW)
1423ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeNearestNeighbourInt8NchRsqrtw,
1424 AlignCornersResizeNearestNeighbourTest<DataType::QAsymmS8>,
1425 DataLayout::NCHW)
1426ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeNearestNeighbourUint8Nchw,
1427 AlignCornersResizeNearestNeighbourTest<DataType::QAsymmU8>,
1428 DataLayout::NCHW)
1429ARMNN_AUTO_TEST_CASE_WITH_THF(AlignCornersResizeNearestNeighbourUint16Nchw,
1430 AlignCornersResizeNearestNeighbourTest<DataType::QSymmS16>,
1431 DataLayout::NCHW)
Teresa Charlin970f43b2019-07-01 13:51:07 +0100[diff] [blame]1432
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1433// Fake Quantization
Finn Williams87020072020-08-26 16:19:15 +0100[diff] [blame]1434ARMNN_AUTO_TEST_CASE_WITH_THF(FakeQuantization, FakeQuantizationTest)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1435
Matteo Martincigh539b44d2018-10-01 09:26:39 +0100[diff] [blame]1436// L2 Normalization
Finn Williamsc43de6a2020-08-27 11:13:25 +0100[diff] [blame]1437ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization1d, L2Normalization1dTest, DataLayout::NCHW)
1438ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization2d, L2Normalization2dTest, DataLayout::NCHW)
1439ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization3d, L2Normalization3dTest, DataLayout::NCHW)
1440ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization4d, L2Normalization4dTest, DataLayout::NCHW)
Ferran Balaguerc6138d82019-06-13 17:23:50 +0100[diff] [blame]1441
Finn Williamsc43de6a2020-08-27 11:13:25 +0100[diff] [blame]1442ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization1dInt16, L2Normalization1dInt16Test, DataLayout::NCHW)
1443ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization2dInt16, L2Normalization2dInt16Test, DataLayout::NCHW)
1444ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization3dInt16, L2Normalization3dInt16Test, DataLayout::NCHW)
1445ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization4dInt16, L2Normalization4dInt16Test, DataLayout::NCHW)
jimfly013aab7c32018-11-12 13:32:08 +0000[diff] [blame]1446
Finn Williamsc43de6a2020-08-27 11:13:25 +0100[diff] [blame]1447ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization1dUint8, L2Normalization1dUint8Test, DataLayout::NCHW)
1448ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization2dUint8, L2Normalization2dUint8Test, DataLayout::NCHW)
1449ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization3dUint8, L2Normalization3dUint8Test, DataLayout::NCHW)
1450ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization4dUint8, L2Normalization4dUint8Test, DataLayout::NCHW)
Ferran Balaguerc6138d82019-06-13 17:23:50 +0100[diff] [blame]1451
Finn Williamsc43de6a2020-08-27 11:13:25 +0100[diff] [blame]1452ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization1dNhwc, L2Normalization1dTest, DataLayout::NHWC)
1453ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization2dNhwc, L2Normalization2dTest, DataLayout::NHWC)
1454ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization3dNhwc, L2Normalization3dTest, DataLayout::NHWC)
1455ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization4dNhwc, L2Normalization4dTest, DataLayout::NHWC)
Ferran Balaguerc6138d82019-06-13 17:23:50 +0100[diff] [blame]1456
Finn Williamsc43de6a2020-08-27 11:13:25 +0100[diff] [blame]1457ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization1dInt16Nhwc, L2Normalization1dInt16Test, DataLayout::NHWC)
1458ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization2dInt16Nhwc, L2Normalization2dInt16Test, DataLayout::NHWC)
1459ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization3dInt16Nhwc, L2Normalization3dInt16Test, DataLayout::NHWC)
1460ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization4dInt16Nhwc, L2Normalization4dInt16Test, DataLayout::NHWC)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1461
Finn Williamsc43de6a2020-08-27 11:13:25 +0100[diff] [blame]1462ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization1dUint8Nhwc, L2Normalization1dUint8Test, DataLayout::NHWC)
1463ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization2dUint8Nhwc, L2Normalization2dUint8Test, DataLayout::NHWC)
1464ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization3dUint8Nhwc, L2Normalization3dUint8Test, DataLayout::NHWC)
1465ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization4dUint8Nhwc, L2Normalization4dUint8Test, DataLayout::NHWC)
Ferran Balaguerc6138d82019-06-13 17:23:50 +0100[diff] [blame]1466
Finn Williamsc43de6a2020-08-27 11:13:25 +0100[diff] [blame]1467ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization2dShape, L2Normalization2dShapeTest);
Matthew Jackson82b15ed2019-07-25 16:14:30 +0100[diff] [blame]1468
Finn Williamsc43de6a2020-08-27 11:13:25 +0100[diff] [blame]1469ARMNN_AUTO_TEST_CASE_WITH_THF(L2NormalizationDefaultEpsilon, L2NormalizationDefaultEpsilonTest, DataLayout::NCHW)
1470ARMNN_AUTO_TEST_CASE_WITH_THF(L2NormalizationNonDefaultEpsilon, L2NormalizationNonDefaultEpsilonTest, DataLayout::NCHW)
Ferran Balaguere52211e2019-06-17 12:23:52 +0100[diff] [blame]1471
Aron Virginas-Tare662a942019-10-14 15:12:00 +0100[diff] [blame]1472// LogSoftmax
Finn Williamsc43de6a2020-08-27 11:13:25 +0100[diff] [blame]1473ARMNN_AUTO_TEST_CASE_WITH_THF(LogSoftmaxFloat32_1, LogSoftmaxTest1<DataType::Float32>)
1474ARMNN_AUTO_TEST_CASE_WITH_THF(LogSoftmaxFloat32_2, LogSoftmaxTest2<DataType::Float32>)
1475ARMNN_AUTO_TEST_CASE_WITH_THF(LogSoftmaxFloat32_3, LogSoftmaxTest3<DataType::Float32>)
1476ARMNN_AUTO_TEST_CASE_WITH_THF(LogSoftmaxFloat32_4, LogSoftmaxTest4<DataType::Float32>)
Aron Virginas-Tare662a942019-10-14 15:12:00 +0100[diff] [blame]1477
Finn Williamsc43de6a2020-08-27 11:13:25 +0100[diff] [blame]1478ARMNN_AUTO_TEST_CASE_WITH_THF(LogSoftmaxFloat16_1, LogSoftmaxTest1<DataType::Float16>)
1479ARMNN_AUTO_TEST_CASE_WITH_THF(LogSoftmaxFloat16_2, LogSoftmaxTest2<DataType::Float16>)
1480ARMNN_AUTO_TEST_CASE_WITH_THF(LogSoftmaxFloat16_3, LogSoftmaxTest3<DataType::Float16>)
1481ARMNN_AUTO_TEST_CASE_WITH_THF(LogSoftmaxFloat16_4, LogSoftmaxTest4<DataType::Float16>)
Aron Virginas-Tare662a942019-10-14 15:12:00 +0100[diff] [blame]1482
Matthew Sloyan2e5d0b22021-10-21 14:05:31 +0100[diff] [blame]1483// Pad - Constant
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]1484ARMNN_AUTO_TEST_CASE_WITH_THF(PadBFloat162d, PadBFloat162dTest)
1485ARMNN_AUTO_TEST_CASE_WITH_THF(PadBFloat162dCustomPadding, PadBFloat162dCustomPaddingTest)
1486ARMNN_AUTO_TEST_CASE_WITH_THF(PadBFloat163d, PadBFloat163dTest)
1487ARMNN_AUTO_TEST_CASE_WITH_THF(PadBFloat164d, PadBFloat164dTest)
Narumol Prangnawarat44179c32020-03-11 14:51:27 +0000[diff] [blame]1488
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]1489ARMNN_AUTO_TEST_CASE_WITH_THF(PadFloat322d, PadFloat322dTest)
1490ARMNN_AUTO_TEST_CASE_WITH_THF(PadFloat322dCustomPadding, PadFloat322dCustomPaddingTest)
1491ARMNN_AUTO_TEST_CASE_WITH_THF(PadFloat323d, PadFloat323dTest)
1492ARMNN_AUTO_TEST_CASE_WITH_THF(PadFloat324d, PadFloat324dTest)
Mohamed Nour Abouelseouddd6acea2018-10-18 12:26:19 +0100[diff] [blame]1493
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]1494ARMNN_AUTO_TEST_CASE_WITH_THF(PadUint82d, PadUint82dTest)
1495ARMNN_AUTO_TEST_CASE_WITH_THF(PadUint82dCustomPadding, PadUint82dCustomPaddingTest)
1496ARMNN_AUTO_TEST_CASE_WITH_THF(PadUint83d, PadUint83dTest)
1497ARMNN_AUTO_TEST_CASE_WITH_THF(PadUint84d, PadUint84dTest)
Narumol Prangnawarat8719d222020-11-27 16:57:56 +0000[diff] [blame]1498ARMNN_AUTO_TEST_CASE_WITH_THF(PadUint8Asymm, PadQAsymmTestCommon<DataType::QAsymmU8>, -2.0f, 3, 0.0f)
1499ARMNN_AUTO_TEST_CASE_WITH_THF(PadUint8CustomPaddingAsymm, PadQAsymmTestCommon<DataType::QAsymmU8>, -2.0f, 3, 2.0f)
Mohamed Nour Abouelseoud7420e552018-10-12 12:26:24 +0100[diff] [blame]1500
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]1501ARMNN_AUTO_TEST_CASE_WITH_THF(Pad2dQSymm16, Pad2dTestCommon<DataType::QSymmS16>, 2.0f, 0, 0.0f)
1502ARMNN_AUTO_TEST_CASE_WITH_THF(Pad2dQSymm16CustomPadding, Pad2dTestCommon<DataType::QSymmS16>, 2.0f, 0, 1.0f)
1503ARMNN_AUTO_TEST_CASE_WITH_THF(Pad3dQSymm16, Pad3dTestCommon<DataType::QSymmS16>, 2.0f, 0)
1504ARMNN_AUTO_TEST_CASE_WITH_THF(Pad4dQSymm16, Pad4dTestCommon<DataType::QSymmS16>, 2.0f, 0)
Narumol Prangnawarate6eaf662019-07-08 08:57:17 +0100[diff] [blame]1505
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]1506ARMNN_AUTO_TEST_CASE_WITH_THF(PadInt82d, PadInt82dTest)
1507ARMNN_AUTO_TEST_CASE_WITH_THF(PadInt82dCustomPadding, PadInt82dCustomPaddingTest)
1508ARMNN_AUTO_TEST_CASE_WITH_THF(PadInt83d, PadInt83dTest)
1509ARMNN_AUTO_TEST_CASE_WITH_THF(PadInt84d, PadInt84dTest)
Narumol Prangnawarat8719d222020-11-27 16:57:56 +0000[diff] [blame]1510ARMNN_AUTO_TEST_CASE_WITH_THF(PadQAsymmS8, PadQAsymmTestCommon<DataType::QAsymmS8>, -2.0f, 3, 0.0f)
1511ARMNN_AUTO_TEST_CASE_WITH_THF(PadQAsymmS8CustomPadding, PadQAsymmTestCommon<DataType::QAsymmS8>, -2.0f, 3, 2.0f)
Sadik Armagan041b3c02020-06-04 10:32:18 +0100[diff] [blame]1512
Matthew Sloyan2e5d0b22021-10-21 14:05:31 +0100[diff] [blame]1513// Pad - Symmetric & Reflect
1514ARMNN_AUTO_TEST_CASE_WITH_THF(PadSymmetric2dFloat32, PadSymmetric2dFloat32Test)
1515ARMNN_AUTO_TEST_CASE_WITH_THF(PadReflect2dFloat32, PadReflect2dFloat32Test)
1516
1517ARMNN_AUTO_TEST_CASE_WITH_THF(PadSymmetric3dFloat32, PadSymmetric3dFloat32Test)
1518ARMNN_AUTO_TEST_CASE_WITH_THF(PadReflect3dFloat32, PadReflect3dFloat32Test)
1519ARMNN_AUTO_TEST_CASE_WITH_THF(PadSymmetric3dUint8, PadSymmetric3dUint8Test)
1520ARMNN_AUTO_TEST_CASE_WITH_THF(PadReflect3dUint8, PadReflect3dUint8Test)
1521ARMNN_AUTO_TEST_CASE_WITH_THF(PadSymmetric3dInt8, PadSymmetric3dInt8Test)
1522ARMNN_AUTO_TEST_CASE_WITH_THF(PadReflect3dInt8, PadReflect3dInt8Test)
1523
1524ARMNN_AUTO_TEST_CASE_WITH_THF(PadSymmetric4dFloat32, PadSymmetric4dFloat32Test)
1525ARMNN_AUTO_TEST_CASE_WITH_THF(PadReflect4dFloat32, PadReflect4dFloat32Test)
1526ARMNN_AUTO_TEST_CASE_WITH_THF(PadSymmetric4dBFloat16, PadSymmetric4dBFloat16Test)
1527ARMNN_AUTO_TEST_CASE_WITH_THF(PadReflect4dBFloat16, PadReflect4dBFloat16Test)
1528ARMNN_AUTO_TEST_CASE_WITH_THF(PadSymmetric4dUint8, PadSymmetric4dUint8Test)
1529ARMNN_AUTO_TEST_CASE_WITH_THF(PadReflect4dUint8, PadReflect4dUint8Test)
1530ARMNN_AUTO_TEST_CASE_WITH_THF(PadSymmetric4dInt8, PadSymmetric4dInt8Test)
1531ARMNN_AUTO_TEST_CASE_WITH_THF(PadReflect4dInt8, PadReflect4dInt8Test)
1532ARMNN_AUTO_TEST_CASE_WITH_THF(PadSymmetric4dInt16, PadSymmetric4dInt16Test)
1533ARMNN_AUTO_TEST_CASE_WITH_THF(PadReflect4dInt16, PadReflect4dInt16Test)
1534
1535ARMNN_AUTO_TEST_CASE_WITH_THF(PadSymmetricFloat16, PadSymmetricFloat16Test)
1536ARMNN_AUTO_TEST_CASE_WITH_THF(PadReflectFloat16, PadReflectFloat16Test)
1537
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1538// Constant
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]1539ARMNN_AUTO_TEST_CASE_WITH_THF(Constant, ConstantTest)
1540ARMNN_AUTO_TEST_CASE_WITH_THF(ConstantUint8, ConstantUint8CustomQuantizationScaleAndOffsetTest)
1541ARMNN_AUTO_TEST_CASE_WITH_THF(ConstantInt16, ConstantInt16CustomQuantizationScaleAndOffsetTest)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1542
1543// Concat
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]1544ARMNN_AUTO_TEST_CASE_WITH_THF(Concat1d, Concat1dTest)
1545ARMNN_AUTO_TEST_CASE_WITH_THF(Concat1dUint8, Concat1dUint8Test)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1546
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]1547ARMNN_AUTO_TEST_CASE_WITH_THF(Concat2dDim0, Concat2dDim0Test)
1548ARMNN_AUTO_TEST_CASE_WITH_THF(Concat2dDim0Uint8, Concat2dDim0Uint8Test)
1549ARMNN_AUTO_TEST_CASE_WITH_THF(Concat2dDim1, Concat2dDim1Test)
1550ARMNN_AUTO_TEST_CASE_WITH_THF(Concat2dDim1Uint8, Concat2dDim1Uint8Test)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1551
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]1552ARMNN_AUTO_TEST_CASE_WITH_THF(Concat2dDim0DiffInputDims, Concat2dDim0DiffInputDimsTest)
1553ARMNN_AUTO_TEST_CASE_WITH_THF(Concat2dDim0DiffInputDimsUint8, Concat2dDim0DiffInputDimsUint8Test)
1554ARMNN_AUTO_TEST_CASE_WITH_THF(Concat2dDim1DiffInputDims, Concat2dDim1DiffInputDimsTest)
1555ARMNN_AUTO_TEST_CASE_WITH_THF(Concat2dDim1DiffInputDimsUint8, Concat2dDim1DiffInputDimsUint8Test)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1556
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]1557ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim0, Concat3dDim0Test)
1558ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim0Uint8, Concat3dDim0Uint8Test)
1559ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim1, Concat3dDim1Test)
1560ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim1Uint8, Concat3dDim1Uint8Test)
1561ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim2, Concat3dDim2Test, true)
1562ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim2Uint8, Concat3dDim2Uint8Test, true)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1563
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]1564ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim0DiffInputDims, Concat3dDim0DiffInputDimsTest)
1565ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim0DiffInputDimsUint8, Concat3dDim0DiffInputDimsUint8Test)
1566ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim1DiffInputDims, Concat3dDim1DiffInputDimsTest)
1567ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim1DiffInputDimsUint8, Concat3dDim1DiffInputDimsUint8Test)
1568ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim2DiffInputDims, Concat3dDim2DiffInputDimsTest, true)
1569ARMNN_AUTO_TEST_CASE_WITH_THF(Concat3dDim2DiffInputDimsUint8, Concat3dDim2DiffInputDimsUint8Test, true)
narpra015cdda352018-11-19 15:30:27 +0000[diff] [blame]1570
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]1571ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDim0, Concat4dDim0Test)
1572ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDim1, Concat4dDim1Test)
1573ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDim2, Concat4dDim2Test)
1574ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDim3, Concat4dDim3Test, true)
1575ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDim0Uint8, Concat4dDim0Uint8Test)
1576ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDim1Uint8, Concat4dDim1Uint8Test)
1577ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDim2Uint8, Concat4dDim2Uint8Test)
1578ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDim3Uint8, Concat4dDim3Uint8Test, true)
narpra015cdda352018-11-19 15:30:27 +0000[diff] [blame]1579
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]1580ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDiffShapeDim0, Concat4dDiffShapeDim0Test)
1581ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDiffShapeDim1, Concat4dDiffShapeDim1Test)
1582ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDiffShapeDim2, Concat4dDiffShapeDim2Test)
1583ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDiffShapeDim3, Concat4dDiffShapeDim3Test, true)
1584ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDiffShapeDim0Uint8, Concat4dDiffShapeDim0Uint8Test)
1585ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDiffShapeDim1Uint8, Concat4dDiffShapeDim1Uint8Test)
1586ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDiffShapeDim2Uint8, Concat4dDiffShapeDim2Uint8Test)
1587ARMNN_AUTO_TEST_CASE_WITH_THF(Concat4dDiffShapeDim3Uint8, Concat4dDiffShapeDim3Uint8Test, true)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1588
Ryan OSheaf4bfa6a2020-06-10 11:33:37 +0100[diff] [blame]1589// Fill
Finn Williams87020072020-08-26 16:19:15 +0100[diff] [blame]1590ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleFill, SimpleFillTest<DataType::Float32>)
1591ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleFillF16, SimpleFillTest<DataType::Float16>)
1592ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleFillS32, SimpleFillTest<DataType::Signed32>)
Ryan OSheaf4bfa6a2020-06-10 11:33:37 +0100[diff] [blame]1593
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1594// Floor
Finn Williams7faf9a82020-08-27 10:37:36 +0100[diff] [blame]1595ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleFloor, SimpleFloorTest<DataType::Float32>)
1596ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleFloorFloat16, SimpleFloorTest<DataType::Float16>)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1597
1598// Reshape
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]1599ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleReshapeFloat32, SimpleReshapeTest<DataType::Float32>)
1600ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleReshapeQuantisedAsymmS8, SimpleReshapeTest<DataType::QAsymmS8>)
1601ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleReshapeQuantisedAsymm8, SimpleReshapeTest<DataType::QAsymmU8>)
1602ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleReshapeQuantisedSymm16, SimpleReshapeTest<DataType::QSymmS16>)
1603ARMNN_AUTO_TEST_CASE_WITH_THF(Reshape5d, Reshape5dTest<DataType::Float32>)
Narumol Prangnawarat0c95f4c2020-11-18 16:52:07 +0000[diff] [blame]1604ARMNN_AUTO_TEST_CASE_WITH_THF(ReshapeBoolean, ReshapeBooleanTest)
1605
Simon Obute51f67772021-09-03 15:50:13 +0100[diff] [blame]1606// ChannelShuffle
1607ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleChannelShuffleFloat32, SimpleChannelShuffleTest<DataType::Float32>)
1608ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleChannelShuffleQAsymmU8, SimpleChannelShuffleTest<DataType::QAsymmU8>)
Teresa Charlineb852bb2021-09-30 12:44:33 +0100[diff] [blame]1609ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleChannelShuffleQAsymmS8, SimpleChannelShuffleTest<DataType::QAsymmS8>)
Simon Obute51f67772021-09-03 15:50:13 +0100[diff] [blame]1610ARMNN_AUTO_TEST_CASE_WITH_THF(ChannelShuffle2DFloat32, ChannelShuffle2DTest<DataType::Float32>)
1611ARMNN_AUTO_TEST_CASE_WITH_THF(ChannelShuffle2DQAsymmU8, ChannelShuffle2DTest<DataType::QAsymmU8>)
Teresa Charlineb852bb2021-09-30 12:44:33 +0100[diff] [blame]1612ARMNN_AUTO_TEST_CASE_WITH_THF(ChannelShuffle2DQAsymmS8, ChannelShuffle2DTest<DataType::QAsymmS8>)
Simon Obute51f67772021-09-03 15:50:13 +0100[diff] [blame]1613ARMNN_AUTO_TEST_CASE_WITH_THF(ChannelShuffle4DFloat32, ChannelShuffle4DTest<DataType::Float32>)
1614ARMNN_AUTO_TEST_CASE_WITH_THF(ChannelShuffle4DQAsymmU8, ChannelShuffle4DTest<DataType::QAsymmU8>)
Teresa Charlineb852bb2021-09-30 12:44:33 +0100[diff] [blame]1615ARMNN_AUTO_TEST_CASE_WITH_THF(ChannelShuffle4DQAsymmS8, ChannelShuffle4DTest<DataType::QAsymmS8>)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1616
Mohamed Nour Abouelseouda1d3c6a2018-12-27 12:39:16 +0000[diff] [blame]1617// Rsqrt
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]1618ARMNN_AUTO_TEST_CASE_WITH_THF(Rsqrt2d, Rsqrt2dTest<DataType::Float32>)
1619ARMNN_AUTO_TEST_CASE_WITH_THF(Rsqrt3d, Rsqrt3dTest<DataType::Float32>)
1620ARMNN_AUTO_TEST_CASE_WITH_THF(RsqrtZero, RsqrtZeroTest<DataType::Float32>)
1621ARMNN_AUTO_TEST_CASE_WITH_THF(RsqrtNegative, RsqrtNegativeTest<DataType::Float32>)
1622ARMNN_AUTO_TEST_CASE_WITH_THF(Rsqrt2dFloat16, Rsqrt2dTest<DataType::Float16>)
1623ARMNN_AUTO_TEST_CASE_WITH_THF(Rsqrt3dFloat16, Rsqrt3dTest<DataType::Float16>)
1624ARMNN_AUTO_TEST_CASE_WITH_THF(Rsqrt2dQuantisedAsymmS8, Rsqrt2dTest<DataType::QAsymmS8>)
1625ARMNN_AUTO_TEST_CASE_WITH_THF(Rsqrt3dQuantisedAsymmS8, Rsqrt3dTest<DataType::QAsymmS8>)
1626ARMNN_AUTO_TEST_CASE_WITH_THF(Rsqrt2dQuantisedAsymm8, Rsqrt2dTest<DataType::QAsymmU8>)
1627ARMNN_AUTO_TEST_CASE_WITH_THF(Rsqrt3dQuantisedAsymm8, Rsqrt3dTest<DataType::QAsymmU8>)
1628ARMNN_AUTO_TEST_CASE_WITH_THF(Rsqrt2dQuantisedSymm16, Rsqrt2dTest<DataType::QSymmS16>)
1629ARMNN_AUTO_TEST_CASE_WITH_THF(Rsqrt3dQuantisedSymm16, Rsqrt3dTest<DataType::QSymmS16>)
Mohamed Nour Abouelseouda1d3c6a2018-12-27 12:39:16 +0000[diff] [blame]1630
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1631// Permute
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]1632ARMNN_AUTO_TEST_CASE_WITH_THF(SimplePermuteBFloat16, SimplePermuteTest<DataType::BFloat16>)
1633ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteBFloat16ValueSet1Test, PermuteValueSet1Test<DataType::BFloat16>)
1634ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteBFloat16ValueSet2Test, PermuteValueSet2Test<DataType::BFloat16>)
1635ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteBFloat16ValueSet3Test, PermuteValueSet3Test<DataType::BFloat16>)
1636ARMNN_AUTO_TEST_CASE_WITH_THF(SimplePermuteFloat32, SimplePermuteTest<DataType::Float32>)
1637ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteFloat32ValueSet1Test, PermuteValueSet1Test<DataType::Float32>)
1638ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteFloat32ValueSet2Test, PermuteValueSet2Test<DataType::Float32>)
1639ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteFloat32ValueSet3Test, PermuteValueSet3Test<DataType::Float32>)
1640ARMNN_AUTO_TEST_CASE_WITH_THF(SimplePermuteQASymS8, SimplePermuteTest<DataType::QAsymmS8>)
1641ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteQASymmS8ValueSet1Test, PermuteValueSet1Test<DataType::QAsymmS8>)
1642ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteQASymmS8ValueSet2Test, PermuteValueSet2Test<DataType::QAsymmS8>)
1643ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteQASymmS8ValueSet3Test, PermuteValueSet3Test<DataType::QAsymmS8>)
1644ARMNN_AUTO_TEST_CASE_WITH_THF(SimplePermuteQASymm8, SimplePermuteTest<DataType::QAsymmU8>)
1645ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteQASymm8ValueSet1Test, PermuteValueSet1Test<DataType::QAsymmU8>)
1646ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteQASymm8ValueSet2Test, PermuteValueSet2Test<DataType::QAsymmU8>)
1647ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteQASymm8ValueSet3Test, PermuteValueSet3Test<DataType::QAsymmU8>)
1648ARMNN_AUTO_TEST_CASE_WITH_THF(SimplePermuteQSymm16, SimplePermuteTest<DataType::QSymmS16>)
1649ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteQSymm16ValueSet1Test, PermuteValueSet1Test<DataType::QSymmS16>)
1650ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteQSymm16ValueSet2Test, PermuteValueSet2Test<DataType::QSymmS16>)
1651ARMNN_AUTO_TEST_CASE_WITH_THF(PermuteQSymm16ValueSet3Test, PermuteValueSet3Test<DataType::QSymmS16>)
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1652
Matteo Martincigha65b7ae2018-11-14 12:39:55 +0000[diff] [blame]1653// Lstm
Sadik Armagan1625efc2021-06-10 18:24:34 +0100[diff] [blame]1654TEST_CASE("LstmUtilsZeroVector") {
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]1655 LstmUtilsZeroVectorTest(); }
Sadik Armagan1625efc2021-06-10 18:24:34 +0100[diff] [blame]1656TEST_CASE("LstmUtilsMeanStddevNormalization") {
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]1657 LstmUtilsMeanStddevNormalizationNoneZeroInputTest();
1658 LstmUtilsMeanStddevNormalizationAllZeroInputTest();
1659 LstmUtilsMeanStddevNormalizationMixedZeroInputTest(); }
Sadik Armagan1625efc2021-06-10 18:24:34 +0100[diff] [blame]1660TEST_CASE("LstmUtilsVectorBatchVectorCwiseProduct") {
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]1661 LstmUtilsVectorBatchVectorCwiseProductTest(); }
Sadik Armagan1625efc2021-06-10 18:24:34 +0100[diff] [blame]1662TEST_CASE("LstmUtilsVectorBatchVectorAdd") {
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]1663 LstmUtilsVectorBatchVectorAddTest(); }
Jan Eilers38e05bd2019-06-26 13:10:09 +0100[diff] [blame]1664
Finn Williamsc43de6a2020-08-27 11:13:25 +0100[diff] [blame]1665ARMNN_AUTO_TEST_CASE_WITH_THF(LstmLayerFloat32WithCifgWithPeepholeNoProjection,
1666 LstmLayerFloat32WithCifgWithPeepholeNoProjectionTest)
1667ARMNN_AUTO_TEST_CASE_WITH_THF(LstmLayerFloat32NoCifgNoPeepholeNoProjection,
1668 LstmLayerFloat32NoCifgNoPeepholeNoProjectionTest)
1669ARMNN_AUTO_TEST_CASE_WITH_THF(LstmLayerFloat32NoCifgWithPeepholeWithProjection,
1670 LstmLayerFloat32NoCifgWithPeepholeWithProjectionTest)
Matteo Martincigha65b7ae2018-11-14 12:39:55 +0000[diff] [blame]1671
Finn Williamsc43de6a2020-08-27 11:13:25 +0100[diff] [blame]1672ARMNN_AUTO_TEST_CASE_WITH_THF(LstmLayerFloat32NoCifgWithPeepholeWithProjectionWithLayerNorm,
1673 LstmLayerFloat32NoCifgWithPeepholeWithProjectionWithLayerNormTest)
Jan Eilers38e05bd2019-06-26 13:10:09 +0100[diff] [blame]1674
Finn Williamsc43de6a2020-08-27 11:13:25 +0100[diff] [blame]1675ARMNN_AUTO_TEST_CASE_WITH_THF(LstmLayerInt16NoCifgNoPeepholeNoProjection,
1676 LstmLayerInt16NoCifgNoPeepholeNoProjectionTest)
1677ARMNN_AUTO_TEST_CASE_WITH_THF(LstmLayerInt16WithCifgWithPeepholeNoProjection,
1678 LstmLayerInt16WithCifgWithPeepholeNoProjectionTest)
1679ARMNN_AUTO_TEST_CASE_WITH_THF(LstmLayerInt16NoCifgWithPeepholeWithProjection,
1680 LstmLayerInt16NoCifgWithPeepholeWithProjectionTest)
1681ARMNN_AUTO_TEST_CASE_WITH_THF(LstmLayerInt16NoCifgNoPeepholeNoProjectionInt16Constant,
1682 LstmLayerInt16NoCifgNoPeepholeNoProjectionInt16ConstantTest)
Conor Kennedyb9971c92019-05-07 07:14:23 +0100[diff] [blame]1683
James Conroy4f1f8992020-04-29 20:01:10 +0100[diff] [blame]1684// QLstm
Finn Williamsc43de6a2020-08-27 11:13:25 +0100[diff] [blame]1685ARMNN_AUTO_TEST_CASE_WITH_THF(QLstm, QLstmTest)
1686ARMNN_AUTO_TEST_CASE_WITH_THF(QLstm1, QLstmTest1)
1687ARMNN_AUTO_TEST_CASE_WITH_THF(QLstm2, QLstmTest2)
James Conroy4f1f8992020-04-29 20:01:10 +0100[diff] [blame]1688
mathad01b392e982021-04-07 12:07:30 +0100[diff] [blame]1689// Cast
1690ARMNN_AUTO_TEST_CASE_WITH_THF(CastInt32ToFloat, CastInt32ToFloat2dTest)
1691ARMNN_AUTO_TEST_CASE_WITH_THF(CastInt16ToFloat, CastInt16ToFloat2dTest)
1692ARMNN_AUTO_TEST_CASE_WITH_THF(CastInt8ToFloat, CastInt8ToFloat2dTest)
1693ARMNN_AUTO_TEST_CASE_WITH_THF(CastInt8AsymmToFloat, CastInt8AsymmToFloat2dTest)
1694ARMNN_AUTO_TEST_CASE_WITH_THF(CastUIntToFloat, CastUInt8ToFloat2dTest)
1695ARMNN_AUTO_TEST_CASE_WITH_THF(CastInt8ToUInt, CastInt8ToUInt82dTest)
1696ARMNN_AUTO_TEST_CASE_WITH_THF(CastInt8AsymmToUInt, CastInt8AsymmToUInt82dTest)
1697ARMNN_AUTO_TEST_CASE_WITH_THF(CastFloat16ToFloat32, CastFloat16ToFloat322dTest)
1698ARMNN_AUTO_TEST_CASE_WITH_THF(CastBFloat16ToFloat32, CastBFloat16ToFloat322dTest)
1699ARMNN_AUTO_TEST_CASE_WITH_THF(CastFloatToFloat16, CastFloat32ToFloat162dTest)
1700ARMNN_AUTO_TEST_CASE_WITH_THF(CastFloatToIn8, CastFloat32ToInt82dTest)
1701ARMNN_AUTO_TEST_CASE_WITH_THF(CastFloatToUInt8, CastFloat32ToUInt82dTest)
1702
Narumol Prangnawarat7ddbbae2020-03-13 10:26:05 +0000[diff] [blame]1703// Convert from BFloat16 to Float32
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]1704ARMNN_AUTO_TEST_CASE_WITH_THF(ConvertBf16ToFp32, ConvertBf16ToFp32Test)
Narumol Prangnawarat7ddbbae2020-03-13 10:26:05 +0000[diff] [blame]1705
Narumol Prangnawaratea54a012020-03-16 16:36:10 +0000[diff] [blame]1706// Convert from Float32 to BFloat16
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]1707ARMNN_AUTO_TEST_CASE_WITH_THF(ConvertFp32ToBf16, ConvertFp32ToBf16Test)
Narumol Prangnawaratea54a012020-03-16 16:36:10 +0000[diff] [blame]1708
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]1709// Convert from Float16 to Float32
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]1710ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvertFp16ToFp32, SimpleConvertFp16ToFp32Test)
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]1711// Convert from Float32 to Float16
Keith Davisf500d6c2020-08-31 08:32:55 +0100[diff] [blame]1712ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleConvertFp32ToFp16, SimpleConvertFp32ToFp16Test)
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]1713
narpra011e4c31d2018-09-28 11:07:51 +0100[diff] [blame]1714// Mean
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]1715ARMNN_AUTO_TEST_CASE_WITH_THF(MeanSimpleFloat32, MeanSimpleTest<DataType::Float32>)
1716ARMNN_AUTO_TEST_CASE_WITH_THF(MeanSimpleAxisFloat32, MeanSimpleAxisTest<DataType::Float32>)
1717ARMNN_AUTO_TEST_CASE_WITH_THF(MeanKeepDimsFloat32, MeanKeepDimsTest<DataType::Float32>)
1718ARMNN_AUTO_TEST_CASE_WITH_THF(MeanMultipleDimsFloat32, MeanMultipleDimsTest<DataType::Float32>)
1719ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts1Float32, MeanVts1Test<DataType::Float32>)
1720ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts2Float32, MeanVts2Test<DataType::Float32>)
1721ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts3Float32, MeanVts3Test<DataType::Float32>)
narpra011e4c31d2018-09-28 11:07:51 +0100[diff] [blame]1722
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]1723ARMNN_AUTO_TEST_CASE_WITH_THF(MeanSimpleQuantisedAsymmS8, MeanSimpleTest<DataType::QAsymmS8>)
1724ARMNN_AUTO_TEST_CASE_WITH_THF(MeanSimpleAxisQuantisedAsymmS8, MeanSimpleAxisTest<DataType::QAsymmS8>)
1725ARMNN_AUTO_TEST_CASE_WITH_THF(MeanKeepDimsQuantisedAsymmS8, MeanKeepDimsTest<DataType::QAsymmS8>)
1726ARMNN_AUTO_TEST_CASE_WITH_THF(MeanMultipleDimsQuantisedAsymmS8, MeanMultipleDimsTest<DataType::QAsymmS8>)
1727ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts1QuantisedAsymmS8, MeanVts1Test<DataType::QAsymmS8>)
1728ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts2QuantisedAsymmS8, MeanVts2Test<DataType::QAsymmS8>)
1729ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts3QuantisedAsymmS8, MeanVts3Test<DataType::QAsymmS8>)
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]1730
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]1731ARMNN_AUTO_TEST_CASE_WITH_THF(MeanSimpleQuantisedAsymm8, MeanSimpleTest<DataType::QAsymmU8>)
1732ARMNN_AUTO_TEST_CASE_WITH_THF(MeanSimpleAxisQuantisedAsymm8, MeanSimpleAxisTest<DataType::QAsymmU8>)
1733ARMNN_AUTO_TEST_CASE_WITH_THF(MeanKeepDimsQuantisedAsymm8, MeanKeepDimsTest<DataType::QAsymmU8>)
1734ARMNN_AUTO_TEST_CASE_WITH_THF(MeanMultipleDimsQuantisedAsymm8, MeanMultipleDimsTest<DataType::QAsymmU8>)
1735ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts1QuantisedAsymm8, MeanVts1Test<DataType::QAsymmU8>)
1736ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts2QuantisedAsymm8, MeanVts2Test<DataType::QAsymmU8>)
1737ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts3QuantisedAsymm8, MeanVts3Test<DataType::QAsymmU8>)
narpra011e4c31d2018-09-28 11:07:51 +0100[diff] [blame]1738
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]1739ARMNN_AUTO_TEST_CASE_WITH_THF(MeanSimpleQuantisedSymm16, MeanSimpleTest<DataType::QSymmS16>)
1740ARMNN_AUTO_TEST_CASE_WITH_THF(MeanSimpleAxisQuantisedSymm16, MeanSimpleAxisTest<DataType::QSymmS16>)
1741ARMNN_AUTO_TEST_CASE_WITH_THF(MeanKeepDimsQuantisedSymm16, MeanKeepDimsTest<DataType::QSymmS16>)
1742ARMNN_AUTO_TEST_CASE_WITH_THF(MeanMultipleDimsQuantisedSymm16, MeanMultipleDimsTest<DataType::QSymmS16>)
1743ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts1QuantisedSymm16, MeanVts1Test<DataType::QSymmS16>)
1744ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts2QuantisedSymm16, MeanVts2Test<DataType::QSymmS16>)
1745ARMNN_AUTO_TEST_CASE_WITH_THF(MeanVts3QuantisedSymm16, MeanVts3Test<DataType::QSymmS16>)
James Conroyb80775f2019-06-11 11:25:30 +0100[diff] [blame]1746
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]1747ARMNN_AUTO_TEST_CASE_WITH_THF(AdditionAfterMaxPool, AdditionAfterMaxPoolTest)
Éanna Ó Catháin47c1ddb2018-10-12 14:24:13 +0100[diff] [blame]1748
Nikhil Raj68c2c902019-09-19 11:21:11 +0100[diff] [blame]1749// ArgMinMax
Sadik Armagan43528ac2020-08-27 09:06:40 +0100[diff] [blame]1750ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxFloat32, ArgMaxSimpleTest<DataType::Float32>)
1751ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinFloat32, ArgMinSimpleTest<DataType::Float32>)
1752ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinChannelFloat32, ArgMinChannelTest<DataType::Float32>)
1753ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxChannelFloat32, ArgMaxChannelTest<DataType::Float32>)
1754ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxHeightFloat32, ArgMaxHeightTest<DataType::Float32>)
1755ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinWidthFloat32, ArgMinWidthTest<DataType::Float32>)
Nikhil Raj68c2c902019-09-19 11:21:11 +0100[diff] [blame]1756
Sadik Armagan43528ac2020-08-27 09:06:40 +0100[diff] [blame]1757ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxFloat16, ArgMaxSimpleTest<DataType::Float16>)
1758ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinFloat16, ArgMinSimpleTest<DataType::Float16>)
1759ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinChannelFloat16, ArgMinChannelTest<DataType::Float16>)
1760ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxChannelFloat16, ArgMaxChannelTest<DataType::Float16>)
1761ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxHeightFloat16, ArgMaxHeightTest<DataType::Float16>)
1762ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinWidthFloat16, ArgMinWidthTest<DataType::Float16>)
Teresa Charline300b362020-05-25 10:01:03 +0100[diff] [blame]1763
Sadik Armagan43528ac2020-08-27 09:06:40 +0100[diff] [blame]1764ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxSigned32, ArgMaxSimpleTest<DataType::Signed32>)
1765ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinSigned32, ArgMinSimpleTest<DataType::Signed32>)
1766ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinChannelSigned32, ArgMinChannelTest<DataType::Signed32>)
1767ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxChannelSigned32, ArgMaxChannelTest<DataType::Signed32>)
1768ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxHeightSigned32, ArgMaxHeightTest<DataType::Signed32>)
1769ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinWidthSigned32, ArgMinWidthTest<DataType::Signed32>)
Francis Murtagh1939df52019-11-13 15:21:09 +0000[diff] [blame]1770
Sadik Armagan43528ac2020-08-27 09:06:40 +0100[diff] [blame]1771ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxSimpleQuantisedAsymmS8, ArgMaxSimpleTest<DataType::QAsymmS8>)
1772ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinSimpleQuantisedAsymmS8, ArgMinSimpleTest<DataType::QAsymmS8>)
1773ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinChannelQuantisedAsymmS8, ArgMinChannelTest<DataType::QAsymmS8>)
1774ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxChannelQuantisedAsymmS8, ArgMaxChannelTest<DataType::QAsymmS8>)
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]1775
Sadik Armagan43528ac2020-08-27 09:06:40 +0100[diff] [blame]1776ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxSimpleQuantisedAsymm8, ArgMaxSimpleTest<DataType::QAsymmU8>)
1777ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinSimpleQuantisedAsymm8, ArgMinSimpleTest<DataType::QAsymmU8>)
1778ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinChannelQuantisedAsymm8, ArgMinChannelTest<DataType::QAsymmU8>)
1779ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxChannelQuantisedAsymm8, ArgMaxChannelTest<DataType::QAsymmU8>)
Nikhil Raj68c2c902019-09-19 11:21:11 +0100[diff] [blame]1780
Sadik Armagan43528ac2020-08-27 09:06:40 +0100[diff] [blame]1781ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxSimpleQuantisedSymm16, ArgMaxSimpleTest<DataType::QSymmS16>)
1782ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinSimpleQuantisedSymm16, ArgMinSimpleTest<DataType::QSymmS16>)
1783ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMinChannelQuantisedSymm16, ArgMinChannelTest<DataType::QSymmS16>)
1784ARMNN_AUTO_TEST_CASE_WITH_THF(ArgMaxChannelQuantisedSymm16, ArgMaxChannelTest<DataType::QSymmS16>)
Nikhil Raj68c2c902019-09-19 11:21:11 +0100[diff] [blame]1785
Nattapat Chaimanowong3ea76d52018-11-09 14:10:38 +0000[diff] [blame]1786// Space To Batch Nd
Sadik Armagan714fe5b2020-08-27 12:46:36 +0100[diff] [blame]1787ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdSimpleFloat32, SpaceToBatchNdSimpleFloat32Test)
1788ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiChannelsFloat32, SpaceToBatchNdMultiChannelsFloat32Test)
1789ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiBlockFloat32, SpaceToBatchNdMultiBlockFloat32Test)
1790ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdPaddingFloat32, SpaceToBatchNdPaddingFloat32Test)
Nattapat Chaimanowong3ea76d52018-11-09 14:10:38 +0000[diff] [blame]1791
Sadik Armagan714fe5b2020-08-27 12:46:36 +0100[diff] [blame]1792ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdSimpleFloat16, SpaceToBatchNdSimpleFloat16Test)
1793ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiChannelsFloat16, SpaceToBatchNdMultiChannelsFloat16Test)
1794ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiBlockFloat16, SpaceToBatchNdMultiBlockFloat16Test)
1795ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdPaddingFloat16, SpaceToBatchNdPaddingFloat16Test)
Matthew Jackson9bff1442019-09-12 09:08:23 +0100[diff] [blame]1796
Sadik Armagan714fe5b2020-08-27 12:46:36 +0100[diff] [blame]1797ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdSimpleUint8, SpaceToBatchNdSimpleUint8Test)
1798ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiChannelsUint8, SpaceToBatchNdMultiChannelsUint8Test)
1799ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiBlockUint8, SpaceToBatchNdMultiBlockUint8Test)
1800ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdPaddingUint8, SpaceToBatchNdPaddingUint8Test)
Nattapat Chaimanowong3ea76d52018-11-09 14:10:38 +0000[diff] [blame]1801
Sadik Armagan714fe5b2020-08-27 12:46:36 +0100[diff] [blame]1802ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdSimpleNhwcFloat32, SpaceToBatchNdSimpleNhwcFloat32Test)
1803ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiChannelsNhwcFloat32, SpaceToBatchNdMultiChannelsNhwcFloat32Test)
1804ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiBlockNhwcFloat32, SpaceToBatchNdMultiBlockNhwcFloat32Test)
1805ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdPaddingNhwcFloat32, SpaceToBatchNdPaddingNhwcFloat32Test)
Nattapat Chaimanowong3ea76d52018-11-09 14:10:38 +0000[diff] [blame]1806
Sadik Armagan714fe5b2020-08-27 12:46:36 +0100[diff] [blame]1807ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdSimpleNhwcFloat16, SpaceToBatchNdSimpleNhwcFloat16Test)
1808ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiChannelsNhwcFloat16, SpaceToBatchNdMultiChannelsNhwcFloat16Test)
1809ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiBlockNhwcFloat16, SpaceToBatchNdMultiBlockNhwcFloat16Test)
1810ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdPaddingNhwcFloat16, SpaceToBatchNdPaddingNhwcFloat16Test)
Matthew Jackson9bff1442019-09-12 09:08:23 +0100[diff] [blame]1811
Sadik Armagan714fe5b2020-08-27 12:46:36 +0100[diff] [blame]1812ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdSimpleNhwcUint8, SpaceToBatchNdSimpleNhwcUint8Test)
1813ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiChannelsNhwcUint8, SpaceToBatchNdMultiChannelsNhwcUint8Test)
1814ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiBlockNhwcUint8, SpaceToBatchNdMultiBlockNhwcUint8Test)
1815ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdPaddingNhwcUint8, SpaceToBatchNdPaddingNhwcUint8Test)
Nattapat Chaimanowong3ea76d52018-11-09 14:10:38 +0000[diff] [blame]1816
Sadik Armagan714fe5b2020-08-27 12:46:36 +0100[diff] [blame]1817ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdSimpleUint16, SpaceToBatchNdSimpleUint16Test)
1818ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiChannelsUint16, SpaceToBatchNdMultiChannelsUint16Test)
1819ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiBlockUint16, SpaceToBatchNdMultiBlockUint16Test)
1820ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdPaddingUint16, SpaceToBatchNdPaddingUint16Test)
nikraj01120522a2019-05-31 11:33:07 +0100[diff] [blame]1821
Sadik Armagan714fe5b2020-08-27 12:46:36 +0100[diff] [blame]1822ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdSimpleNhwcUint16, SpaceToBatchNdSimpleNhwcUint16Test)
1823ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiChannelsNhwcUint16, SpaceToBatchNdMultiChannelsNhwcUint16Test)
1824ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdMultiBlockNhwcUint16, SpaceToBatchNdMultiBlockNhwcUint16Test)
1825ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToBatchNdPaddingNhwcUint16, SpaceToBatchNdPaddingNhwcUint16Test)
nikraj01120522a2019-05-31 11:33:07 +0100[diff] [blame]1826
Francis Murtagh57f13d52019-06-24 14:24:36 +0100[diff] [blame]1827// BatchToSpace
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]1828ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcFloat32_1, BatchToSpaceNdNhwcTest1<DataType::Float32>)
1829ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcFloat32_2, BatchToSpaceNdNhwcTest2<DataType::Float32>)
1830ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcFloat32_3, BatchToSpaceNdNhwcTest3<DataType::Float32>)
1831ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcFloat32_4, BatchToSpaceNdNhwcTest4<DataType::Float32>)
1832ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcFloat32_5, BatchToSpaceNdNhwcTest5<DataType::Float32>)
1833ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcFloat32_6, BatchToSpaceNdNhwcTest6<DataType::Float32>)
1834ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcFloat32_7, BatchToSpaceNdNhwcTest7<DataType::Float32>)
Matthew Jackson9bff1442019-09-12 09:08:23 +0100[diff] [blame]1835
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]1836ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcFloat16_1, BatchToSpaceNdNhwcTest1<DataType::Float16>)
1837ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcFloat16_2, BatchToSpaceNdNhwcTest2<DataType::Float16>)
1838ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcFloat16_3, BatchToSpaceNdNhwcTest3<DataType::Float16>)
1839ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcFloat16_4, BatchToSpaceNdNhwcTest4<DataType::Float16>)
1840ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcFloat16_5, BatchToSpaceNdNhwcTest5<DataType::Float16>)
1841ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcFloat16_6, BatchToSpaceNdNhwcTest6<DataType::Float16>)
1842ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcFloat16_7, BatchToSpaceNdNhwcTest7<DataType::Float16>)
Éanna Ó Catháin4e1e1362018-11-12 11:36:34 +0000[diff] [blame]1843
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]1844ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcInt1, BatchToSpaceNdNhwcTest1<DataType::QAsymmS8>)
1845ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcInt2, BatchToSpaceNdNhwcTest2<DataType::QAsymmS8>)
1846ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcInt3, BatchToSpaceNdNhwcTest3<DataType::QAsymmS8>)
1847ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcInt4, BatchToSpaceNdNhwcTest4<DataType::QAsymmS8>)
1848ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcInt5, BatchToSpaceNdNhwcTest5<DataType::QAsymmS8>)
1849ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcInt6, BatchToSpaceNdNhwcTest6<DataType::QAsymmS8>)
1850ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcInt7, BatchToSpaceNdNhwcTest7<DataType::QAsymmS8>)
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]1851
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]1852ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcUint1, BatchToSpaceNdNhwcTest1<DataType::QAsymmU8>)
1853ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcUint2, BatchToSpaceNdNhwcTest2<DataType::QAsymmU8>)
1854ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcUint3, BatchToSpaceNdNhwcTest3<DataType::QAsymmU8>)
1855ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcUint4, BatchToSpaceNdNhwcTest4<DataType::QAsymmU8>)
1856ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcUint5, BatchToSpaceNdNhwcTest5<DataType::QAsymmU8>)
1857ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcUint6, BatchToSpaceNdNhwcTest6<DataType::QAsymmU8>)
1858ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcUint7, BatchToSpaceNdNhwcTest7<DataType::QAsymmU8>)
Éanna Ó Catháin4e1e1362018-11-12 11:36:34 +0000[diff] [blame]1859
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]1860ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcQsymm16_1, BatchToSpaceNdNhwcTest1<DataType::QSymmS16>)
1861ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcQsymm16_2, BatchToSpaceNdNhwcTest2<DataType::QSymmS16>)
1862ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcQsymm16_3, BatchToSpaceNdNhwcTest3<DataType::QSymmS16>)
1863ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcQsymm16_4, BatchToSpaceNdNhwcTest4<DataType::QSymmS16>)
1864ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcQsymm16_5, BatchToSpaceNdNhwcTest5<DataType::QSymmS16>)
1865ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcQsymm16_6, BatchToSpaceNdNhwcTest6<DataType::QSymmS16>)
1866ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNhwcQsymm16_7, BatchToSpaceNdNhwcTest7<DataType::QSymmS16>)
Francis Murtaghd0dfe172019-06-25 10:57:10 +0100[diff] [blame]1867
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]1868ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwFloat16_1, BatchToSpaceNdNchwTest1<DataType::Float16>)
1869ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwFloat16_2, BatchToSpaceNdNchwTest2<DataType::Float16>)
1870ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwFloat16_3, BatchToSpaceNdNchwTest3<DataType::Float16>)
1871ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwFloat16_4, BatchToSpaceNdNchwTest4<DataType::Float16>)
1872ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwFloat16_5, BatchToSpaceNdNchwTest5<DataType::Float16>)
1873ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwFloat16_6, BatchToSpaceNdNchwTest6<DataType::Float16>)
1874ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwFloat16_7, BatchToSpaceNdNchwTest7<DataType::Float16>)
Mike Kelly831faed2018-11-28 11:52:08 +0000[diff] [blame]1875
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]1876ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwInt1, BatchToSpaceNdNchwTest1<DataType::QAsymmS8>)
1877ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwInt2, BatchToSpaceNdNchwTest2<DataType::QAsymmS8>)
1878ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwInt3, BatchToSpaceNdNchwTest3<DataType::QAsymmS8>)
1879ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwInt4, BatchToSpaceNdNchwTest4<DataType::QAsymmS8>)
1880ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwInt5, BatchToSpaceNdNchwTest5<DataType::QAsymmS8>)
1881ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwInt6, BatchToSpaceNdNchwTest6<DataType::QAsymmS8>)
1882ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwInt7, BatchToSpaceNdNchwTest7<DataType::QAsymmS8>)
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]1883
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]1884ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwUint1, BatchToSpaceNdNchwTest1<DataType::QAsymmU8>)
1885ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwUint2, BatchToSpaceNdNchwTest2<DataType::QAsymmU8>)
1886ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwUint3, BatchToSpaceNdNchwTest3<DataType::QAsymmU8>)
1887ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwUint4, BatchToSpaceNdNchwTest4<DataType::QAsymmU8>)
1888ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwUint5, BatchToSpaceNdNchwTest5<DataType::QAsymmU8>)
1889ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwUint6, BatchToSpaceNdNchwTest6<DataType::QAsymmU8>)
1890ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwUint7, BatchToSpaceNdNchwTest7<DataType::QAsymmU8>)
Éanna Ó Catháin262553e2018-11-14 11:26:23 +0000[diff] [blame]1891
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]1892ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwQsymm16_1, BatchToSpaceNdNchwTest1<DataType::QSymmS16>)
1893ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwQsymm16_2, BatchToSpaceNdNchwTest2<DataType::QSymmS16>)
1894ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwQsymm16_3, BatchToSpaceNdNchwTest3<DataType::QSymmS16>)
1895ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwQsymm16_4, BatchToSpaceNdNchwTest4<DataType::QSymmS16>)
1896ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwQsymm16_5, BatchToSpaceNdNchwTest5<DataType::QSymmS16>)
1897ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwQsymm16_6, BatchToSpaceNdNchwTest6<DataType::QSymmS16>)
1898ARMNN_AUTO_TEST_CASE_WITH_THF(BatchToSpaceNdNchwQsymm16_7, BatchToSpaceNdNchwTest7<DataType::QSymmS16>)
Francis Murtaghd0dfe172019-06-25 10:57:10 +0100[diff] [blame]1899
Aron Virginas-Tar73f66422019-09-23 19:11:59 +0100[diff] [blame]1900// DepthToSpace
1901ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwFloat32_1, DepthToSpaceTest1<DataType::Float32>, DataLayout::NCHW);
1902ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwFloat32_2, DepthToSpaceTest2<DataType::Float32>, DataLayout::NCHW);
1903ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwFloat32_3, DepthToSpaceTest3<DataType::Float32>, DataLayout::NCHW);
1904ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwFloat32_4, DepthToSpaceTest4<DataType::Float32>, DataLayout::NCHW);
1905
1906ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwFloat16_1, DepthToSpaceTest1<DataType::Float16>, DataLayout::NCHW);
1907ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwFloat16_2, DepthToSpaceTest2<DataType::Float16>, DataLayout::NCHW);
1908ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwFloat16_3, DepthToSpaceTest3<DataType::Float16>, DataLayout::NCHW);
1909ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwFloat16_4, DepthToSpaceTest4<DataType::Float16>, DataLayout::NCHW);
1910
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]1911ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwInt8_1, DepthToSpaceTest1<DataType::QAsymmS8>, DataLayout::NCHW);
1912ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwInt8_2, DepthToSpaceTest2<DataType::QAsymmS8>, DataLayout::NCHW);
1913ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwInt8_3, DepthToSpaceTest3<DataType::QAsymmS8>, DataLayout::NCHW);
1914ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwInt8_4, DepthToSpaceTest4<DataType::QAsymmS8>, DataLayout::NCHW);
1915
Derek Lambertif90c56d2020-01-10 17:14:08 +0000[diff] [blame]1916ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwUint8_1, DepthToSpaceTest1<DataType::QAsymmU8>, DataLayout::NCHW);
1917ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwUint8_2, DepthToSpaceTest2<DataType::QAsymmU8>, DataLayout::NCHW);
1918ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwUint8_3, DepthToSpaceTest3<DataType::QAsymmU8>, DataLayout::NCHW);
1919ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwUint8_4, DepthToSpaceTest4<DataType::QAsymmU8>, DataLayout::NCHW);
Aron Virginas-Tar73f66422019-09-23 19:11:59 +0100[diff] [blame]1920
Derek Lambertif90c56d2020-01-10 17:14:08 +0000[diff] [blame]1921ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwInt16_1, DepthToSpaceTest1<DataType::QSymmS16>, DataLayout::NCHW);
1922ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwInt16_2, DepthToSpaceTest2<DataType::QSymmS16>, DataLayout::NCHW);
1923ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwInt16_3, DepthToSpaceTest3<DataType::QSymmS16>, DataLayout::NCHW);
1924ARMNN_AUTO_TEST_CASE(DepthToSpaceNchwInt16_4, DepthToSpaceTest4<DataType::QSymmS16>, DataLayout::NCHW);
Aron Virginas-Tar73f66422019-09-23 19:11:59 +0100[diff] [blame]1925
1926ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcFloat32_1, DepthToSpaceTest1<DataType::Float32>, DataLayout::NHWC);
1927ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcFloat32_2, DepthToSpaceTest2<DataType::Float32>, DataLayout::NHWC);
1928ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcFloat32_3, DepthToSpaceTest3<DataType::Float32>, DataLayout::NHWC);
1929ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcFloat32_4, DepthToSpaceTest4<DataType::Float32>, DataLayout::NHWC);
1930
1931ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcFloat16_1, DepthToSpaceTest1<DataType::Float16>, DataLayout::NHWC);
1932ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcFloat16_2, DepthToSpaceTest2<DataType::Float16>, DataLayout::NHWC);
1933ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcFloat16_3, DepthToSpaceTest3<DataType::Float16>, DataLayout::NHWC);
1934ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcFloat16_4, DepthToSpaceTest4<DataType::Float16>, DataLayout::NHWC);
1935
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]1936ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcInt8_1, DepthToSpaceTest1<DataType::QAsymmS8>, DataLayout::NHWC);
1937ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcInt8_2, DepthToSpaceTest2<DataType::QAsymmS8>, DataLayout::NHWC);
1938ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcInt8_3, DepthToSpaceTest3<DataType::QAsymmS8>, DataLayout::NHWC);
1939ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcInt8_4, DepthToSpaceTest4<DataType::QAsymmS8>, DataLayout::NHWC);
1940
Derek Lambertif90c56d2020-01-10 17:14:08 +0000[diff] [blame]1941ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcUint8_1, DepthToSpaceTest1<DataType::QAsymmU8>, DataLayout::NHWC);
1942ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcUint8_2, DepthToSpaceTest2<DataType::QAsymmU8>, DataLayout::NHWC);
1943ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcUint8_3, DepthToSpaceTest3<DataType::QAsymmU8>, DataLayout::NHWC);
1944ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcUint8_4, DepthToSpaceTest4<DataType::QAsymmU8>, DataLayout::NHWC);
Aron Virginas-Tar73f66422019-09-23 19:11:59 +0100[diff] [blame]1945
Derek Lambertif90c56d2020-01-10 17:14:08 +0000[diff] [blame]1946ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcInt16_1, DepthToSpaceTest1<DataType::QSymmS16>, DataLayout::NHWC);
1947ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcInt16_2, DepthToSpaceTest2<DataType::QSymmS16>, DataLayout::NHWC);
1948ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcInt16_3, DepthToSpaceTest3<DataType::QSymmS16>, DataLayout::NHWC);
1949ARMNN_AUTO_TEST_CASE(DepthToSpaceNhwcInt16_4, DepthToSpaceTest4<DataType::QSymmS16>, DataLayout::NHWC);
Francis Murtaghd0dfe172019-06-25 10:57:10 +0100[diff] [blame]1950
Keith Davis3ae3f972021-05-21 16:33:48 +0100[diff] [blame]1951// Shape
1952ARMNN_AUTO_TEST_CASE_WITH_THF(ShapeDimSize1Float16, ShapeDimSize1Test<DataType::Float16>)
1953ARMNN_AUTO_TEST_CASE_WITH_THF(ShapeDimSize1Float32, ShapeDimSize1Test<DataType::Float32>)
1954ARMNN_AUTO_TEST_CASE_WITH_THF(ShapeDimSize1QAsymmU8, ShapeDimSize1Test<DataType::QAsymmU8>)
1955ARMNN_AUTO_TEST_CASE_WITH_THF(ShapeDimSize1Signed32, ShapeDimSize1Test<DataType::Signed32>)
1956ARMNN_AUTO_TEST_CASE_WITH_THF(ShapeDimSize1QSymmS16, ShapeDimSize1Test<DataType::QSymmS16>)
1957ARMNN_AUTO_TEST_CASE_WITH_THF(ShapeDimSize1QSymmS8, ShapeDimSize1Test<DataType::QSymmS8>)
1958ARMNN_AUTO_TEST_CASE_WITH_THF(ShapeDimSize1QAsymmS8, ShapeDimSize1Test<DataType::QAsymmS8>)
1959ARMNN_AUTO_TEST_CASE_WITH_THF(ShapeDimSize1BFloat16, ShapeDimSize1Test<DataType::BFloat16>)
1960
1961ARMNN_AUTO_TEST_CASE_WITH_THF(ShapeDimSize2Float16, ShapeDimSize2Test<DataType::Float16>)
1962ARMNN_AUTO_TEST_CASE_WITH_THF(ShapeDimSize2Float32, ShapeDimSize2Test<DataType::Float32>)
1963ARMNN_AUTO_TEST_CASE_WITH_THF(ShapeDimSize2QAsymmU8, ShapeDimSize2Test<DataType::QAsymmU8>)
1964ARMNN_AUTO_TEST_CASE_WITH_THF(ShapeDimSize2Signed32, ShapeDimSize2Test<DataType::Signed32>)
1965ARMNN_AUTO_TEST_CASE_WITH_THF(ShapeDimSize2QSymmS16, ShapeDimSize2Test<DataType::QSymmS16>)
1966ARMNN_AUTO_TEST_CASE_WITH_THF(ShapeDimSize2QSymmS8, ShapeDimSize2Test<DataType::QSymmS8>)
1967ARMNN_AUTO_TEST_CASE_WITH_THF(ShapeDimSize2QAsymmS8, ShapeDimSize2Test<DataType::QAsymmS8>)
1968ARMNN_AUTO_TEST_CASE_WITH_THF(ShapeDimSize2BFloat16, ShapeDimSize2Test<DataType::BFloat16>)
1969
1970ARMNN_AUTO_TEST_CASE_WITH_THF(ShapeDimSize3Float16, ShapeDimSize3Test<DataType::Float16>)
1971ARMNN_AUTO_TEST_CASE_WITH_THF(ShapeDimSize3Float32, ShapeDimSize3Test<DataType::Float32>)
1972ARMNN_AUTO_TEST_CASE_WITH_THF(ShapeDimSize3QAsymmU8, ShapeDimSize3Test<DataType::QAsymmU8>)
1973ARMNN_AUTO_TEST_CASE_WITH_THF(ShapeDimSize3Signed32, ShapeDimSize3Test<DataType::Signed32>)
1974ARMNN_AUTO_TEST_CASE_WITH_THF(ShapeDimSize3QSymmS16, ShapeDimSize3Test<DataType::QSymmS16>)
1975ARMNN_AUTO_TEST_CASE_WITH_THF(ShapeDimSize3QSymmS8, ShapeDimSize3Test<DataType::QSymmS8>)
1976ARMNN_AUTO_TEST_CASE_WITH_THF(ShapeDimSize3QAsymmS8, ShapeDimSize3Test<DataType::QAsymmS8>)
1977ARMNN_AUTO_TEST_CASE_WITH_THF(ShapeDimSize3BFloat16, ShapeDimSize3Test<DataType::BFloat16>)
1978
1979ARMNN_AUTO_TEST_CASE_WITH_THF(ShapeDimSize4Float16, ShapeDimSize4Test<DataType::Float16>)
1980ARMNN_AUTO_TEST_CASE_WITH_THF(ShapeDimSize4Float32, ShapeDimSize4Test<DataType::Float32>)
1981ARMNN_AUTO_TEST_CASE_WITH_THF(ShapeDimSize4QAsymmU8, ShapeDimSize4Test<DataType::QAsymmU8>)
1982ARMNN_AUTO_TEST_CASE_WITH_THF(ShapeDimSize4Signed32, ShapeDimSize4Test<DataType::Signed32>)
1983ARMNN_AUTO_TEST_CASE_WITH_THF(ShapeDimSize4QSymmS16, ShapeDimSize4Test<DataType::QSymmS16>)
1984ARMNN_AUTO_TEST_CASE_WITH_THF(ShapeDimSize4QSymmS8, ShapeDimSize4Test<DataType::QSymmS8>)
1985ARMNN_AUTO_TEST_CASE_WITH_THF(ShapeDimSize4QAsymmS8, ShapeDimSize4Test<DataType::QAsymmS8>)
1986ARMNN_AUTO_TEST_CASE_WITH_THF(ShapeDimSize4BFloat16, ShapeDimSize4Test<DataType::BFloat16>)
1987
Francis Murtagh57f13d52019-06-24 14:24:36 +0100[diff] [blame]1988// SpaceToDepth
Sadik Armagan5e719842020-08-28 12:17:33 +0100[diff] [blame]1989ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToDepthNchwAsymmQ8, SpaceToDepthNchwAsymmQ8Test)
1990ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToDepthNhwcAsymmQ8, SpaceToDepthNhwcAsymmQ8Test)
Keith Davisa57eccb2019-06-14 17:33:22 +0100[diff] [blame]1991
Sadik Armagan5e719842020-08-28 12:17:33 +0100[diff] [blame]1992ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToDepthNhwc1Float32, SpaceToDepthNhwcFloat32Test1)
1993ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToDepthNchw1Float32, SpaceToDepthNchwFloat32Test1)
James Conroyd2aa85e2019-07-01 17:12:40 +0100[diff] [blame]1994
Sadik Armagan5e719842020-08-28 12:17:33 +0100[diff] [blame]1995ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToDepthNhwc2Float32, SpaceToDepthNhwcFloat32Test2)
1996ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToDepthNchw2Float32, SpaceToDepthNchwFloat32Test2)
James Conroyd2aa85e2019-07-01 17:12:40 +0100[diff] [blame]1997
Sadik Armagan5e719842020-08-28 12:17:33 +0100[diff] [blame]1998ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToDepthNhwcQSymm16, SpaceToDepthNhwcQSymm16Test)
1999ARMNN_AUTO_TEST_CASE_WITH_THF(SpaceToDepthNchwQSymm16, SpaceToDepthNchwQSymm16Test)
Keith Davisa57eccb2019-06-14 17:33:22 +0100[diff] [blame]2000
Nattapat Chaimanowong1216b582018-11-23 15:33:41 +0000[diff] [blame]2001// Strided Slice
Sadik Armagan5e719842020-08-28 12:17:33 +0100[diff] [blame]2002ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice4dFloat32, StridedSlice4dFloat32Test)
2003ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice4dReverseFloat32, StridedSlice4dReverseFloat32Test)
2004ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSliceSimpleStrideFloat32, StridedSliceSimpleStrideFloat32Test)
2005ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSliceSimpleRangeMaskFloat32, StridedSliceSimpleRangeMaskFloat32Test)
2006ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSliceShrinkAxisMaskFloat32, StridedSliceShrinkAxisMaskFloat32Test)
2007ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSliceShrinkAxisMaskCTSFloat32, StridedSliceShrinkAxisMaskCTSFloat32Test)
2008ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSliceShrinkAxisMaskBitPosition0Dim3Float32,
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]2009 StridedSliceShrinkAxisMaskBitPosition0Dim3Float32Test)
Sadik Armagan5e719842020-08-28 12:17:33 +0100[diff] [blame]2010ARMNN_AUTO_TEST_CASE_WITH_THF(
2011 StridedSliceShrinkAxisMaskBitPosition0Float32, StridedSliceShrinkAxisMaskBitPosition0Float32Test)
2012ARMNN_AUTO_TEST_CASE_WITH_THF(
2013 StridedSliceShrinkAxisMaskBitPosition1Float32, StridedSliceShrinkAxisMaskBitPosition1Float32Test)
2014ARMNN_AUTO_TEST_CASE_WITH_THF(
2015 StridedSliceShrinkAxisMaskBitPosition2Float32, StridedSliceShrinkAxisMaskBitPosition2Float32Test)
2016ARMNN_AUTO_TEST_CASE_WITH_THF(
2017 StridedSliceShrinkAxisMaskBitPosition3Float32, StridedSliceShrinkAxisMaskBitPosition3Float32Test)
2018ARMNN_AUTO_TEST_CASE_WITH_THF(
2019 StridedSliceShrinkAxisMaskBitPosition0And1Float32, StridedSliceShrinkAxisMaskBitPosition0And1Float32Test)
2020ARMNN_AUTO_TEST_CASE_WITH_THF(
2021 StridedSliceShrinkAxisMaskBitPosition0And2Float32, StridedSliceShrinkAxisMaskBitPosition0And2Float32Test)
2022ARMNN_AUTO_TEST_CASE_WITH_THF(
2023 StridedSliceShrinkAxisMaskBitPosition0And3Float32, StridedSliceShrinkAxisMaskBitPosition0And3Float32Test)
2024ARMNN_AUTO_TEST_CASE_WITH_THF(
2025 StridedSliceShrinkAxisMaskBitPosition0And1And3Float32, StridedSliceShrinkAxisMaskBitPosition0And1And3Float32Test)
2026ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice3dFloat32, StridedSlice3dFloat32Test)
2027ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice3dReverseFloat32, StridedSlice3dReverseFloat32Test)
2028ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice2dFloat32, StridedSlice2dFloat32Test)
2029ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice2dReverseFloat32, StridedSlice2dReverseFloat32Test)
Nattapat Chaimanowong1216b582018-11-23 15:33:41 +0000[diff] [blame]2030
Sadik Armagan5e719842020-08-28 12:17:33 +0100[diff] [blame]2031ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice4dUint8, StridedSlice4dUint8Test)
2032ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice4dReverseUint8, StridedSlice4dReverseUint8Test)
2033ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSliceSimpleStrideUint8, StridedSliceSimpleStrideUint8Test)
2034ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSliceSimpleRangeMaskUint8, StridedSliceSimpleRangeMaskUint8Test)
2035ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSliceShrinkAxisMaskUint8, StridedSliceShrinkAxisMaskUint8Test)
2036ARMNN_AUTO_TEST_CASE_WITH_THF(
2037 StridedSliceShrinkAxisMaskBitPosition0Dim3Uint8, StridedSliceShrinkAxisMaskBitPosition0Dim3Uint8Test)
2038ARMNN_AUTO_TEST_CASE_WITH_THF(
2039 StridedSliceShrinkAxisMaskBitPosition0Uint8, StridedSliceShrinkAxisMaskBitPosition0Uint8Test)
2040ARMNN_AUTO_TEST_CASE_WITH_THF(
2041 StridedSliceShrinkAxisMaskBitPosition1Uint8, StridedSliceShrinkAxisMaskBitPosition1Uint8Test)
2042ARMNN_AUTO_TEST_CASE_WITH_THF(
2043 StridedSliceShrinkAxisMaskBitPosition2Uint8, StridedSliceShrinkAxisMaskBitPosition2Uint8Test)
2044ARMNN_AUTO_TEST_CASE_WITH_THF(
2045 StridedSliceShrinkAxisMaskBitPosition3Uint8, StridedSliceShrinkAxisMaskBitPosition3Uint8Test)
2046ARMNN_AUTO_TEST_CASE_WITH_THF(
2047 StridedSliceShrinkAxisMaskBitPosition0And1Uint8, StridedSliceShrinkAxisMaskBitPosition0And1Uint8Test)
2048ARMNN_AUTO_TEST_CASE_WITH_THF(
2049 StridedSliceShrinkAxisMaskBitPosition0And2Uint8, StridedSliceShrinkAxisMaskBitPosition0And2Uint8Test)
2050ARMNN_AUTO_TEST_CASE_WITH_THF(
2051 StridedSliceShrinkAxisMaskBitPosition0And3Uint8, StridedSliceShrinkAxisMaskBitPosition0And3Uint8Test)
2052ARMNN_AUTO_TEST_CASE_WITH_THF(
2053 StridedSliceShrinkAxisMaskBitPosition0And1And3Uint8, StridedSliceShrinkAxisMaskBitPosition0And1And3Uint8Test)
2054ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice3dUint8, StridedSlice3dUint8Test)
2055ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice3dReverseUint8, StridedSlice3dReverseUint8Test)
2056ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice2dUint8, StridedSlice2dUint8Test)
2057ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice2dReverseUint8, StridedSlice2dReverseUint8Test)
Nattapat Chaimanowong1216b582018-11-23 15:33:41 +0000[diff] [blame]2058
Sadik Armagan5e719842020-08-28 12:17:33 +0100[diff] [blame]2059ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice4dInt16, StridedSlice4dInt16Test)
2060ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice4dReverseInt16, StridedSlice4dReverseInt16Test)
2061ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSliceSimpleStrideInt16, StridedSliceSimpleStrideInt16Test)
2062ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSliceSimpleRangeMaskInt16, StridedSliceSimpleRangeMaskInt16Test)
2063ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSliceShrinkAxisMaskInt16, StridedSliceShrinkAxisMaskInt16Test)
2064ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice3dInt16, StridedSlice3dInt16Test)
2065ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice3dReverseInt16, StridedSlice3dReverseInt16Test)
2066ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice2dInt16, StridedSlice2dInt16Test)
2067ARMNN_AUTO_TEST_CASE_WITH_THF(StridedSlice2dReverseInt16, StridedSlice2dReverseInt16Test)
Matteo Martincigh42666a12019-05-29 08:53:41 +0100[diff] [blame]2068
Nattapat Chaimanowongcfdcadf2018-12-06 11:54:33 +0000[diff] [blame]2069// Debug
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]2070ARMNN_AUTO_TEST_CASE(Debug4dFloat32, Debug4dFloat32Test)
2071ARMNN_AUTO_TEST_CASE(Debug3dFloat32, Debug3dFloat32Test)
2072ARMNN_AUTO_TEST_CASE(Debug2dFloat32, Debug2dFloat32Test)
2073ARMNN_AUTO_TEST_CASE(Debug1dFloat32, Debug1dFloat32Test)
Nattapat Chaimanowongcfdcadf2018-12-06 11:54:33 +0000[diff] [blame]2074
Narumol Prangnawarat403a1852020-03-12 14:24:13 +0000[diff] [blame]2075ARMNN_AUTO_TEST_CASE(Debug4dBFloat16, Debug4dBFloat16Test)
2076ARMNN_AUTO_TEST_CASE(Debug3dBFloat16, Debug3dBFloat16Test)
2077ARMNN_AUTO_TEST_CASE(Debug2dBFloat16, Debug2dBFloat16Test)
2078ARMNN_AUTO_TEST_CASE(Debug1dBFloat16, Debug1dBFloat16Test)
2079
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]2080ARMNN_AUTO_TEST_CASE(Debug4dUint8, Debug4dUint8Test)
2081ARMNN_AUTO_TEST_CASE(Debug3dUint8, Debug3dUint8Test)
2082ARMNN_AUTO_TEST_CASE(Debug2dUint8, Debug2dUint8Test)
2083ARMNN_AUTO_TEST_CASE(Debug1dUint8, Debug1dUint8Test)
Nattapat Chaimanowongcfdcadf2018-12-06 11:54:33 +0000[diff] [blame]2084
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]2085ARMNN_AUTO_TEST_CASE(Debug4dQSymm16, Debug4dInt16Test)
2086ARMNN_AUTO_TEST_CASE(Debug3dQSymm16, Debug3dInt16Test)
2087ARMNN_AUTO_TEST_CASE(Debug2dQSymm16, Debug2dInt16Test)
2088ARMNN_AUTO_TEST_CASE(Debug1dQSymm16, Debug1dInt16Test)
Narumol Prangnawarat47cfee92019-07-04 10:29:00 +0100[diff] [blame]2089
narpra014951d842019-01-18 16:53:53 +0000[diff] [blame]2090// Gather
Finn Williamsc43de6a2020-08-27 11:13:25 +0100[diff] [blame]2091ARMNN_AUTO_TEST_CASE_WITH_THF(Gather1dParamsFloat32, Gather1dParamsFloat32Test)
2092ARMNN_AUTO_TEST_CASE_WITH_THF(Gather1dParamsFloat16, Gather1dParamsFloat16Test)
2093ARMNN_AUTO_TEST_CASE_WITH_THF(Gather1dParamsUint8, Gather1dParamsUint8Test)
2094ARMNN_AUTO_TEST_CASE_WITH_THF(Gather1dParamsInt16, Gather1dParamsInt16Test)
2095ARMNN_AUTO_TEST_CASE_WITH_THF(Gather1dParamsInt32, Gather1dParamsInt32Test)
2096ARMNN_AUTO_TEST_CASE_WITH_THF(GatherMultiDimParamsFloat32, GatherMultiDimParamsFloat32Test)
2097ARMNN_AUTO_TEST_CASE_WITH_THF(GatherMultiDimParamsFloat16, GatherMultiDimParamsFloat16Test)
2098ARMNN_AUTO_TEST_CASE_WITH_THF(GatherMultiDimParamsUint8, GatherMultiDimParamsUint8Test)
2099ARMNN_AUTO_TEST_CASE_WITH_THF(GatherMultiDimParamsInt16, GatherMultiDimParamsInt16Test)
2100ARMNN_AUTO_TEST_CASE_WITH_THF(GatherMultiDimParamsInt32, GatherMultiDimParamsInt32Test)
2101ARMNN_AUTO_TEST_CASE_WITH_THF(GatherMultiDimParamsMultiDimIndicesFloat32,
2102 GatherMultiDimParamsMultiDimIndicesFloat32Test)
2103ARMNN_AUTO_TEST_CASE_WITH_THF(GatherMultiDimParamsMultiDimIndicesFloat16,
2104 GatherMultiDimParamsMultiDimIndicesFloat16Test)
2105ARMNN_AUTO_TEST_CASE_WITH_THF(GatherMultiDimParamsMultiDimIndicesUint8, GatherMultiDimParamsMultiDimIndicesUint8Test)
2106ARMNN_AUTO_TEST_CASE_WITH_THF(GatherMultiDimParamsMultiDimIndicesInt16, GatherMultiDimParamsMultiDimIndicesInt16Test)
2107ARMNN_AUTO_TEST_CASE_WITH_THF(GatherMultiDimParamsMultiDimIndicesInt32, GatherMultiDimParamsMultiDimIndicesInt32Test)
narpra014951d842019-01-18 16:53:53 +0000[diff] [blame]2108
Sadik Armagan9199e582019-09-05 17:35:31 +0100[diff] [blame]2109// Abs
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]2110ARMNN_AUTO_TEST_CASE_WITH_THF(Abs2d, Abs2dTest<DataType::Float32>)
2111ARMNN_AUTO_TEST_CASE_WITH_THF(Abs3d, Abs3dTest<DataType::Float32>)
2112ARMNN_AUTO_TEST_CASE_WITH_THF(AbsZero, AbsZeroTest<DataType::Float32>)
2113ARMNN_AUTO_TEST_CASE_WITH_THF(Abs2dFloat16, Abs2dTest<DataType::Float16>)
2114ARMNN_AUTO_TEST_CASE_WITH_THF(Abs3dFloat16, Abs3dTest<DataType::Float16>)
2115ARMNN_AUTO_TEST_CASE_WITH_THF(Abs2dSigned32, Abs2dTest<DataType::Signed32>)
2116ARMNN_AUTO_TEST_CASE_WITH_THF(Abs3dSigned32, Abs3dTest<DataType::Signed32>)
2117ARMNN_AUTO_TEST_CASE_WITH_THF(AbsZeroSigned32, AbsZeroTest<DataType::Signed32>)
Kevin Mayec52c3a2020-04-24 09:42:31 +0100[diff] [blame]2118
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]2119ARMNN_AUTO_TEST_CASE_WITH_THF(Abs2dQuantisedAsymmS8, Abs2dTest<DataType::QAsymmS8>)
2120ARMNN_AUTO_TEST_CASE_WITH_THF(Abs3dQuantisedAsymmS8, Abs3dTest<DataType::QAsymmS8>)
2121ARMNN_AUTO_TEST_CASE_WITH_THF(Abs2dQuantisedAsymm8, Abs2dTest<DataType::QAsymmU8>)
2122ARMNN_AUTO_TEST_CASE_WITH_THF(Abs3dQuantisedAsymm8, Abs3dTest<DataType::QAsymmU8>)
2123ARMNN_AUTO_TEST_CASE_WITH_THF(Abs2dQuantisedSymm16, Abs2dTest<DataType::QSymmS16>)
2124ARMNN_AUTO_TEST_CASE_WITH_THF(Abs3dQuantisedSymm16, Abs3dTest<DataType::QSymmS16>)
Sadik Armagan9199e582019-09-05 17:35:31 +0100[diff] [blame]2125
Narumol Prangnawarate0a4ad82019-02-04 19:05:27 +0000[diff] [blame]2126// Detection PostProcess
Sadik Armagan1625efc2021-06-10 18:24:34 +0100[diff] [blame]2127TEST_CASE("DetectionPostProcessRegularNmsFloat")
Narumol Prangnawarate0a4ad82019-02-04 19:05:27 +0000[diff] [blame]2128{
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]2129 DetectionPostProcessRegularNmsFloatTest<RefWorkloadFactory>();
Narumol Prangnawarate0a4ad82019-02-04 19:05:27 +0000[diff] [blame]2130}
Sadik Armagan1625efc2021-06-10 18:24:34 +0100[diff] [blame]2131TEST_CASE("DetectionPostProcessFastNmsFloat")
Narumol Prangnawarate0a4ad82019-02-04 19:05:27 +0000[diff] [blame]2132{
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]2133 DetectionPostProcessFastNmsFloatTest<RefWorkloadFactory>();
Narumol Prangnawarate0a4ad82019-02-04 19:05:27 +0000[diff] [blame]2134}
Sadik Armagan1625efc2021-06-10 18:24:34 +0100[diff] [blame]2135TEST_CASE("DetectionPostProcessRegularNmsInt8")
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]2136{
Sadik Armagan1625efc2021-06-10 18:24:34 +0100[diff] [blame]2137 DetectionPostProcessRegularNmsQuantizedTest<RefWorkloadFactory, DataType::QAsymmS8>();
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]2138}
Sadik Armagan1625efc2021-06-10 18:24:34 +0100[diff] [blame]2139TEST_CASE("DetectionPostProcessFastNmsInt8")
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]2140{
Sadik Armagan1625efc2021-06-10 18:24:34 +0100[diff] [blame]2141 DetectionPostProcessRegularNmsQuantizedTest<RefWorkloadFactory, DataType::QAsymmS8>();
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]2142}
Sadik Armagan1625efc2021-06-10 18:24:34 +0100[diff] [blame]2143TEST_CASE("DetectionPostProcessRegularNmsUint8")
Narumol Prangnawarate0a4ad82019-02-04 19:05:27 +0000[diff] [blame]2144{
Sadik Armagan1625efc2021-06-10 18:24:34 +0100[diff] [blame]2145 DetectionPostProcessRegularNmsQuantizedTest<RefWorkloadFactory, DataType::QAsymmU8>();
Narumol Prangnawarate0a4ad82019-02-04 19:05:27 +0000[diff] [blame]2146}
Sadik Armagan1625efc2021-06-10 18:24:34 +0100[diff] [blame]2147TEST_CASE("DetectionPostProcessFastNmsUint8")
Narumol Prangnawarate0a4ad82019-02-04 19:05:27 +0000[diff] [blame]2148{
Sadik Armagan1625efc2021-06-10 18:24:34 +0100[diff] [blame]2149 DetectionPostProcessRegularNmsQuantizedTest<RefWorkloadFactory, DataType::QAsymmU8>();
Aron Virginas-Tar6331f912019-06-03 17:10:02 +0100[diff] [blame]2150}
Sadik Armagan1625efc2021-06-10 18:24:34 +0100[diff] [blame]2151TEST_CASE("DetectionPostProcessRegularNmsInt16")
Aron Virginas-Tar6331f912019-06-03 17:10:02 +0100[diff] [blame]2152{
Sadik Armagan1625efc2021-06-10 18:24:34 +0100[diff] [blame]2153 DetectionPostProcessRegularNmsQuantizedTest<RefWorkloadFactory, DataType::QSymmS16>();
Aron Virginas-Tar6331f912019-06-03 17:10:02 +0100[diff] [blame]2154}
Sadik Armagan1625efc2021-06-10 18:24:34 +0100[diff] [blame]2155TEST_CASE("DetectionPostProcessFastNmsInt16")
Aron Virginas-Tar6331f912019-06-03 17:10:02 +0100[diff] [blame]2156{
Sadik Armagan1625efc2021-06-10 18:24:34 +0100[diff] [blame]2157 DetectionPostProcessFastNmsQuantizedTest<RefWorkloadFactory, DataType::QSymmS16>();
Narumol Prangnawarate0a4ad82019-02-04 19:05:27 +0000[diff] [blame]2158}
2159
Nattapat Chaimanowong8a54ac02019-03-29 15:25:04 +0000[diff] [blame]2160// Dequantize
Nattapat Chaimanowongafa4e3a2019-04-02 11:41:45 +0100[diff] [blame]2161ARMNN_AUTO_TEST_CASE(DequantizeSimpleUint8, DequantizeSimpleUint8Test)
2162ARMNN_AUTO_TEST_CASE(DequantizeOffsetUint8, DequantizeOffsetUint8Test)
Ryan OShea9add1202020-02-07 10:06:33 +0000[diff] [blame]2163ARMNN_AUTO_TEST_CASE(DequantizeSimpleAsymmInt8, DequantizeSimpleAsymmInt8Test)
2164ARMNN_AUTO_TEST_CASE(DequantizeOffsetAsymmInt8, DequantizeOffsetAsymmInt8Test)
Finn Williamsfd271062019-12-04 14:27:27 +0000[diff] [blame]2165ARMNN_AUTO_TEST_CASE(DequantizeSimpleInt8, DequantizeSimpleInt8Test)
Nattapat Chaimanowongafa4e3a2019-04-02 11:41:45 +0100[diff] [blame]2166ARMNN_AUTO_TEST_CASE(DequantizeSimpleInt16, DequantizeSimpleInt16Test)
Jan Eilersf7107932019-11-01 11:09:36 +0000[diff] [blame]2167ARMNN_AUTO_TEST_CASE(DequantizeSimpleUint8ToFp16, DequantizeSimpleUint8ToFp16Test)
Finn Williamsfd271062019-12-04 14:27:27 +0000[diff] [blame]2168ARMNN_AUTO_TEST_CASE(DequantizeSimpleInt8ToFp16, DequantizeSimpleInt8ToFp16Test)
Jan Eilersf7107932019-11-01 11:09:36 +0000[diff] [blame]2169ARMNN_AUTO_TEST_CASE(DequantizeSimpleInt16ToFp16, DequantizeSimpleInt16ToFp16Test)
Narumol Prangnawarate0a4ad82019-02-04 19:05:27 +0000[diff] [blame]2170
Nattapat Chaimanowonga0beb3b2019-04-01 17:04:53 +0100[diff] [blame]2171// Quantize
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]2172ARMNN_AUTO_TEST_CASE_WITH_THF(QuantizeSimpleUint8, QuantizeSimpleUint8Test)
2173ARMNN_AUTO_TEST_CASE_WITH_THF(QuantizeClampUint8, QuantizeClampUint8Test)
2174ARMNN_AUTO_TEST_CASE_WITH_THF(QuantizeClampAsymmInt8, QuantizeClampAsymmInt8Test)
2175ARMNN_AUTO_TEST_CASE_WITH_THF(QuantizeClampInt8, QuantizeClampInt8Test)
2176ARMNN_AUTO_TEST_CASE_WITH_THF(QuantizeClampInt16, QuantizeClampInt16Test)
Nattapat Chaimanowonga0beb3b2019-04-01 17:04:53 +0100[diff] [blame]2177
Matteo Martincighab9e5252019-06-13 17:27:46 +0100[diff] [blame]2178// PReLU
Finn Williams826a5432020-08-27 16:15:20 +0100[diff] [blame]2179ARMNN_AUTO_TEST_CASE_WITH_THF(PreluFloat32, PreluTest<DataType::Float32>)
2180ARMNN_AUTO_TEST_CASE_WITH_THF(PreluFloat16, PreluTest<DataType::Float16>)
2181ARMNN_AUTO_TEST_CASE_WITH_THF(PreluUint8, PreluTest<DataType::QAsymmU8>)
2182ARMNN_AUTO_TEST_CASE_WITH_THF(PreluInt16, PreluTest<DataType::QSymmS16>)
Matteo Martincighab9e5252019-06-13 17:27:46 +0100[diff] [blame]2183
Aron Virginas-Tar92b9f872019-09-17 17:27:04 +0100[diff] [blame]2184// Slice
2185ARMNN_AUTO_TEST_CASE(Slice4dFloat32, Slice4dFloat32Test)
2186ARMNN_AUTO_TEST_CASE(Slice3dFloat32, Slice3dFloat32Test)
2187ARMNN_AUTO_TEST_CASE(Slice2dFloat32, Slice2dFloat32Test)
2188ARMNN_AUTO_TEST_CASE(Slice1dFloat32, Slice1dFloat32Test)
2189
2190ARMNN_AUTO_TEST_CASE(Slice4dUint8, Slice4dUint8Test)
2191ARMNN_AUTO_TEST_CASE(Slice3dUint8, Slice3dUint8Test)
2192ARMNN_AUTO_TEST_CASE(Slice2dUint8, Slice2dUint8Test)
2193ARMNN_AUTO_TEST_CASE(Slice1dUint8, Slice1dUint8Test)
2194
2195ARMNN_AUTO_TEST_CASE(Slice4dInt16, Slice4dInt16Test)
2196ARMNN_AUTO_TEST_CASE(Slice3dInt16, Slice3dInt16Test)
2197ARMNN_AUTO_TEST_CASE(Slice2dInt16, Slice2dInt16Test)
2198ARMNN_AUTO_TEST_CASE(Slice1dInt16, Slice1dInt16Test)
2199
Mike Kellyc9ea45a2020-02-28 18:11:58 +0000[diff] [blame]2200// Transpose
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]2201ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleTransposeBFloat16, SimpleTransposeTest<DataType::BFloat16>)
2202ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeBFloat16ValueSet1Test, TransposeValueSet1Test<DataType::BFloat16>)
2203ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeBFloat16ValueSet2Test, TransposeValueSet2Test<DataType::BFloat16>)
2204ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeBFloat16ValueSet3Test, TransposeValueSet3Test<DataType::BFloat16>)
2205ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleTransposeFloat32, SimpleTransposeTest<DataType::Float32>)
2206ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeFloat32ValueSet1Test, TransposeValueSet1Test<DataType::Float32>)
2207ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeFloat32ValueSet2Test, TransposeValueSet2Test<DataType::Float32>)
2208ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeFloat32ValueSet3Test, TransposeValueSet3Test<DataType::Float32>)
2209ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleTransposeQASymmS8, SimpleTransposeTest<DataType::QAsymmS8>)
2210ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeQASymmS8ValueSet1Test, TransposeValueSet1Test<DataType::QAsymmS8>)
2211ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeQASymmS8ValueSet2Test, TransposeValueSet2Test<DataType::QAsymmS8>)
2212ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeQASymmS8ValueSet3Test, TransposeValueSet3Test<DataType::QAsymmS8>)
2213ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleTransposeQASymm8, SimpleTransposeTest<DataType::QAsymmU8>)
2214ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeQASymm8ValueSet1Test, TransposeValueSet1Test<DataType::QAsymmU8>)
2215ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeQASymm8ValueSet2Test, TransposeValueSet2Test<DataType::QAsymmU8>)
2216ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeQASymm8ValueSet3Test, TransposeValueSet3Test<DataType::QAsymmU8>)
2217ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleTransposeQSymm16, SimpleTransposeTest<DataType::QSymmS16>)
2218ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeQSymm16ValueSet1Test, TransposeValueSet1Test<DataType::QSymmS16>)
2219ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeQSymm16ValueSet2Test, TransposeValueSet2Test<DataType::QSymmS16>)
2220ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeQSymm16ValueSet3Test, TransposeValueSet3Test<DataType::QSymmS16>)
Mike Kellyc9ea45a2020-02-28 18:11:58 +0000[diff] [blame]2221
Aron Virginas-Tar735a4502019-06-26 15:02:47 +0100[diff] [blame]2222// TransposeConvolution2d
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]2223ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleTransposeConvolution2dFloatNchw,
2224 SimpleTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
2225 true,
2226 DataLayout::NCHW)
2227ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleTransposeConvolution2dFloatNhwc,
2228 SimpleTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
2229 true,
2230 DataLayout::NHWC)
2231ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleTransposeConvolution2dInt8Nchw,
2232 SimpleTransposeConvolution2dTest<DataType::QAsymmS8, DataType::Signed32>,
2233 true,
2234 DataLayout::NCHW)
2235ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleTransposeConvolution2dInt8Nhwc,
2236 SimpleTransposeConvolution2dTest<DataType::QAsymmS8, DataType::Signed32>,
2237 true,
2238 DataLayout::NHWC)
2239ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleTransposeConvolution2dUint8Nchw,
2240 SimpleTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
2241 true,
2242 DataLayout::NCHW)
2243ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleTransposeConvolution2dUint8Nhwc,
2244 SimpleTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
2245 true,
2246 DataLayout::NHWC)
2247ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleTransposeConvolution2dInt16Nchw,
2248 SimpleTransposeConvolution2dTest<DataType::QSymmS16, DataType::Signed32>,
2249 true,
2250 DataLayout::NCHW)
2251ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleTransposeConvolution2dInt16Nhwc,
2252 SimpleTransposeConvolution2dTest<DataType::QSymmS16, DataType::Signed32>,
2253 true,
2254 DataLayout::NCHW)
Aron Virginas-Tar735a4502019-06-26 15:02:47 +0100[diff] [blame]2255
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]2256ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedSimpleTransposeConvolution2dFloatNchw,
2257 SimpleTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
2258 false,
2259 DataLayout::NCHW)
2260ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedSimpleTransposeConvolution2dFloatNhwc,
2261 SimpleTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
2262 true,
2263 DataLayout::NHWC)
2264ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedSimpleTransposeConvolution2dInt8Nchw,
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]2265 SimpleTransposeConvolution2dTest<DataType::QAsymmS8, DataType::Signed32>,
2266 true,
2267 DataLayout::NCHW)
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]2268ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedSimpleTransposeConvolution2dInt8Nhwc,
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]2269 SimpleTransposeConvolution2dTest<DataType::QAsymmS8, DataType::Signed32>,
2270 true,
2271 DataLayout::NHWC)
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]2272ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedSimpleTransposeConvolution2dUint8Nchw,
2273 SimpleTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
2274 true,
2275 DataLayout::NCHW)
2276ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedSimpleTransposeConvolution2dUint8Nhwc,
2277 SimpleTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
2278 true,
2279 DataLayout::NHWC)
2280ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedSimpleTransposeConvolution2dInt16Nchw,
2281 SimpleTransposeConvolution2dTest<DataType::QSymmS16, DataType::Signed32>,
2282 true,
2283 DataLayout::NCHW)
2284ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedSimpleTransposeConvolution2dInt16Nhwc,
2285 SimpleTransposeConvolution2dTest<DataType::QSymmS16, DataType::Signed32>,
2286 true,
2287 DataLayout::NCHW)
Aron Virginas-Tar735a4502019-06-26 15:02:47 +0100[diff] [blame]2288
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]2289ARMNN_AUTO_TEST_CASE_WITH_THF(PaddedTransposeConvolution2dFloatNchw,
2290 PaddedTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
2291 true,
2292 DataLayout::NCHW)
2293ARMNN_AUTO_TEST_CASE_WITH_THF(PaddedTransposeConvolution2dFloatNhwc,
2294 PaddedTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
2295 true,
2296 DataLayout::NHWC)
2297ARMNN_AUTO_TEST_CASE_WITH_THF(PaddedTransposeConvolution2dInt8Nchw,
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]2298 PaddedTransposeConvolution2dTest<DataType::QAsymmS8, DataType::Signed32>,
2299 true,
2300 DataLayout::NCHW)
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]2301ARMNN_AUTO_TEST_CASE_WITH_THF(PaddedTransposeConvolution2dInt8Nhwc,
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]2302 PaddedTransposeConvolution2dTest<DataType::QAsymmS8, DataType::Signed32>,
2303 true,
2304 DataLayout::NHWC)
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]2305ARMNN_AUTO_TEST_CASE_WITH_THF(PaddedTransposeConvolution2dUint8Nchw,
2306 PaddedTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
2307 true,
2308 DataLayout::NCHW)
2309ARMNN_AUTO_TEST_CASE_WITH_THF(PaddedTransposeConvolution2dUint8Nhwc,
2310 PaddedTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
2311 true,
2312 DataLayout::NHWC)
2313ARMNN_AUTO_TEST_CASE_WITH_THF(PaddedTransposeConvolution2dInt16Nchw,
2314 PaddedTransposeConvolution2dTest<DataType::QSymmS16, DataType::Signed32>,
2315 true,
2316 DataLayout::NCHW)
2317ARMNN_AUTO_TEST_CASE_WITH_THF(PaddedTransposeConvolution2dInt16Nhwc,
2318 PaddedTransposeConvolution2dTest<DataType::QSymmS16, DataType::Signed32>,
2319 true,
2320 DataLayout::NCHW)
Aron Virginas-Tar735a4502019-06-26 15:02:47 +0100[diff] [blame]2321
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]2322ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedPaddedTransposeConvolution2dFloatNchw,
2323 PaddedTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
2324 false,
2325 DataLayout::NCHW)
2326ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedPaddedTransposeConvolution2dFloatNhwc,
2327 PaddedTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
2328 true,
2329 DataLayout::NHWC)
2330ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedPaddedTransposeConvolution2dInt8Nchw,
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]2331 PaddedTransposeConvolution2dTest<DataType::QAsymmS8, DataType::Signed32>,
2332 true,
2333 DataLayout::NCHW)
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]2334ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedPaddedTransposeConvolution2dInt8Nhwc,
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]2335 PaddedTransposeConvolution2dTest<DataType::QAsymmS8, DataType::Signed32>,
2336 true,
2337 DataLayout::NHWC)
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]2338ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedPaddedTransposeConvolution2dUint8Nchw,
2339 PaddedTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
2340 true,
2341 DataLayout::NCHW)
2342ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedPaddedTransposeConvolution2dUint8Nhwc,
2343 PaddedTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
2344 true,
2345 DataLayout::NHWC)
2346ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedPaddedTransposeConvolution2dInt16Nchw,
2347 PaddedTransposeConvolution2dTest<DataType::QSymmS16, DataType::Signed32>,
2348 true,
2349 DataLayout::NCHW)
2350ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedPaddedTransposeConvolution2dInt16Nhwc,
2351 PaddedTransposeConvolution2dTest<DataType::QSymmS16, DataType::Signed32>,
2352 true,
2353 DataLayout::NCHW)
Aron Virginas-Tar735a4502019-06-26 15:02:47 +0100[diff] [blame]2354
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]2355ARMNN_AUTO_TEST_CASE_WITH_THF(StridedTransposeConvolution2dFloatNchw,
2356 StridedTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
2357 true,
2358 DataLayout::NCHW)
2359ARMNN_AUTO_TEST_CASE_WITH_THF(StridedTransposeConvolution2dFloatNhwc,
2360 StridedTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
2361 true,
2362 DataLayout::NHWC)
2363ARMNN_AUTO_TEST_CASE_WITH_THF(StridedTransposeConvolution2dInt8Nchw,
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]2364 StridedTransposeConvolution2dTest<DataType::QAsymmS8, DataType::Signed32>,
2365 true,
2366 DataLayout::NCHW)
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]2367ARMNN_AUTO_TEST_CASE_WITH_THF(StridedTransposeConvolution2dInt8Nhwc,
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]2368 StridedTransposeConvolution2dTest<DataType::QAsymmS8, DataType::Signed32>,
2369 true,
2370 DataLayout::NHWC)
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]2371ARMNN_AUTO_TEST_CASE_WITH_THF(StridedTransposeConvolution2dUint8Nchw,
2372 StridedTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
2373 true,
2374 DataLayout::NCHW)
2375ARMNN_AUTO_TEST_CASE_WITH_THF(StridedTransposeConvolution2dUint8Nhwc,
2376 StridedTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
2377 true,
2378 DataLayout::NHWC)
2379ARMNN_AUTO_TEST_CASE_WITH_THF(StridedTransposeConvolution2dInt16Nchw,
2380 StridedTransposeConvolution2dTest<DataType::QSymmS16, DataType::Signed32>,
2381 true,
2382 DataLayout::NCHW)
2383ARMNN_AUTO_TEST_CASE_WITH_THF(StridedTransposeConvolution2dInt16Nhwc,
2384 StridedTransposeConvolution2dTest<DataType::QSymmS16, DataType::Signed32>,
2385 true,
2386 DataLayout::NCHW)
Aron Virginas-Tar735a4502019-06-26 15:02:47 +0100[diff] [blame]2387
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]2388ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedStridedTransposeConvolution2dFloatNchw,
2389 StridedTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
2390 false,
2391 DataLayout::NCHW)
2392ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedStridedTransposeConvolution2dFloatNhwc,
2393 StridedTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
2394 true,
2395 DataLayout::NHWC)
2396ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedStridedTransposeConvolution2dInt8Nchw,
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]2397 StridedTransposeConvolution2dTest<DataType::QAsymmS8, DataType::Signed32>,
2398 true,
2399 DataLayout::NCHW)
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]2400ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedStridedTransposeConvolution2dInt8Nhwc,
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]2401 StridedTransposeConvolution2dTest<DataType::QAsymmS8, DataType::Signed32>,
2402 true,
2403 DataLayout::NHWC)
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]2404ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedStridedTransposeConvolution2dUint8Nchw,
2405 StridedTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
2406 true,
2407 DataLayout::NCHW)
2408ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedStridedTransposeConvolution2dUint8Nhwc,
2409 StridedTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
2410 true,
2411 DataLayout::NHWC)
2412ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedStridedTransposeConvolution2dInt16Nchw,
2413 StridedTransposeConvolution2dTest<DataType::QSymmS16, DataType::Signed32>,
2414 true,
2415 DataLayout::NCHW)
2416ARMNN_AUTO_TEST_CASE_WITH_THF(UnbiasedStridedTransposeConvolution2dInt16Nhwc,
2417 StridedTransposeConvolution2dTest<DataType::QSymmS16, DataType::Signed32>,
2418 true,
2419 DataLayout::NCHW)
Aron Virginas-Tard8edabb2019-08-12 14:29:59 +0100[diff] [blame]2420
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]2421ARMNN_AUTO_TEST_CASE_WITH_THF(MultiChannelTransposeConvolution2dFloatNchw,
2422 MultiChannelTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
2423 DataLayout::NCHW)
2424ARMNN_AUTO_TEST_CASE_WITH_THF(MultiChannelTransposeConvolution2dFloatNhwc,
2425 MultiChannelTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
2426 DataLayout::NHWC)
2427ARMNN_AUTO_TEST_CASE_WITH_THF(MultiChannelTransposeConvolution2dInt8Nchw,
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]2428 MultiChannelTransposeConvolution2dTest<DataType::QAsymmS8, DataType::Signed32>,
2429 DataLayout::NCHW)
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]2430ARMNN_AUTO_TEST_CASE_WITH_THF(MultiChannelTransposeConvolution2dInt8Nhwc,
Sadik Armagan303980c2020-04-17 12:45:14 +0100[diff] [blame]2431 MultiChannelTransposeConvolution2dTest<DataType::QAsymmS8, DataType::Signed32>,
2432 DataLayout::NHWC)
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]2433ARMNN_AUTO_TEST_CASE_WITH_THF(MultiChannelTransposeConvolution2dUint8Nchw,
2434 MultiChannelTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
2435 DataLayout::NCHW)
2436ARMNN_AUTO_TEST_CASE_WITH_THF(MultiChannelTransposeConvolution2dUint8Nhwc,
2437 MultiChannelTransposeConvolution2dTest<DataType::QAsymmU8, DataType::Signed32>,
2438 DataLayout::NHWC)
2439ARMNN_AUTO_TEST_CASE_WITH_THF(MultiChannelTransposeConvolution2dInt16Nchw,
2440 MultiChannelTransposeConvolution2dTest<DataType::QSymmS16, DataType::Signed32>,
2441 DataLayout::NCHW)
2442ARMNN_AUTO_TEST_CASE_WITH_THF(MultiChannelTransposeConvolution2dInt16Nhwc,
2443 MultiChannelTransposeConvolution2dTest<DataType::QSymmS16, DataType::Signed32>,
2444 DataLayout::NCHW)
Aron Virginas-Tar735a4502019-06-26 15:02:47 +0100[diff] [blame]2445
Finn Williamsec36d3e2020-08-28 13:17:05 +0100[diff] [blame]2446ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeConvolution2dPerAxisQuantTestNchw,
2447 TransposeConvolution2dPerAxisQuantTest,
2448 DataLayout::NCHW);
2449ARMNN_AUTO_TEST_CASE_WITH_THF(TransposeConvolution2dPerAxisQuantTestNhwc,
2450 TransposeConvolution2dPerAxisQuantTest,
2451 DataLayout::NHWC);
Aron Virginas-Tar94d3b932019-11-11 12:54:47 +0000[diff] [blame]2452
Matthew Jackson81e601c2019-07-11 12:07:09 +0100[diff] [blame]2453// Stack
Sadik Armagan5e719842020-08-28 12:17:33 +0100[diff] [blame]2454ARMNN_AUTO_TEST_CASE_WITH_THF(Stack0Axis, StackAxis0Float32Test)
2455ARMNN_AUTO_TEST_CASE_WITH_THF(StackOutput4DAxis1, StackOutput4DAxis1Float32Test)
2456ARMNN_AUTO_TEST_CASE_WITH_THF(StackOutput4DAxis2, StackOutput4DAxis2Float32Test)
2457ARMNN_AUTO_TEST_CASE_WITH_THF(StackOutput4DAxis3, StackOutput4DAxis3Float32Test)
2458ARMNN_AUTO_TEST_CASE_WITH_THF(StackOutput3DInputs3, StackOutput3DInputs3Float32Test)
2459ARMNN_AUTO_TEST_CASE_WITH_THF(StackOutput5D, StackOutput5DFloat32Test)
2460ARMNN_AUTO_TEST_CASE_WITH_THF(StackFloat16, StackFloat16Test)
Sadik Armagan529195f2022-01-14 12:56:35 +0000[diff] [blame^]2461ARMNN_AUTO_TEST_CASE_WITH_THF(StackInt32, StackInt32Test)
Matthew Jackson81e601c2019-07-11 12:07:09 +0100[diff] [blame]2462
Sadik Armaganac472102020-03-24 09:54:36 +0000[diff] [blame]2463// Neg
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]2464ARMNN_AUTO_TEST_CASE_WITH_THF(Neg2d, Neg2dTest<DataType::Float32>)
2465ARMNN_AUTO_TEST_CASE_WITH_THF(Neg3d, Neg3dTest<DataType::Float32>)
2466ARMNN_AUTO_TEST_CASE_WITH_THF(NegZero, NegZeroTest<DataType::Float32>)
2467ARMNN_AUTO_TEST_CASE_WITH_THF(NegNegative, NegNegativeTest<DataType::Float32>)
2468ARMNN_AUTO_TEST_CASE_WITH_THF(Neg2dFloat16, Neg2dTest<DataType::Float16>)
2469ARMNN_AUTO_TEST_CASE_WITH_THF(Neg3dFloat16, Neg3dTest<DataType::Float16>)
2470ARMNN_AUTO_TEST_CASE_WITH_THF(Neg2dQuantisedAsymmS8, Neg2dTest<DataType::QAsymmS8>)
2471ARMNN_AUTO_TEST_CASE_WITH_THF(Neg3dQuantisedAsymmS8, Neg3dTest<DataType::QAsymmS8>)
2472ARMNN_AUTO_TEST_CASE_WITH_THF(Neg2dQuantisedAsymm8, Neg2dTest<DataType::QAsymmU8>)
2473ARMNN_AUTO_TEST_CASE_WITH_THF(Neg3dQuantisedAsymm8, Neg3dTest<DataType::QAsymmU8>)
2474ARMNN_AUTO_TEST_CASE_WITH_THF(Neg2dQuantisedSymm16, Neg2dTest<DataType::QSymmS16>)
2475ARMNN_AUTO_TEST_CASE_WITH_THF(Neg3dQuantisedSymm16, Neg3dTest<DataType::QSymmS16>)
Sadik Armaganac472102020-03-24 09:54:36 +0000[diff] [blame]2476
Sadik Armagan9fabf432020-05-27 13:40:58 +0100[diff] [blame]2477// Exp
Keith Davis33a626f2020-08-27 15:38:12 +0100[diff] [blame]2478ARMNN_AUTO_TEST_CASE_WITH_THF(Exp2d, Exp2dTest<DataType::Float32>)
2479ARMNN_AUTO_TEST_CASE_WITH_THF(Exo3d, Exp3dTest<DataType::Float32>)
2480ARMNN_AUTO_TEST_CASE_WITH_THF(ExpZero, ExpZeroTest<DataType::Float32>)
2481ARMNN_AUTO_TEST_CASE_WITH_THF(ExpNegative, ExpNegativeTest<DataType::Float32>)
2482ARMNN_AUTO_TEST_CASE_WITH_THF(Exp2dFloat16, Exp2dTest<DataType::Float16>)
2483ARMNN_AUTO_TEST_CASE_WITH_THF(Exp3dFloat16, Exp3dTest<DataType::Float16>)
2484ARMNN_AUTO_TEST_CASE_WITH_THF(Exp2dQuantisedAsymmS8, Exp2dTest<DataType::QAsymmS8>)
2485ARMNN_AUTO_TEST_CASE_WITH_THF(Exp3dQuantisedAsymmS8, Exp3dTest<DataType::QAsymmS8>)
2486ARMNN_AUTO_TEST_CASE_WITH_THF(Exp2dQuantisedAsymm8, Exp2dTest<DataType::QAsymmU8>)
2487ARMNN_AUTO_TEST_CASE_WITH_THF(Exp3dQuantisedAsymm8, Exp3dTest<DataType::QAsymmU8>)
2488ARMNN_AUTO_TEST_CASE_WITH_THF(Exp2dQuantisedSymm16, Exp2dTest<DataType::QSymmS16>)
2489ARMNN_AUTO_TEST_CASE_WITH_THF(Exp3dQuantisedSymm16, Exp3dTest<DataType::QSymmS16>)
Sadik Armagan9fabf432020-05-27 13:40:58 +0100[diff] [blame]2490
Teresa Charlin50de4fa2021-05-31 18:47:33 +0100[diff] [blame]2491// Log
2492ARMNN_AUTO_TEST_CASE_WITH_THF(Log2d, Log2dTest<DataType::Float32>)
2493ARMNN_AUTO_TEST_CASE_WITH_THF(Log3d, Log3dTest<DataType::Float32>)
2494ARMNN_AUTO_TEST_CASE_WITH_THF(LogZero, LogZeroTest<DataType::Float32>)
2495ARMNN_AUTO_TEST_CASE_WITH_THF(LogNegative, LogNegativeTest<DataType::Float32>)
2496ARMNN_AUTO_TEST_CASE_WITH_THF(Log2dFloat16, Log2dTest<DataType::Float16>)
2497ARMNN_AUTO_TEST_CASE_WITH_THF(Log3dFloat16, Log3dTest<DataType::Float16>)
2498ARMNN_AUTO_TEST_CASE_WITH_THF(Log2dQuantisedAsymmS8, Log2dTest<DataType::QAsymmS8>)
2499ARMNN_AUTO_TEST_CASE_WITH_THF(Log3dQuantisedAsymmS8, Log3dTest<DataType::QAsymmS8>)
2500ARMNN_AUTO_TEST_CASE_WITH_THF(Log2dQuantisedAsymm8, Log2dTest<DataType::QAsymmU8>)
2501ARMNN_AUTO_TEST_CASE_WITH_THF(Log3dQuantisedAsymm8, Log3dTest<DataType::QAsymmU8>)
2502ARMNN_AUTO_TEST_CASE_WITH_THF(Log2dQuantisedSymm16, Log2dTest<DataType::QSymmS16>)
2503ARMNN_AUTO_TEST_CASE_WITH_THF(Log3dQuantisedSymm16, Log3dTest<DataType::QSymmS16>)
2504
2505// Sin
2506ARMNN_AUTO_TEST_CASE_WITH_THF(Sin2d, Sin2dTest<DataType::Float32>)
2507ARMNN_AUTO_TEST_CASE_WITH_THF(Sin3d, Sin3dTest<DataType::Float32>)
2508ARMNN_AUTO_TEST_CASE_WITH_THF(SinZero, SinZeroTest<DataType::Float32>)
2509ARMNN_AUTO_TEST_CASE_WITH_THF(SinNegative, SinNegativeTest<DataType::Float32>)
2510ARMNN_AUTO_TEST_CASE_WITH_THF(Sin2dFloat16, Sin2dTest<DataType::Float16>)
2511ARMNN_AUTO_TEST_CASE_WITH_THF(Sin3dFloat16, Sin3dTest<DataType::Float16>)
2512ARMNN_AUTO_TEST_CASE_WITH_THF(Sin2dQuantisedAsymmS8, Sin2dTest<DataType::QAsymmS8>)
2513ARMNN_AUTO_TEST_CASE_WITH_THF(Sin3dQuantisedAsymmS8, Sin3dTest<DataType::QAsymmS8>)
2514ARMNN_AUTO_TEST_CASE_WITH_THF(Sin2dQuantisedAsymm8, Sin2dTest<DataType::QAsymmU8>)
2515ARMNN_AUTO_TEST_CASE_WITH_THF(Sin3dQuantisedAsymm8, Sin3dTest<DataType::QAsymmU8>)
2516ARMNN_AUTO_TEST_CASE_WITH_THF(Sin2dQuantisedSymm16, Sin2dTest<DataType::QSymmS16>)
2517ARMNN_AUTO_TEST_CASE_WITH_THF(Sin3dQuantisedSymm16, Sin3dTest<DataType::QSymmS16>)
2518
2519//Logical
James Conroyaba90cd2020-11-06 16:28:18 +0000[diff] [blame]2520ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalNot, LogicalNotTest)
2521ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalNotInt, LogicalNotIntTest)
2522ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalAnd, LogicalAndTest)
2523ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalOr, LogicalOrTest)
2524ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalAndInt, LogicalAndIntTest)
2525ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalOrInt, LogicalOrIntTest)
2526ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalAndBroadcast1, LogicalAndBroadcast1Test)
2527ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalOrBroadcast1, LogicalOrBroadcast1Test)
2528ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalAndBroadcast2, LogicalAndBroadcast2Test)
2529ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalOrBroadcast2, LogicalOrBroadcast2Test)
2530ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalAndBroadcast3, LogicalAndBroadcast3Test)
2531ARMNN_AUTO_TEST_CASE_WITH_THF(LogicalOrBroadcast3, LogicalOrBroadcast3Test)
2532
Sadik Armagan0c3ea5b2021-02-03 09:29:30 +0000[diff] [blame]2533// ReduceSum
2534ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceSumFloat32, ReduceSumSimpleTest<DataType::Float32>)
2535ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceSumSingleAxisFloat32_1, ReduceSumSingleAxisTest1<DataType::Float32>)
2536ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceSumSingleAxisFloat32_2, ReduceSumSingleAxisTest2<DataType::Float32>)
2537ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceSumSingleAxisFloat32_3, ReduceSumSingleAxisTest3<DataType::Float32>)
2538ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceSumMultipleAxisFloat32, ReduceSumMultipleAxisTest<DataType::Float32>)
2539
Teresa Charlin4e3e8312021-08-05 12:34:37 +0100[diff] [blame]2540// ReduceProd
2541ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceProdFloat32, ReduceProdSimpleTest<DataType::Float32>)
2542ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceProdSingleAxisFloat32_1, ReduceProdSingleAxisTest1<DataType::Float32>)
2543ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceProdSingleAxisFloat32_2, ReduceProdSingleAxisTest2<DataType::Float32>)
2544ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceProdSingleAxisFloat32_3, ReduceProdSingleAxisTest3<DataType::Float32>)
2545ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceProdMultipleAxisFloat32, ReduceProdMultipleAxisTest<DataType::Float32>)
2546
Sadik Armagana2747482021-02-09 10:28:54 +0000[diff] [blame]2547// ReduceMax
2548ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceMaxFloat32, ReduceMaxSimpleTest<DataType::Float32>)
2549ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceMaxNegativeAxisFloat32, ReduceMaxNegativeAxisTest<DataType::Float32>)
2550ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceMax2Float32, ReduceMaxSimpleTest2<DataType::Float32>)
2551
2552// ReduceMin
2553ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceMinFloat32, ReduceMinSimpleTest<DataType::Float32>)
2554ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceMinNegativeAxisFloat32, ReduceMinNegativeAxisTest<DataType::Float32>)
2555
Narumol Prangnawarate5339e72021-07-28 17:33:28 +0100[diff] [blame]2556// Unidirectional Sequence Lstm
2557ARMNN_AUTO_TEST_CASE_WITH_THF(UnidirectionalSequenceLstmLayerFloat32,
2558 UnidirectionalSequenceLstmLayerFloat32Test)
2559ARMNN_AUTO_TEST_CASE_WITH_THF(UnidirectionalSequenceLstmLayerFloat32TimeMajor,
2560 UnidirectionalSequenceLstmLayerFloat32TimeMajorTest)
2561ARMNN_AUTO_TEST_CASE_WITH_THF(UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjection,
2562 UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest)
2563ARMNN_AUTO_TEST_CASE_WITH_THF(UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNorm,
2564 UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest)
2565ARMNN_AUTO_TEST_CASE_WITH_THF(UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjection,
2566 UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest)
Narumol Prangnawaratbd575b22021-08-31 16:53:54 +0100[diff] [blame]2567ARMNN_AUTO_TEST_CASE_WITH_THF(UnidirectionalSequenceLstmLayerInt8,
2568 UnidirectionalSequenceLstmLayerInt8Test)
2569ARMNN_AUTO_TEST_CASE_WITH_THF(UnidirectionalSequenceLstmLayerInt8TimeMajor,
2570 UnidirectionalSequenceLstmLayerInt8TimeMajorTest)
2571ARMNN_AUTO_TEST_CASE_WITH_THF(UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjection,
2572 UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionTest)
2573ARMNN_AUTO_TEST_CASE_WITH_THF(UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionWithLayerNorm,
2574 UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionWithLayerNormTest)
2575ARMNN_AUTO_TEST_CASE_WITH_THF(UnidirectionalSequenceLstmInt8WithCifgWithPeepholeNoProjection,
2576 UnidirectionalSequenceLstmInt8WithCifgWithPeepholeNoProjectionTest)
Narumol Prangnawarate5339e72021-07-28 17:33:28 +0100[diff] [blame]2577
Sadik Armagan1625efc2021-06-10 18:24:34 +0100[diff] [blame]2578}