Gitiles
Code Review Sign In
review.mlplatform.org / ml / ComputeLibrary / 670c04ac4e99ee4b9eeb231e9d980c927bddfaa0 / . / tests / benchmark / NEON / ConvolutionLayer.cpp
blob: 2ae893bdec5083bb9506aa01557277071761ce7d [file] [log] [blame]
Moritz Pflanzeree493ae2017-07-05 10:52:21 +0100[diff] [blame]1/*
giuros01154bc1c2019-03-26 17:44:40 +0000[diff] [blame]2 * Copyright (c) 2017-2019 ARM Limited.
Moritz Pflanzeree493ae2017-07-05 10:52:21 +0100[diff] [blame]3 *
4 * SPDX-License-Identifier: MIT
5 *
6 * Permission is hereby granted, free of charge, to any person obtaining a copy
7 * of this software and associated documentation files (the "Software"), to
8 * deal in the Software without restriction, including without limitation the
9 * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
10 * sell copies of the Software, and to permit persons to whom the Software is
11 * furnished to do so, subject to the following conditions:
12 *
13 * The above copyright notice and this permission notice shall be included in all
14 * copies or substantial portions of the Software.
15 *
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22 * SOFTWARE.
23 */
24#include "arm_compute/core/TensorShape.h"
25#include "arm_compute/core/Types.h"
26#include "arm_compute/runtime/NEON/functions/NEConvolutionLayer.h"
giuros01154bc1c2019-03-26 17:44:40 +0000[diff] [blame]27#include "arm_compute/runtime/NEON/functions/NEFFTConvolutionLayer.h"
Georgios Pinitas9fb11592018-04-26 20:34:58 +0100[diff] [blame]28#include "arm_compute/runtime/NEON/functions/NEWinogradConvolutionLayer.h"
Moritz Pflanzeree493ae2017-07-05 10:52:21 +0100[diff] [blame]29#include "arm_compute/runtime/Tensor.h"
30#include "arm_compute/runtime/TensorAllocator.h"
Moritz Pflanzerd58cec02017-07-18 15:44:21 +0100[diff] [blame]31#include "tests/NEON/Accessor.h"
Moritz Pflanzera09de0c2017-09-01 20:41:12 +0100[diff] [blame]32#include "tests/benchmark/fixtures/ConvolutionLayerFixture.h"
giuros01154bc1c2019-03-26 17:44:40 +0000[diff] [blame]33#include "tests/benchmark/fixtures/FFTConvolutionLayerFixture.h"
Georgios Pinitas9fb11592018-04-26 20:34:58 +0100[diff] [blame]34#include "tests/benchmark/fixtures/WinogradConvolutionLayerFixture.h"
giuros01154bc1c2019-03-26 17:44:40 +0000[diff] [blame]35#include "tests/datasets/SmallConvolutionLayerDataset.h"
Moritz Pflanzera09de0c2017-09-01 20:41:12 +0100[diff] [blame]36#include "tests/datasets/system_tests/alexnet/AlexNetConvolutionLayerDataset.h"
37#include "tests/datasets/system_tests/googlenet/inceptionv1/GoogLeNetInceptionV1ConvolutionLayerDataset.h"
38#include "tests/datasets/system_tests/googlenet/inceptionv4/GoogLeNetInceptionV4ConvolutionLayerDataset.h"
39#include "tests/datasets/system_tests/lenet5/LeNet5ConvolutionLayerDataset.h"
Gian Marco Iodice2bc74412018-03-01 16:32:10 +0000[diff] [blame]40#include "tests/datasets/system_tests/mobilenet/MobileNetConvolutionLayerDataset.h"
giuros01154bc1c2019-03-26 17:44:40 +0000[diff] [blame]41#include "tests/datasets/system_tests/resnet12/ResNet12ConvolutionLayerDataset.h"
Moritz Pflanzera09de0c2017-09-01 20:41:12 +0100[diff] [blame]42#include "tests/datasets/system_tests/squeezenet/SqueezeNetConvolutionLayerDataset.h"
43#include "tests/datasets/system_tests/vgg/vgg16/VGG16ConvolutionLayerDataset.h"
44#include "tests/datasets/system_tests/yolo/v2/YOLOV2ConvolutionLayerDataset.h"
45#include "tests/framework/Macros.h"
46#include "tests/framework/datasets/Datasets.h"
Anthony Barbier2a07e182017-08-04 18:20:27 +0100[diff] [blame]47#include "utils/TypePrinter.h"
Moritz Pflanzeree493ae2017-07-05 10:52:21 +0100[diff] [blame]48
49namespace arm_compute
50{
51namespace test
52{
Michalis Spyrou724079d2017-12-15 11:37:37 +0000[diff] [blame]53namespace benchmark
54{
Pablo Tello40b29272017-07-19 16:56:05 +0100[diff] [blame]55namespace
56{
Ioan-Cristian Szabo5edbd1c2017-11-13 13:34:08 +0000[diff] [blame]57#ifdef __ARM_FEATURE_FP16_VECTOR_ARITHMETIC
Isabella Gottardie6630e42018-01-18 15:50:39 +0000[diff] [blame]58const auto data_types = framework::dataset::make("DataType", { DataType::F16, DataType::F32, DataType::QASYMM8 });
59#else /* __ARM_FEATURE_FP16_VECTOR_ARITHMETIC */
60const auto data_types = framework::dataset::make("DataType", { DataType::F32, DataType::QASYMM8 });
61
Ioan-Cristian Szabo5edbd1c2017-11-13 13:34:08 +0000[diff] [blame]62#endif /* __ARM_FEATURE_FP16_VECTOR_ARITHMETIC */
Pablo Tello40b29272017-07-19 16:56:05 +0100[diff] [blame]63} // namespace
64
Isabella Gottardi6acc6ad2018-02-02 17:19:18 +0000[diff] [blame]65using NEGEMMConvolutionLayerFixture = ConvolutionLayerFixture<Tensor, NEGEMMConvolutionLayer, Accessor>;
giuros01154bc1c2019-03-26 17:44:40 +0000[diff] [blame]66using NEFFTConvolutionLayerFixture = FFTConvolutionLayerFixture<Tensor, NEFFTConvolutionLayer, Accessor>;
Moritz Pflanzeree493ae2017-07-05 10:52:21 +0100[diff] [blame]67
68TEST_SUITE(NEON)
Pablo Tello8108d2e2017-12-01 09:58:32 +0000[diff] [blame]69#if defined(__aarch64__)
Georgios Pinitas9fb11592018-04-26 20:34:58 +0100[diff] [blame]70using NEWinogradConvolutionLayerFixture = WinogradConvolutionLayerFixture<Tensor, NEWinogradConvolutionLayer, Accessor>;
Pablo Tello8108d2e2017-12-01 09:58:32 +0000[diff] [blame]71
Georgios Pinitas9fb11592018-04-26 20:34:58 +0100[diff] [blame]72REGISTER_FIXTURE_DATA_TEST_CASE(AlexNetWinogradLayer, NEWinogradConvolutionLayerFixture, framework::DatasetMode::ALL,
Isabella Gottardi3f217ec2018-02-12 14:59:19 +0000[diff] [blame]73 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::AlexNetWinogradLayerDataset(),
74 framework::dataset::make("ActivationInfo", ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU))),
75 framework::dataset::make("DataType", DataType::F32)),
Pablo Tello8108d2e2017-12-01 09:58:32 +0000[diff] [blame]76 framework::dataset::make("Batches", 1)));
77
Georgios Pinitas9fb11592018-04-26 20:34:58 +0100[diff] [blame]78REGISTER_FIXTURE_DATA_TEST_CASE(GoogLeNetInceptionV1WinogradLayer, NEWinogradConvolutionLayerFixture, framework::DatasetMode::ALL,
Isabella Gottardi3f217ec2018-02-12 14:59:19 +0000[diff] [blame]79 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::GoogLeNetInceptionV1WinogradLayerDataset(),
80 framework::dataset::make("ActivationInfo", ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU))),
81 framework::dataset::make("DataType", DataType::F32)),
Pablo Tello8108d2e2017-12-01 09:58:32 +0000[diff] [blame]82 framework::dataset::make("Batches", 1)));
83
Georgios Pinitas9fb11592018-04-26 20:34:58 +0100[diff] [blame]84REGISTER_FIXTURE_DATA_TEST_CASE(GoogLeNetInceptionV4WinogradLayer, NEWinogradConvolutionLayerFixture, framework::DatasetMode::ALL,
Isabella Gottardi3f217ec2018-02-12 14:59:19 +0000[diff] [blame]85 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::GoogLeNetInceptionV4WinogradLayerDataset(),
86 framework::dataset::make("ActivationInfo", ActivationLayerInfo())),
87 framework::dataset::make("DataType", DataType::F32)),
Pablo Tello8108d2e2017-12-01 09:58:32 +0000[diff] [blame]88 framework::dataset::make("Batches", 1)));
89
Georgios Pinitas9fb11592018-04-26 20:34:58 +0100[diff] [blame]90REGISTER_FIXTURE_DATA_TEST_CASE(SqueezeNetWinogradLayer, NEWinogradConvolutionLayerFixture, framework::DatasetMode::ALL,
Isabella Gottardi3f217ec2018-02-12 14:59:19 +0000[diff] [blame]91 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::SqueezeNetWinogradLayerDataset(),
92 framework::dataset::make("ActivationInfo", ActivationLayerInfo())),
93 framework::dataset::make("DataType", DataType::F32)),
Pablo Tello8108d2e2017-12-01 09:58:32 +0000[diff] [blame]94 framework::dataset::make("Batches", 1)));
95#endif /* __aarch64__ */
Moritz Pflanzeree493ae2017-07-05 10:52:21 +0100[diff] [blame]96
giuros01154bc1c2019-03-26 17:44:40 +0000[diff] [blame]97REGISTER_FIXTURE_DATA_TEST_CASE(ResNet12FFTLayer, NEFFTConvolutionLayerFixture, framework::DatasetMode::ALL,
98 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::ResNet12FFTConvolutionLayerDataset(),
99 framework::dataset::make("ActivationInfo", ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU))),
100 framework::dataset::make("DataType", { DataType::F32 })),
101 framework::dataset::make("Batches", 1)));
102
Isabella Gottardi6acc6ad2018-02-02 17:19:18 +0000[diff] [blame]103REGISTER_FIXTURE_DATA_TEST_CASE(AlexNetConvolutionLayer, NEGEMMConvolutionLayerFixture, framework::DatasetMode::ALL,
Isabella Gottardi3f217ec2018-02-12 14:59:19 +0000[diff] [blame]104 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::AlexNetConvolutionLayerDataset(),
105 framework::dataset::make("ActivationInfo", ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU))),
106 data_types),
SiCong Licc65bbe2017-08-18 11:58:21 +0100[diff] [blame]107 framework::dataset::make("Batches", 1)));
Moritz Pflanzeree493ae2017-07-05 10:52:21 +0100[diff] [blame]108
Isabella Gottardi6acc6ad2018-02-02 17:19:18 +0000[diff] [blame]109REGISTER_FIXTURE_DATA_TEST_CASE(LeNet5ConvolutionLayer, NEGEMMConvolutionLayerFixture, framework::DatasetMode::ALL,
Isabella Gottardi3f217ec2018-02-12 14:59:19 +0000[diff] [blame]110 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::LeNet5ConvolutionLayerDataset(),
111 framework::dataset::make("ActivationInfo", ActivationLayerInfo())),
112 data_types),
SiCong Licc65bbe2017-08-18 11:58:21 +0100[diff] [blame]113 framework::dataset::make("Batches", 1)));
Moritz Pflanzeree493ae2017-07-05 10:52:21 +0100[diff] [blame]114
Isabella Gottardi6acc6ad2018-02-02 17:19:18 +0000[diff] [blame]115REGISTER_FIXTURE_DATA_TEST_CASE(GoogLeNetInceptionV1ConvolutionLayer, NEGEMMConvolutionLayerFixture, framework::DatasetMode::ALL,
Isabella Gottardi3f217ec2018-02-12 14:59:19 +0000[diff] [blame]116 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::GoogLeNetInceptionV1ConvolutionLayerDataset(),
117 framework::dataset::make("ActivationInfo", ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU))),
118 data_types),
SiCong Licc65bbe2017-08-18 11:58:21 +0100[diff] [blame]119 framework::dataset::make("Batches", 1)));
Moritz Pflanzeree493ae2017-07-05 10:52:21 +0100[diff] [blame]120
Isabella Gottardi6acc6ad2018-02-02 17:19:18 +0000[diff] [blame]121REGISTER_FIXTURE_DATA_TEST_CASE(GoogLeNetInceptionV4ConvolutionLayer, NEGEMMConvolutionLayerFixture, framework::DatasetMode::ALL,
Isabella Gottardi3f217ec2018-02-12 14:59:19 +0000[diff] [blame]122 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::GoogLeNetInceptionV4ConvolutionLayerDataset(),
123 framework::dataset::make("ActivationInfo", ActivationLayerInfo())),
124 data_types),
SiCong Licc65bbe2017-08-18 11:58:21 +0100[diff] [blame]125 framework::dataset::make("Batches", 1)));
SiCong Lif583fb72017-08-15 15:09:18 +0100[diff] [blame]126
Isabella Gottardi6acc6ad2018-02-02 17:19:18 +0000[diff] [blame]127REGISTER_FIXTURE_DATA_TEST_CASE(SqueezeNetConvolutionLayer, NEGEMMConvolutionLayerFixture, framework::DatasetMode::ALL,
Isabella Gottardi3f217ec2018-02-12 14:59:19 +0000[diff] [blame]128 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::SqueezeNetConvolutionLayerDataset(),
129 framework::dataset::make("ActivationInfo", ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU))),
130 data_types),
SiCong Licc65bbe2017-08-18 11:58:21 +0100[diff] [blame]131 framework::dataset::make("Batches", 1)));
SiCong Li0aea8902017-07-21 18:24:36 +0100[diff] [blame]132
Gian Marco Iodice2bc74412018-03-01 16:32:10 +0000[diff] [blame]133REGISTER_FIXTURE_DATA_TEST_CASE(MobileNetConvolutionLayer, NEGEMMConvolutionLayerFixture, framework::DatasetMode::ALL,
Isabella Gottardi3f217ec2018-02-12 14:59:19 +0000[diff] [blame]134 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::MobileNetConvolutionLayerDataset(),
135 framework::dataset::make("ActivationInfo", ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU))),
136 data_types),
Gian Marco Iodice2bc74412018-03-01 16:32:10 +0000[diff] [blame]137 framework::dataset::make("Batches", 1)));
138
SiCong Li1e5c1572017-07-27 17:58:52 +0100[diff] [blame]139TEST_SUITE(NIGHTLY)
Isabella Gottardi6acc6ad2018-02-02 17:19:18 +0000[diff] [blame]140REGISTER_FIXTURE_DATA_TEST_CASE(AlexNetConvolutionLayer, NEGEMMConvolutionLayerFixture, framework::DatasetMode::NIGHTLY,
Isabella Gottardi3f217ec2018-02-12 14:59:19 +0000[diff] [blame]141 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::AlexNetConvolutionLayerDataset(),
142 framework::dataset::make("ActivationInfo", ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU))),
143 data_types),
SiCong Licc65bbe2017-08-18 11:58:21 +0100[diff] [blame]144 framework::dataset::make("Batches", { 4, 8 })));
SiCong Li1e5c1572017-07-27 17:58:52 +0100[diff] [blame]145
Isabella Gottardi6acc6ad2018-02-02 17:19:18 +0000[diff] [blame]146REGISTER_FIXTURE_DATA_TEST_CASE(LeNet5ConvolutionLayer, NEGEMMConvolutionLayerFixture, framework::DatasetMode::NIGHTLY,
Isabella Gottardi3f217ec2018-02-12 14:59:19 +0000[diff] [blame]147 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::LeNet5ConvolutionLayerDataset(),
148 framework::dataset::make("ActivationInfo", ActivationLayerInfo())),
149 data_types),
SiCong Licc65bbe2017-08-18 11:58:21 +0100[diff] [blame]150 framework::dataset::make("Batches", { 4, 8 })));
SiCong Li1e5c1572017-07-27 17:58:52 +0100[diff] [blame]151
Isabella Gottardi6acc6ad2018-02-02 17:19:18 +0000[diff] [blame]152REGISTER_FIXTURE_DATA_TEST_CASE(GoogLeNetInceptionV1ConvolutionLayer, NEGEMMConvolutionLayerFixture, framework::DatasetMode::NIGHTLY,
Isabella Gottardi3f217ec2018-02-12 14:59:19 +0000[diff] [blame]153 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::GoogLeNetInceptionV1ConvolutionLayerDataset(),
154 framework::dataset::make("ActivationInfo", ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU))),
155 data_types),
SiCong Licc65bbe2017-08-18 11:58:21 +0100[diff] [blame]156 framework::dataset::make("Batches", { 4, 8 })));
SiCong Li1e5c1572017-07-27 17:58:52 +0100[diff] [blame]157
Isabella Gottardi6acc6ad2018-02-02 17:19:18 +0000[diff] [blame]158REGISTER_FIXTURE_DATA_TEST_CASE(GoogLeNetInceptionV4ConvolutionLayer, NEGEMMConvolutionLayerFixture, framework::DatasetMode::NIGHTLY,
Isabella Gottardi3f217ec2018-02-12 14:59:19 +0000[diff] [blame]159 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::GoogLeNetInceptionV4ConvolutionLayerDataset(),
160 framework::dataset::make("ActivationInfo", ActivationLayerInfo())),
161 data_types),
SiCong Licc65bbe2017-08-18 11:58:21 +0100[diff] [blame]162 framework::dataset::make("Batches", { 4, 8 })));
SiCong Lif583fb72017-08-15 15:09:18 +0100[diff] [blame]163
Isabella Gottardi6acc6ad2018-02-02 17:19:18 +0000[diff] [blame]164REGISTER_FIXTURE_DATA_TEST_CASE(SqueezeNetConvolutionLayer, NEGEMMConvolutionLayerFixture, framework::DatasetMode::NIGHTLY,
Isabella Gottardi3f217ec2018-02-12 14:59:19 +0000[diff] [blame]165 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::SqueezeNetConvolutionLayerDataset(),
166 framework::dataset::make("ActivationInfo", ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU))),
167 data_types),
SiCong Licc65bbe2017-08-18 11:58:21 +0100[diff] [blame]168 framework::dataset::make("Batches", { 4, 8 })));
169
Moritz Pflanzer70e9bc22017-09-08 16:07:10 +0100[diff] [blame]170// 8 batches use about 2GB of memory which is too much for most devices!
Isabella Gottardi6acc6ad2018-02-02 17:19:18 +0000[diff] [blame]171REGISTER_FIXTURE_DATA_TEST_CASE(VGG16ConvolutionLayer, NEGEMMConvolutionLayerFixture, framework::DatasetMode::NIGHTLY,
Isabella Gottardi3f217ec2018-02-12 14:59:19 +0000[diff] [blame]172 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::VGG16ConvolutionLayerDataset(),
173 framework::dataset::make("ActivationInfo", ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU))),
174 data_types),
Gian Marco33da05b2017-11-08 14:13:33 +0000[diff] [blame]175 framework::dataset::make("Batches", { 1, 2 })));
SiCong Li1e5c1572017-07-27 17:58:52 +0100[diff] [blame]176
Isabella Gottardi6acc6ad2018-02-02 17:19:18 +0000[diff] [blame]177REGISTER_FIXTURE_DATA_TEST_CASE(YOLOV2ConvolutionLayer, NEGEMMConvolutionLayerFixture, framework::DatasetMode::NIGHTLY,
Isabella Gottardi3f217ec2018-02-12 14:59:19 +0000[diff] [blame]178 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::YOLOV2ConvolutionLayerDataset(),
179 framework::dataset::make("ActivationInfo", ActivationLayerInfo())),
180 data_types),
SiCong Li1e5c1572017-07-27 17:58:52 +0100[diff] [blame]181 framework::dataset::make("Batches", { 1, 4, 8 })));
SiCong Licc65bbe2017-08-18 11:58:21 +0100[diff] [blame]182
Pablo Tello8108d2e2017-12-01 09:58:32 +0000[diff] [blame]183#if defined(__aarch64__)
Georgios Pinitas9fb11592018-04-26 20:34:58 +0100[diff] [blame]184REGISTER_FIXTURE_DATA_TEST_CASE(AlexNetWinogradLayer, NEWinogradConvolutionLayerFixture, framework::DatasetMode::NIGHTLY,
Isabella Gottardi3f217ec2018-02-12 14:59:19 +0000[diff] [blame]185 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::AlexNetWinogradLayerDataset(),
186 framework::dataset::make("ActivationInfo", ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU))),
187 framework::dataset::make("DataType", DataType::F32)),
Pablo Tello8108d2e2017-12-01 09:58:32 +0000[diff] [blame]188 framework::dataset::make("Batches", { 4, 8 })));
189
Georgios Pinitas9fb11592018-04-26 20:34:58 +0100[diff] [blame]190REGISTER_FIXTURE_DATA_TEST_CASE(GoogLeNetInceptionV1WinogradLayer, NEWinogradConvolutionLayerFixture, framework::DatasetMode::NIGHTLY,
Isabella Gottardi3f217ec2018-02-12 14:59:19 +0000[diff] [blame]191 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::GoogLeNetInceptionV1WinogradLayerDataset(),
192 framework::dataset::make("ActivationInfo", ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU))),
193 framework::dataset::make("DataType", DataType::F32)),
Pablo Tello8108d2e2017-12-01 09:58:32 +0000[diff] [blame]194 framework::dataset::make("Batches", { 4, 8 })));
195
Georgios Pinitas9fb11592018-04-26 20:34:58 +0100[diff] [blame]196REGISTER_FIXTURE_DATA_TEST_CASE(GoogLeNetInceptionV4WinogradLayer, NEWinogradConvolutionLayerFixture, framework::DatasetMode::NIGHTLY,
Isabella Gottardi3f217ec2018-02-12 14:59:19 +0000[diff] [blame]197 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::GoogLeNetInceptionV4WinogradLayerDataset(),
198 framework::dataset::make("ActivationInfo", ActivationLayerInfo())),
199 framework::dataset::make("DataType", DataType::F32)),
Pablo Tello8108d2e2017-12-01 09:58:32 +0000[diff] [blame]200 framework::dataset::make("Batches", { 4, 8 })));
201
Georgios Pinitas9fb11592018-04-26 20:34:58 +0100[diff] [blame]202REGISTER_FIXTURE_DATA_TEST_CASE(SqueezeNetWinogradLayer, NEWinogradConvolutionLayerFixture, framework::DatasetMode::NIGHTLY,
Isabella Gottardi3f217ec2018-02-12 14:59:19 +0000[diff] [blame]203 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::SqueezeNetWinogradLayerDataset(),
204 framework::dataset::make("ActivationInfo", ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU))),
205 framework::dataset::make("DataType", DataType::F32)),
Pablo Tello8108d2e2017-12-01 09:58:32 +0000[diff] [blame]206 framework::dataset::make("Batches", { 4, 8 })));
207#endif /* __aarch64__ */
208
SiCong Li1e5c1572017-07-27 17:58:52 +0100[diff] [blame]209TEST_SUITE_END()
Moritz Pflanzeree493ae2017-07-05 10:52:21 +0100[diff] [blame]210TEST_SUITE_END()
Michalis Spyrou724079d2017-12-15 11:37:37 +0000[diff] [blame]211} // namespace benchmark
Moritz Pflanzeree493ae2017-07-05 10:52:21 +0100[diff] [blame]212} // namespace test
213} // namespace arm_compute