blob: ddaa3533f53344f200020cd20f1d153438995476 [file] [log] [blame]
Moritz Pflanzer6db73ce2017-07-19 10:18:42 +01001/*
Matthew Bentham945b8da2023-07-12 11:54:59 +00002 * Copyright (c) 2017-2021, 2023 Arm Limited.
Moritz Pflanzer6db73ce2017-07-19 10:18:42 +01003 *
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#ifndef ARM_COMPUTE_TEST_NORMALIZATION_LAYER_FIXTURE
25#define ARM_COMPUTE_TEST_NORMALIZATION_LAYER_FIXTURE
26
27#include "arm_compute/core/TensorShape.h"
28#include "arm_compute/core/Types.h"
29#include "arm_compute/runtime/Tensor.h"
Moritz Pflanzer6db73ce2017-07-19 10:18:42 +010030#include "tests/AssetsLibrary.h"
31#include "tests/Globals.h"
32#include "tests/IAccessor.h"
Moritz Pflanzera09de0c2017-09-01 20:41:12 +010033#include "tests/framework/Asserts.h"
34#include "tests/framework/Fixture.h"
Georgios Pinitas5a7e7762017-12-01 16:27:29 +000035#include "tests/validation/reference/NormalizationLayer.h"
Moritz Pflanzer6db73ce2017-07-19 10:18:42 +010036
37#include <random>
38
39namespace arm_compute
40{
41namespace test
42{
43namespace validation
44{
45template <typename TensorType, typename AccessorType, typename FunctionType, typename T>
Vidhya Sudhan Loganathan014333d2018-07-02 09:13:49 +010046class NormalizationValidationGenericFixture : public framework::Fixture
Moritz Pflanzer6db73ce2017-07-19 10:18:42 +010047{
48public:
Michele Di Giorgio9d3a8312018-11-20 12:31:24 +000049 void setup(TensorShape shape, NormType norm_type, int norm_size, float beta, bool is_scaled, DataType data_type, DataLayout data_layout)
Moritz Pflanzer6db73ce2017-07-19 10:18:42 +010050 {
Georgios Pinitas41caa622017-11-16 14:37:08 +000051 NormalizationLayerInfo info(norm_type, norm_size, 5, beta, 1.f, is_scaled);
Moritz Pflanzer6db73ce2017-07-19 10:18:42 +010052
Michele Di Giorgio9d3a8312018-11-20 12:31:24 +000053 _target = compute_target(shape, info, data_type, data_layout);
Vidhya Sudhan Loganathan014333d2018-07-02 09:13:49 +010054 _reference = compute_reference(shape, info, data_type);
Moritz Pflanzer6db73ce2017-07-19 10:18:42 +010055 }
56
57protected:
58 template <typename U>
59 void fill(U &&tensor)
60 {
Giorgio Arena4bdd1772020-12-17 16:47:07 +000061 static_assert(std::is_floating_point<T>::value || std::is_same<T, half>::value, "Only floating point data types supported.");
Giorgio Arena33b103b2021-01-08 10:37:15 +000062 using DistributionType = typename std::conditional<std::is_same<T, half>::value, arm_compute::utils::uniform_real_distribution_16bit<T>, std::uniform_real_distribution<T>>::type;
Giorgio Arena4bdd1772020-12-17 16:47:07 +000063
64 DistributionType distribution{ T(-1.0f), T(1.0f) };
Gian Marco Iodicef2cde9b2018-08-23 15:29:16 +010065 library->fill(tensor, distribution, 0);
Moritz Pflanzer6db73ce2017-07-19 10:18:42 +010066 }
67
Michele Di Giorgio9d3a8312018-11-20 12:31:24 +000068 TensorType compute_target(TensorShape shape, NormalizationLayerInfo info, DataType data_type, DataLayout data_layout)
Moritz Pflanzer6db73ce2017-07-19 10:18:42 +010069 {
Michele Di Giorgio9d3a8312018-11-20 12:31:24 +000070 if(data_layout == DataLayout::NHWC)
71 {
72 permute(shape, PermutationVector(2U, 0U, 1U));
73 }
74
Moritz Pflanzer6db73ce2017-07-19 10:18:42 +010075 // Create tensors
Michele Di Giorgio9d3a8312018-11-20 12:31:24 +000076 TensorType src = create_tensor<TensorType>(shape, data_type, 1, QuantizationInfo(), data_layout);
77 TensorType dst = create_tensor<TensorType>(shape, data_type, 1, QuantizationInfo(), data_layout);
Moritz Pflanzer6db73ce2017-07-19 10:18:42 +010078
79 // Create and configure function
80 FunctionType norm_layer;
81 norm_layer.configure(&src, &dst, info);
82
Michele Di Giorgio4fc10b32021-04-30 18:30:41 +010083 ARM_COMPUTE_ASSERT(src.info()->is_resizable());
84 ARM_COMPUTE_ASSERT(dst.info()->is_resizable());
Moritz Pflanzer6db73ce2017-07-19 10:18:42 +010085
86 // Allocate tensors
87 src.allocator()->allocate();
88 dst.allocator()->allocate();
89
Michele Di Giorgio4fc10b32021-04-30 18:30:41 +010090 ARM_COMPUTE_ASSERT(!src.info()->is_resizable());
91 ARM_COMPUTE_ASSERT(!dst.info()->is_resizable());
Moritz Pflanzer6db73ce2017-07-19 10:18:42 +010092
93 // Fill tensors
94 fill(AccessorType(src));
95
96 // Compute function
97 norm_layer.run();
98
99 return dst;
100 }
101
Vidhya Sudhan Loganathan014333d2018-07-02 09:13:49 +0100102 SimpleTensor<T> compute_reference(const TensorShape &shape, NormalizationLayerInfo info, DataType data_type)
Moritz Pflanzer6db73ce2017-07-19 10:18:42 +0100103 {
104 // Create reference
Vidhya Sudhan Loganathan014333d2018-07-02 09:13:49 +0100105 SimpleTensor<T> src{ shape, data_type, 1 };
Moritz Pflanzer6db73ce2017-07-19 10:18:42 +0100106
107 // Fill reference
108 fill(src);
109
110 return reference::normalization_layer<T>(src, info);
111 }
112
113 TensorType _target{};
114 SimpleTensor<T> _reference{};
Moritz Pflanzer6db73ce2017-07-19 10:18:42 +0100115};
116
117template <typename TensorType, typename AccessorType, typename FunctionType, typename T>
Vidhya Sudhan Loganathan014333d2018-07-02 09:13:49 +0100118class NormalizationValidationFixture : public NormalizationValidationGenericFixture<TensorType, AccessorType, FunctionType, T>
Moritz Pflanzer6db73ce2017-07-19 10:18:42 +0100119{
120public:
Michele Di Giorgio9d3a8312018-11-20 12:31:24 +0000121 void setup(TensorShape shape, NormType norm_type, int norm_size, float beta, bool is_scaled, DataType data_type, DataLayout data_layout)
Moritz Pflanzer6db73ce2017-07-19 10:18:42 +0100122 {
Michele Di Giorgio9d3a8312018-11-20 12:31:24 +0000123 NormalizationValidationGenericFixture<TensorType, AccessorType, FunctionType, T>::setup(shape, norm_type, norm_size, beta, is_scaled, data_type, data_layout);
Moritz Pflanzer6db73ce2017-07-19 10:18:42 +0100124 }
125};
126} // namespace validation
127} // namespace test
128} // namespace arm_compute
129#endif /* ARM_COMPUTE_TEST_NORMALIZATION_LAYER_FIXTURE */