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