2  * Copyright (c) 2018-2021 Arm Limited.
4  * SPDX-License-Identifier: MIT
13  * The above copyright notice and this permission notice shall be included in all
49 // *INDENT-OFF*
50 // clang-format off
52 const auto data_types = framework::dataset::make("DataType", { DataType::QASYMM8, DataType::F16, Da…
55 const auto n_values = framework::dataset::make("NumTensors", { 3, 4 });
58 const auto shapes_1d_small = combine(datasets::Small1DShapes(), framework::dataset::make("Axis", -1…
61 const auto shapes_2d_small = combine(datasets::Small2DShapes(), framework::dataset::make("Axis", -2…
64 const auto shapes_3d_small = combine(datasets::Small3DShapes(), framework::dataset::make("Axis", -3…
67 const auto shapes_4d_small = combine(datasets::Small4DShapes(), framework::dataset::make("Axis", -4…
70 const auto shapes_1d_large = combine(datasets::Large1DShapes(), framework::dataset::make("Axis", -1…
73 const auto shapes_2d_large = combine(datasets::Medium2DShapes(), framework::dataset::make("Axis", -…
76 const auto shapes_3d_large = combine(datasets::Medium3DShapes(), framework::dataset::make("Axis", -…
79 const auto shapes_4d_large = combine(datasets::Medium4DShapes(), framework::dataset::make("Axis", -…
91 DATA_TEST_CASE(Validate, framework::DatasetMode::ALL, zip(zip(zip(  in TEST_SUITE()
92 …                                                              framework::dataset::make("InputInfo",  in TEST_SUITE()
100 framework::dataset::make("OutputInfo",
104     TensorInfo(TensorShape(1U, 2U, 3U), 1, DataType::S32),  // fails axis <  (- input's rank)
108 framework::dataset::make("Axis", { -3, 1, -4, -3, 1 })),
109 framework::dataset::make("Expected", { true, true, false, false, false })),
123 FIXTURE_DATA_TEST_CASE(RunSmall, NEStackLayerFixture<int>, framework::DatasetMode::ALL,  in TEST_SUITE()
125 …                                          framework::dataset::make("DataType", { DataType::S32 })),  in TEST_SUITE()
134 …                                          framework::dataset::make("DataType", { DataType::S32 })),
143 FIXTURE_DATA_TEST_CASE(RunSmall, NEStackLayerFixture<short>, framework::DatasetMode::ALL,  in TEST_SUITE()
145 …                                          framework::dataset::make("DataType", { DataType::S16 })),  in TEST_SUITE()
154 …                                          framework::dataset::make("DataType", { DataType::S16 })),
163 FIXTURE_DATA_TEST_CASE(RunSmall, NEStackLayerFixture<char>, framework::DatasetMode::ALL,  in TEST_SUITE()
165 …                                           framework::dataset::make("DataType", { DataType::S8 })),  in TEST_SUITE()
174 …                                           framework::dataset::make("DataType", { DataType::S8 })),
185 FIXTURE_DATA_TEST_CASE(RunSmall, NEStackLayerFixture<int>, framework::DatasetMode::ALL,  in TEST_SUITE_END()
187 …                                          framework::dataset::make("DataType", { DataType::S32 })),  in TEST_SUITE_END()
196 …                                          framework::dataset::make("DataType", { DataType::S32 })),
205 FIXTURE_DATA_TEST_CASE(RunSmall, NEStackLayerFixture<short>, framework::DatasetMode::ALL,  in TEST_SUITE()
207 …                                          framework::dataset::make("DataType", { DataType::S16 })),  in TEST_SUITE()
216 …                                          framework::dataset::make("DataType", { DataType::S16 })),
225 FIXTURE_DATA_TEST_CASE(RunSmall, NEStackLayerFixture<char>, framework::DatasetMode::ALL,  in TEST_SUITE()
227 …                                           framework::dataset::make("DataType", { DataType::S8 })),  in TEST_SUITE()
236 …                                           framework::dataset::make("DataType", { DataType::S8 })),
247 FIXTURE_DATA_TEST_CASE(RunSmall, NEStackLayerFixture<int>, framework::DatasetMode::ALL,  in TEST_SUITE_END()
249 …                                          framework::dataset::make("DataType", { DataType::S32 })),  in TEST_SUITE_END()
258 …                                          framework::dataset::make("DataType", { DataType::S32 })),
267 FIXTURE_DATA_TEST_CASE(RunSmall, NEStackLayerFixture<short>, framework::DatasetMode::ALL,  in TEST_SUITE()
269 …                                          framework::dataset::make("DataType", { DataType::S16 })),  in TEST_SUITE()
278 …                                          framework::dataset::make("DataType", { DataType::S16 })),
287 FIXTURE_DATA_TEST_CASE(RunSmall, NEStackLayerFixture<char>, framework::DatasetMode::ALL,  in TEST_SUITE()
289 …                                           framework::dataset::make("DataType", { DataType::S8 })),  in TEST_SUITE()
298 …                                           framework::dataset::make("DataType", { DataType::S8 })),
309 FIXTURE_DATA_TEST_CASE(RunSmall, NEStackLayerFixture<int>, framework::DatasetMode::ALL,  in TEST_SUITE_END()
311 …                                          framework::dataset::make("DataType", { DataType::S32 })),  in TEST_SUITE_END()
320 …                                          framework::dataset::make("DataType", { DataType::S32 })),
329 FIXTURE_DATA_TEST_CASE(RunSmall, NEStackLayerFixture<short>, framework::DatasetMode::ALL,  in TEST_SUITE()
331 …                                          framework::dataset::make("DataType", { DataType::S16 })),  in TEST_SUITE()
340 …                                          framework::dataset::make("DataType", { DataType::S16 })),
349 FIXTURE_DATA_TEST_CASE(RunSmall, NEStackLayerFixture<char>, framework::DatasetMode::ALL,  in TEST_SUITE()
351 …                                           framework::dataset::make("DataType", { DataType::S8 })),  in TEST_SUITE()
360 …                                           framework::dataset::make("DataType", { DataType::S8 })),