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