Gitiles
Code Review Sign In
review.mlplatform.org / ml / ComputeLibrary / 670c04ac4e99ee4b9eeb231e9d980c927bddfaa0 / . / tests / benchmark / GLES_COMPUTE / PoolingLayer.cpp
blob: 80a33875e174668a7797c9c5ec6409980e7a23b9 [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{
Michalis Spyrou57dac842018-03-01 16:03:50 +0000[diff] [blame]50const auto data_types = framework::dataset::make("DataType", { DataType::F32 });
51const auto data_layouts = framework::dataset::make("DataLayout", { DataLayout::NCHW });
Anthony Barbier7068f992017-10-26 15:23:08 +0100[diff] [blame]52} // namespace
53
54using GCPoolingLayerFixture = PoolingLayerFixture<GCTensor, GCPoolingLayer, GCAccessor>;
55
56TEST_SUITE(GC)
57
58REGISTER_FIXTURE_DATA_TEST_CASE(AlexNetPoolingLayer, GCPoolingLayerFixture, framework::DatasetMode::ALL,
Michalis Spyrou57dac842018-03-01 16:03:50 +0000[diff] [blame]59 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::AlexNetPoolingLayerDataset(),
60 data_types),
61 data_layouts),
Anthony Barbier7068f992017-10-26 15:23:08 +0100[diff] [blame]62 framework::dataset::make("Batches", 1)));
63
64REGISTER_FIXTURE_DATA_TEST_CASE(LeNet5PoolingLayer, GCPoolingLayerFixture, framework::DatasetMode::ALL,
Michalis Spyrou57dac842018-03-01 16:03:50 +0000[diff] [blame]65 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::LeNet5PoolingLayerDataset(),
66 data_types),
67 data_layouts),
Anthony Barbier7068f992017-10-26 15:23:08 +0100[diff] [blame]68 framework::dataset::make("Batches", 1)));
69
70REGISTER_FIXTURE_DATA_TEST_CASE(GoogLeNetInceptionV1PoolingLayer, GCPoolingLayerFixture, framework::DatasetMode::ALL,
Michalis Spyrou57dac842018-03-01 16:03:50 +0000[diff] [blame]71 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::GoogLeNetInceptionV1PoolingLayerDataset(),
72 data_types),
73 data_layouts),
Anthony Barbier7068f992017-10-26 15:23:08 +0100[diff] [blame]74 framework::dataset::make("Batches", 1)));
75
76REGISTER_FIXTURE_DATA_TEST_CASE(GoogLeNetInceptionV4PoolingLayer, GCPoolingLayerFixture, framework::DatasetMode::ALL,
Michalis Spyrou57dac842018-03-01 16:03:50 +0000[diff] [blame]77 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::GoogLeNetInceptionV4PoolingLayerDataset(),
78 data_types),
79 data_layouts),
Anthony Barbier7068f992017-10-26 15:23:08 +0100[diff] [blame]80 framework::dataset::make("Batches", 1)));
81
82REGISTER_FIXTURE_DATA_TEST_CASE(SqueezeNetPoolingLayer, GCPoolingLayerFixture, framework::DatasetMode::ALL,
Michalis Spyrou57dac842018-03-01 16:03:50 +0000[diff] [blame]83 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::SqueezeNetPoolingLayerDataset(),
84 data_types),
85 data_layouts),
Anthony Barbier7068f992017-10-26 15:23:08 +0100[diff] [blame]86 framework::dataset::make("Batches", 1)));
87
88REGISTER_FIXTURE_DATA_TEST_CASE(VGG16PoolingLayer, GCPoolingLayerFixture, framework::DatasetMode::ALL,
Michalis Spyrou57dac842018-03-01 16:03:50 +0000[diff] [blame]89 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::VGG16PoolingLayerDataset(),
90 data_types),
91 data_layouts),
Anthony Barbier7068f992017-10-26 15:23:08 +0100[diff] [blame]92 framework::dataset::make("Batches", 1)));
93
94REGISTER_FIXTURE_DATA_TEST_CASE(YOLOV2PoolingLayer, GCPoolingLayerFixture, framework::DatasetMode::ALL,
Michalis Spyrou57dac842018-03-01 16:03:50 +0000[diff] [blame]95 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::YOLOV2PoolingLayerDataset(),
96 data_types),
97 data_layouts),
Anthony Barbier7068f992017-10-26 15:23:08 +0100[diff] [blame]98 framework::dataset::make("Batches", 1)));
99
100TEST_SUITE(NIGHTLY)
101REGISTER_FIXTURE_DATA_TEST_CASE(AlexNetPoolingLayer, GCPoolingLayerFixture, framework::DatasetMode::NIGHTLY,
Michalis Spyrou57dac842018-03-01 16:03:50 +0000[diff] [blame]102 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::AlexNetPoolingLayerDataset(),
103 data_types),
104 data_layouts),
Anthony Barbier7068f992017-10-26 15:23:08 +0100[diff] [blame]105 framework::dataset::make("Batches", { 4, 8 })));
106
107REGISTER_FIXTURE_DATA_TEST_CASE(LeNet5PoolingLayer, GCPoolingLayerFixture, framework::DatasetMode::NIGHTLY,
Michalis Spyrou57dac842018-03-01 16:03:50 +0000[diff] [blame]108 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::LeNet5PoolingLayerDataset(),
109 data_types),
110 data_layouts),
Anthony Barbier7068f992017-10-26 15:23:08 +0100[diff] [blame]111 framework::dataset::make("Batches", { 4, 8 })));
112
113REGISTER_FIXTURE_DATA_TEST_CASE(GoogLeNetInceptionV1PoolingLayer, GCPoolingLayerFixture, framework::DatasetMode::NIGHTLY,
Michalis Spyrou57dac842018-03-01 16:03:50 +0000[diff] [blame]114 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::GoogLeNetInceptionV1PoolingLayerDataset(),
115 data_types),
116 data_layouts),
Anthony Barbier7068f992017-10-26 15:23:08 +0100[diff] [blame]117 framework::dataset::make("Batches", { 4, 8 })));
118
119REGISTER_FIXTURE_DATA_TEST_CASE(GoogLeNetInceptionV4PoolingLayer, GCPoolingLayerFixture, framework::DatasetMode::NIGHTLY,
Michalis Spyrou57dac842018-03-01 16:03:50 +0000[diff] [blame]120 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::GoogLeNetInceptionV4PoolingLayerDataset(),
121 data_types),
122 data_layouts),
Anthony Barbier7068f992017-10-26 15:23:08 +0100[diff] [blame]123 framework::dataset::make("Batches", { 4, 8 })));
124
125REGISTER_FIXTURE_DATA_TEST_CASE(SqueezeNetPoolingLayer, GCPoolingLayerFixture, framework::DatasetMode::NIGHTLY,
Michalis Spyrou57dac842018-03-01 16:03:50 +0000[diff] [blame]126 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::SqueezeNetPoolingLayerDataset(),
127 data_types),
128 data_layouts),
Anthony Barbier7068f992017-10-26 15:23:08 +0100[diff] [blame]129 framework::dataset::make("Batches", { 4, 8 })));
130
131REGISTER_FIXTURE_DATA_TEST_CASE(VGG16PoolingLayer, GCPoolingLayerFixture, framework::DatasetMode::NIGHTLY,
Michalis Spyrou57dac842018-03-01 16:03:50 +0000[diff] [blame]132 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::VGG16PoolingLayerDataset(),
133 data_types),
134 data_layouts),
Anthony Barbier7068f992017-10-26 15:23:08 +0100[diff] [blame]135 framework::dataset::make("Batches", { 4, 8 })));
136
137REGISTER_FIXTURE_DATA_TEST_CASE(YOLOV2PoolingLayer, GCPoolingLayerFixture, framework::DatasetMode::NIGHTLY,
Michalis Spyrou57dac842018-03-01 16:03:50 +0000[diff] [blame]138 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::YOLOV2PoolingLayerDataset(),
139 data_types),
140 data_layouts),
Anthony Barbier7068f992017-10-26 15:23:08 +0100[diff] [blame]141 framework::dataset::make("Batches", { 4, 8 })));
142TEST_SUITE_END()
143TEST_SUITE_END()
Michalis Spyrou724079d2017-12-15 11:37:37 +0000[diff] [blame]144} // namespace benchmark
Anthony Barbier7068f992017-10-26 15:23:08 +0100[diff] [blame]145} // namespace test
146} // namespace arm_compute