xref: /aosp_15_r20/external/ComputeLibrary/tests/validation/NEON/SpaceToDepthLayer.cpp (revision c217d954acce2dbc11938adb493fc0abd69584f3)
1*c217d954SCole Faust /*
2*c217d954SCole Faust  * Copyright (c) 2019-2021 Arm Limited.
3*c217d954SCole Faust  *
4*c217d954SCole Faust  * SPDX-License-Identifier: MIT
5*c217d954SCole Faust  *
6*c217d954SCole Faust  * Permission is hereby granted, free of charge, to any person obtaining a copy
7*c217d954SCole Faust  * of this software and associated documentation files (the "Software"), to
8*c217d954SCole Faust  * deal in the Software without restriction, including without limitation the
9*c217d954SCole Faust  * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
10*c217d954SCole Faust  * sell copies of the Software, and to permit persons to whom the Software is
11*c217d954SCole Faust  * furnished to do so, subject to the following conditions:
12*c217d954SCole Faust  *
13*c217d954SCole Faust  * The above copyright notice and this permission notice shall be included in all
14*c217d954SCole Faust  * copies or substantial portions of the Software.
15*c217d954SCole Faust  *
16*c217d954SCole Faust  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17*c217d954SCole Faust  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18*c217d954SCole Faust  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19*c217d954SCole Faust  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20*c217d954SCole Faust  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21*c217d954SCole Faust  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22*c217d954SCole Faust  * SOFTWARE.
23*c217d954SCole Faust  */
24*c217d954SCole Faust #include "arm_compute/core/Types.h"
25*c217d954SCole Faust #include "arm_compute/runtime/NEON/functions/NESpaceToDepthLayer.h"
26*c217d954SCole Faust #include "arm_compute/runtime/Tensor.h"
27*c217d954SCole Faust #include "arm_compute/runtime/TensorAllocator.h"
28*c217d954SCole Faust #include "tests/NEON/Accessor.h"
29*c217d954SCole Faust #include "tests/PaddingCalculator.h"
30*c217d954SCole Faust #include "tests/datasets/ShapeDatasets.h"
31*c217d954SCole Faust #include "tests/datasets/SpaceToDepthDataset.h"
32*c217d954SCole Faust #include "tests/framework/Asserts.h"
33*c217d954SCole Faust #include "tests/framework/Macros.h"
34*c217d954SCole Faust #include "tests/framework/datasets/Datasets.h"
35*c217d954SCole Faust #include "tests/validation/Validation.h"
36*c217d954SCole Faust #include "tests/validation/fixtures/SpaceToDepthFixture.h"
37*c217d954SCole Faust 
38*c217d954SCole Faust namespace arm_compute
39*c217d954SCole Faust {
40*c217d954SCole Faust namespace test
41*c217d954SCole Faust {
42*c217d954SCole Faust namespace validation
43*c217d954SCole Faust {
44*c217d954SCole Faust TEST_SUITE(NEON)
45*c217d954SCole Faust TEST_SUITE(SpaceToDepthLayer)
46*c217d954SCole Faust 
47*c217d954SCole Faust template <typename T>
48*c217d954SCole Faust using NESpaceToDepthLayerFixture = SpaceToDepthLayerValidationFixture<Tensor, Accessor, NESpaceToDepthLayer, T>;
49*c217d954SCole Faust 
50*c217d954SCole Faust // *INDENT-OFF*
51*c217d954SCole Faust // clang-format off
52*c217d954SCole Faust DATA_TEST_CASE(ValidateStatic, framework::DatasetMode::ALL, zip(zip(zip(
53*c217d954SCole Faust                framework::dataset::make("InputInfo", { TensorInfo(TensorShape(32U, 16U, 2U, 1U), 1, DataType::F32),
54*c217d954SCole Faust                                                        TensorInfo(TensorShape(32U, 16U, 2U, 1U), 1, DataType::F32),    // Mismatching data types
55*c217d954SCole Faust                                                        TensorInfo(TensorShape(32U, 16U, 2U, 1U), 1, DataType::F32),    // Negative block shapes
56*c217d954SCole Faust                                                        TensorInfo(TensorShape(32U, 16U, 2U, 1U, 4U), 1, DataType::F32), // Wrong tensor shape
57*c217d954SCole Faust                                                      }),
58*c217d954SCole Faust                framework::dataset::make("OutputInfo",{ TensorInfo(TensorShape(16U, 8U, 8U, 1U), 1, DataType::F32),
59*c217d954SCole Faust                                                        TensorInfo(TensorShape(32U, 8U, 8U, 1U), 1, DataType::F16),
60*c217d954SCole Faust                                                        TensorInfo(TensorShape(32U, 8U, 8U, 1U), 1, DataType::F32),
61*c217d954SCole Faust                                                        TensorInfo(TensorShape(32U, 8U, 8U, 1U), 1, DataType::F32),
62*c217d954SCole Faust                                                      })),
63*c217d954SCole Faust                framework::dataset::make("BlockShape", { 2, 2, -2, 2 })),
64*c217d954SCole Faust                framework::dataset::make("Expected", { true, false, false, false})),
65*c217d954SCole Faust                input_info, output_info, block_shape, expected)
66*c217d954SCole Faust {
67*c217d954SCole Faust     bool has_error = bool(NESpaceToDepthLayer::validate(&input_info.clone()->set_is_resizable(false), &output_info.clone()->set_is_resizable(false), block_shape));
68*c217d954SCole Faust     ARM_COMPUTE_EXPECT(has_error == expected, framework::LogLevel::ERRORS);
69*c217d954SCole Faust }
70*c217d954SCole Faust // clang-format on
71*c217d954SCole Faust // *INDENT-ON*
72*c217d954SCole Faust 
73*c217d954SCole Faust TEST_SUITE(Float)
TEST_SUITE(FP32)74*c217d954SCole Faust TEST_SUITE(FP32)
75*c217d954SCole Faust FIXTURE_DATA_TEST_CASE(Small, NESpaceToDepthLayerFixture<float>, framework::DatasetMode::PRECOMMIT, combine(combine(datasets::SmallSpaceToDepthLayerDataset(), framework::dataset::make("DataType",
76*c217d954SCole Faust                                                                                                                     DataType::F32)),
77*c217d954SCole Faust                                                                                                             framework::dataset::make("DataLayout", { DataLayout::NCHW, DataLayout::NHWC })))
78*c217d954SCole Faust {
79*c217d954SCole Faust     // Validate output
80*c217d954SCole Faust     validate(Accessor(_target), _reference);
81*c217d954SCole Faust }
82*c217d954SCole Faust FIXTURE_DATA_TEST_CASE(Large, NESpaceToDepthLayerFixture<float>, framework::DatasetMode::NIGHTLY, combine(combine(datasets::LargeSpaceToDepthLayerDataset(), framework::dataset::make("DataType",
83*c217d954SCole Faust                                                                                                                   DataType::F32)),
84*c217d954SCole Faust                                                                                                           framework::dataset::make("DataLayout", { DataLayout::NCHW, DataLayout::NHWC })))
85*c217d954SCole Faust {
86*c217d954SCole Faust     // Validate output
87*c217d954SCole Faust     validate(Accessor(_target), _reference);
88*c217d954SCole Faust }
89*c217d954SCole Faust TEST_SUITE_END() // FP32
90*c217d954SCole Faust 
TEST_SUITE(FP16)91*c217d954SCole Faust TEST_SUITE(FP16)
92*c217d954SCole Faust FIXTURE_DATA_TEST_CASE(Small, NESpaceToDepthLayerFixture<half>, framework::DatasetMode::PRECOMMIT, combine(combine(datasets::SmallSpaceToDepthLayerDataset(), framework::dataset::make("DataType",
93*c217d954SCole Faust                                                                                                                    DataType::F16)),
94*c217d954SCole Faust                                                                                                            framework::dataset::make("DataLayout", { DataLayout::NCHW, DataLayout::NHWC })))
95*c217d954SCole Faust {
96*c217d954SCole Faust     // Validate output
97*c217d954SCole Faust     validate(Accessor(_target), _reference);
98*c217d954SCole Faust }
99*c217d954SCole Faust FIXTURE_DATA_TEST_CASE(Large, NESpaceToDepthLayerFixture<half>, framework::DatasetMode::NIGHTLY, combine(combine(datasets::LargeSpaceToDepthLayerDataset(), framework::dataset::make("DataType",
100*c217d954SCole Faust                                                                                                                  DataType::F16)),
101*c217d954SCole Faust                                                                                                          framework::dataset::make("DataLayout", { DataLayout::NCHW, DataLayout::NHWC })))
102*c217d954SCole Faust {
103*c217d954SCole Faust     // Validate output
104*c217d954SCole Faust     validate(Accessor(_target), _reference);
105*c217d954SCole Faust }
106*c217d954SCole Faust TEST_SUITE_END() // FP16
107*c217d954SCole Faust TEST_SUITE_END() // Float
108*c217d954SCole Faust 
109*c217d954SCole Faust TEST_SUITE_END() // SpaceToDepthLayer
110*c217d954SCole Faust TEST_SUITE_END() // Neon
111*c217d954SCole Faust } // namespace validation
112*c217d954SCole Faust } // namespace test
113*c217d954SCole Faust } // namespace arm_compute
114