Gitiles
Code Review Sign In
review.mlplatform.org / ml / ComputeLibrary / 0ef7e670a22248806458d7327db9e8b8c4db4ce6 / . / tests / benchmark / GLES_COMPUTE / PoolingLayer.cpp
blob: 2a8e5dacb3fb63910f68f49b670c9fbde7073d5a [file] [log] [blame]
Anthony Barbier7068f992017-10-26 15:23:08 +0100[diff] [blame]1/*
Michalis Spyrou724079d2017-12-15 11:37:37 +0000[diff] [blame]2 * Copyright (c) 2017-2018 ARM Limited.
Anthony Barbier7068f992017-10-26 15:23:08 +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/GLES_COMPUTE/GCTensor.h"
27#include "arm_compute/runtime/GLES_COMPUTE/GCTensorAllocator.h"
28#include "arm_compute/runtime/GLES_COMPUTE/functions/GCPoolingLayer.h"
29#include "tests/GLES_COMPUTE/GCAccessor.h"
30#include "tests/benchmark/fixtures/PoolingLayerFixture.h"
31#include "tests/datasets/system_tests/alexnet/AlexNetPoolingLayerDataset.h"
32#include "tests/datasets/system_tests/googlenet/inceptionv1/GoogLeNetInceptionV1PoolingLayerDataset.h"
33#include "tests/datasets/system_tests/googlenet/inceptionv4/GoogLeNetInceptionV4PoolingLayerDataset.h"
34#include "tests/datasets/system_tests/lenet5/LeNet5PoolingLayerDataset.h"
35#include "tests/datasets/system_tests/squeezenet/SqueezeNetPoolingLayerDataset.h"
36#include "tests/datasets/system_tests/vgg/vgg16/VGG16PoolingLayerDataset.h"
37#include "tests/datasets/system_tests/yolo/v2/YOLOV2PoolingLayerDataset.h"
38#include "tests/framework/Macros.h"
39#include "tests/framework/datasets/Datasets.h"
40#include "utils/TypePrinter.h"
41
42namespace arm_compute
43{
44namespace test
45{
Michalis Spyrou724079d2017-12-15 11:37:37 +0000[diff] [blame]46namespace benchmark
47{
Anthony Barbier7068f992017-10-26 15:23:08 +0100[diff] [blame]48namespace
49{
50const auto data_types = framework::dataset::make("DataType", { DataType::F32 });
51} // namespace
52
53using GCPoolingLayerFixture = PoolingLayerFixture<GCTensor, GCPoolingLayer, GCAccessor>;
54
55TEST_SUITE(GC)
56
57REGISTER_FIXTURE_DATA_TEST_CASE(AlexNetPoolingLayer, GCPoolingLayerFixture, framework::DatasetMode::ALL,
58 framework::dataset::combine(framework::dataset::combine(datasets::AlexNetPoolingLayerDataset(),
59 data_types),
60 framework::dataset::make("Batches", 1)));
61
62REGISTER_FIXTURE_DATA_TEST_CASE(LeNet5PoolingLayer, GCPoolingLayerFixture, framework::DatasetMode::ALL,
63 framework::dataset::combine(framework::dataset::combine(datasets::LeNet5PoolingLayerDataset(),
64 data_types),
65 framework::dataset::make("Batches", 1)));
66
67REGISTER_FIXTURE_DATA_TEST_CASE(GoogLeNetInceptionV1PoolingLayer, GCPoolingLayerFixture, framework::DatasetMode::ALL,
68 framework::dataset::combine(framework::dataset::combine(datasets::GoogLeNetInceptionV1PoolingLayerDataset(),
69 data_types),
70 framework::dataset::make("Batches", 1)));
71
72REGISTER_FIXTURE_DATA_TEST_CASE(GoogLeNetInceptionV4PoolingLayer, GCPoolingLayerFixture, framework::DatasetMode::ALL,
73 framework::dataset::combine(framework::dataset::combine(datasets::GoogLeNetInceptionV4PoolingLayerDataset(),
74 data_types),
75 framework::dataset::make("Batches", 1)));
76
77REGISTER_FIXTURE_DATA_TEST_CASE(SqueezeNetPoolingLayer, GCPoolingLayerFixture, framework::DatasetMode::ALL,
78 framework::dataset::combine(framework::dataset::combine(datasets::SqueezeNetPoolingLayerDataset(),
79 data_types),
80 framework::dataset::make("Batches", 1)));
81
82REGISTER_FIXTURE_DATA_TEST_CASE(VGG16PoolingLayer, GCPoolingLayerFixture, framework::DatasetMode::ALL,
83 framework::dataset::combine(framework::dataset::combine(datasets::VGG16PoolingLayerDataset(),
84 data_types),
85 framework::dataset::make("Batches", 1)));
86
87REGISTER_FIXTURE_DATA_TEST_CASE(YOLOV2PoolingLayer, GCPoolingLayerFixture, framework::DatasetMode::ALL,
88 framework::dataset::combine(framework::dataset::combine(datasets::YOLOV2PoolingLayerDataset(),
89 data_types),
90 framework::dataset::make("Batches", 1)));
91
92TEST_SUITE(NIGHTLY)
93REGISTER_FIXTURE_DATA_TEST_CASE(AlexNetPoolingLayer, GCPoolingLayerFixture, framework::DatasetMode::NIGHTLY,
94 framework::dataset::combine(framework::dataset::combine(datasets::AlexNetPoolingLayerDataset(),
95 data_types),
96 framework::dataset::make("Batches", { 4, 8 })));
97
98REGISTER_FIXTURE_DATA_TEST_CASE(LeNet5PoolingLayer, GCPoolingLayerFixture, framework::DatasetMode::NIGHTLY,
99 framework::dataset::combine(framework::dataset::combine(datasets::LeNet5PoolingLayerDataset(),
100 data_types),
101 framework::dataset::make("Batches", { 4, 8 })));
102
103REGISTER_FIXTURE_DATA_TEST_CASE(GoogLeNetInceptionV1PoolingLayer, GCPoolingLayerFixture, framework::DatasetMode::NIGHTLY,
104 framework::dataset::combine(framework::dataset::combine(datasets::GoogLeNetInceptionV1PoolingLayerDataset(),
105 data_types),
106 framework::dataset::make("Batches", { 4, 8 })));
107
108REGISTER_FIXTURE_DATA_TEST_CASE(GoogLeNetInceptionV4PoolingLayer, GCPoolingLayerFixture, framework::DatasetMode::NIGHTLY,
109 framework::dataset::combine(framework::dataset::combine(datasets::GoogLeNetInceptionV4PoolingLayerDataset(),
110 data_types),
111 framework::dataset::make("Batches", { 4, 8 })));
112
113REGISTER_FIXTURE_DATA_TEST_CASE(SqueezeNetPoolingLayer, GCPoolingLayerFixture, framework::DatasetMode::NIGHTLY,
114 framework::dataset::combine(framework::dataset::combine(datasets::SqueezeNetPoolingLayerDataset(),
115 data_types),
116 framework::dataset::make("Batches", { 4, 8 })));
117
118REGISTER_FIXTURE_DATA_TEST_CASE(VGG16PoolingLayer, GCPoolingLayerFixture, framework::DatasetMode::NIGHTLY,
119 framework::dataset::combine(framework::dataset::combine(datasets::VGG16PoolingLayerDataset(),
120 data_types),
121 framework::dataset::make("Batches", { 4, 8 })));
122
123REGISTER_FIXTURE_DATA_TEST_CASE(YOLOV2PoolingLayer, GCPoolingLayerFixture, framework::DatasetMode::NIGHTLY,
124 framework::dataset::combine(framework::dataset::combine(datasets::YOLOV2PoolingLayerDataset(),
125 data_types),
126 framework::dataset::make("Batches", { 4, 8 })));
127TEST_SUITE_END()
128TEST_SUITE_END()
Michalis Spyrou724079d2017-12-15 11:37:37 +0000[diff] [blame]129} // namespace benchmark
Anthony Barbier7068f992017-10-26 15:23:08 +0100[diff] [blame]130} // namespace test
131} // namespace arm_compute