1*c217d954SCole Faust /* 2*c217d954SCole Faust * Copyright (c) 2017-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 #ifndef ARM_COMPUTE_TEST_POOLING_LAYER_FIXTURE 25*c217d954SCole Faust #define ARM_COMPUTE_TEST_POOLING_LAYER_FIXTURE 26*c217d954SCole Faust 27*c217d954SCole Faust #include "arm_compute/core/TensorShape.h" 28*c217d954SCole Faust #include "arm_compute/core/Types.h" 29*c217d954SCole Faust #include "arm_compute/core/utils/misc/ShapeCalculator.h" 30*c217d954SCole Faust #include "arm_compute/runtime/Tensor.h" 31*c217d954SCole Faust #include "tests/AssetsLibrary.h" 32*c217d954SCole Faust #include "tests/Globals.h" 33*c217d954SCole Faust #include "tests/IAccessor.h" 34*c217d954SCole Faust #include "tests/framework/Asserts.h" 35*c217d954SCole Faust #include "tests/framework/Fixture.h" 36*c217d954SCole Faust #include "tests/validation/reference/PoolingLayer.h" 37*c217d954SCole Faust #include <random> 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 template <typename TensorType, typename AccessorType, typename FunctionType, typename T> 45*c217d954SCole Faust class PoolingLayerValidationGenericFixture : public framework::Fixture 46*c217d954SCole Faust { 47*c217d954SCole Faust public: 48*c217d954SCole Faust template <typename...> 49*c217d954SCole Faust void setup(TensorShape shape, PoolingLayerInfo pool_info, DataType data_type, DataLayout data_layout, bool indices = false, 50*c217d954SCole Faust QuantizationInfo input_qinfo = QuantizationInfo(), QuantizationInfo output_qinfo = QuantizationInfo(), bool mixed_layout = false) 51*c217d954SCole Faust { 52*c217d954SCole Faust _mixed_layout = mixed_layout; 53*c217d954SCole Faust _pool_info = pool_info; 54*c217d954SCole Faust _target = compute_target(shape, pool_info, data_type, data_layout, input_qinfo, output_qinfo, indices); 55*c217d954SCole Faust _reference = compute_reference(shape, pool_info, data_type, data_layout, input_qinfo, output_qinfo, indices); 56*c217d954SCole Faust } 57*c217d954SCole Faust 58*c217d954SCole Faust protected: mix_layout(FunctionType & layer,TensorType & src,TensorType & dst)59*c217d954SCole Faust void mix_layout(FunctionType &layer, TensorType &src, TensorType &dst) 60*c217d954SCole Faust { 61*c217d954SCole Faust const DataLayout data_layout = src.info()->data_layout(); 62*c217d954SCole Faust // Test Multi DataLayout graph cases, when the data layout changes after configure 63*c217d954SCole Faust src.info()->set_data_layout(data_layout == DataLayout::NCHW ? DataLayout::NHWC : DataLayout::NCHW); 64*c217d954SCole Faust dst.info()->set_data_layout(data_layout == DataLayout::NCHW ? DataLayout::NHWC : DataLayout::NCHW); 65*c217d954SCole Faust 66*c217d954SCole Faust // Compute Convolution function 67*c217d954SCole Faust layer.run(); 68*c217d954SCole Faust 69*c217d954SCole Faust // Reinstating original data layout for the test suite to properly check the values 70*c217d954SCole Faust src.info()->set_data_layout(data_layout); 71*c217d954SCole Faust dst.info()->set_data_layout(data_layout); 72*c217d954SCole Faust } 73*c217d954SCole Faust 74*c217d954SCole Faust template <typename U> fill(U && tensor)75*c217d954SCole Faust void fill(U &&tensor) 76*c217d954SCole Faust { 77*c217d954SCole Faust if(tensor.data_type() == DataType::F32) 78*c217d954SCole Faust { 79*c217d954SCole Faust std::uniform_real_distribution<float> distribution(-1.0f, 1.0f); 80*c217d954SCole Faust library->fill(tensor, distribution, 0); 81*c217d954SCole Faust } 82*c217d954SCole Faust else if(tensor.data_type() == DataType::F16) 83*c217d954SCole Faust { 84*c217d954SCole Faust arm_compute::utils::uniform_real_distribution_16bit<half> distribution{ -1.0f, 1.0f }; 85*c217d954SCole Faust library->fill(tensor, distribution, 0); 86*c217d954SCole Faust } 87*c217d954SCole Faust else // data type is quantized_asymmetric 88*c217d954SCole Faust { 89*c217d954SCole Faust library->fill_tensor_uniform(tensor, 0); 90*c217d954SCole Faust } 91*c217d954SCole Faust } 92*c217d954SCole Faust compute_target(TensorShape shape,PoolingLayerInfo info,DataType data_type,DataLayout data_layout,QuantizationInfo input_qinfo,QuantizationInfo output_qinfo,bool indices)93*c217d954SCole Faust TensorType compute_target(TensorShape shape, PoolingLayerInfo info, 94*c217d954SCole Faust DataType data_type, DataLayout data_layout, 95*c217d954SCole Faust QuantizationInfo input_qinfo, QuantizationInfo output_qinfo, 96*c217d954SCole Faust bool indices) 97*c217d954SCole Faust { 98*c217d954SCole Faust // Change shape in case of NHWC. 99*c217d954SCole Faust if(data_layout == DataLayout::NHWC) 100*c217d954SCole Faust { 101*c217d954SCole Faust permute(shape, PermutationVector(2U, 0U, 1U)); 102*c217d954SCole Faust } 103*c217d954SCole Faust // Create tensors 104*c217d954SCole Faust TensorType src = create_tensor<TensorType>(shape, data_type, 1, input_qinfo, data_layout); 105*c217d954SCole Faust const TensorShape dst_shape = misc::shape_calculator::compute_pool_shape(*(src.info()), info); 106*c217d954SCole Faust TensorType dst = create_tensor<TensorType>(dst_shape, data_type, 1, output_qinfo, data_layout); 107*c217d954SCole Faust _target_indices = create_tensor<TensorType>(dst_shape, DataType::U32, 1, output_qinfo, data_layout); 108*c217d954SCole Faust 109*c217d954SCole Faust // Create and configure function 110*c217d954SCole Faust FunctionType pool_layer; 111*c217d954SCole Faust pool_layer.configure(&src, &dst, info, (indices) ? &_target_indices : nullptr); 112*c217d954SCole Faust 113*c217d954SCole Faust ARM_COMPUTE_ASSERT(src.info()->is_resizable()); 114*c217d954SCole Faust ARM_COMPUTE_ASSERT(dst.info()->is_resizable()); 115*c217d954SCole Faust ARM_COMPUTE_ASSERT(_target_indices.info()->is_resizable()); 116*c217d954SCole Faust 117*c217d954SCole Faust add_padding_x({ &src, &dst, &_target_indices }, data_layout); 118*c217d954SCole Faust 119*c217d954SCole Faust // Allocate tensors 120*c217d954SCole Faust src.allocator()->allocate(); 121*c217d954SCole Faust dst.allocator()->allocate(); 122*c217d954SCole Faust _target_indices.allocator()->allocate(); 123*c217d954SCole Faust 124*c217d954SCole Faust ARM_COMPUTE_ASSERT(!src.info()->is_resizable()); 125*c217d954SCole Faust ARM_COMPUTE_ASSERT(!dst.info()->is_resizable()); 126*c217d954SCole Faust ARM_COMPUTE_ASSERT(!_target_indices.info()->is_resizable()); 127*c217d954SCole Faust 128*c217d954SCole Faust // Fill tensors 129*c217d954SCole Faust fill(AccessorType(src)); 130*c217d954SCole Faust 131*c217d954SCole Faust if(_mixed_layout) 132*c217d954SCole Faust { 133*c217d954SCole Faust mix_layout(pool_layer, src, dst); 134*c217d954SCole Faust } 135*c217d954SCole Faust else 136*c217d954SCole Faust { 137*c217d954SCole Faust // Compute function 138*c217d954SCole Faust pool_layer.run(); 139*c217d954SCole Faust } 140*c217d954SCole Faust return dst; 141*c217d954SCole Faust } 142*c217d954SCole Faust compute_reference(TensorShape shape,PoolingLayerInfo info,DataType data_type,DataLayout data_layout,QuantizationInfo input_qinfo,QuantizationInfo output_qinfo,bool indices)143*c217d954SCole Faust SimpleTensor<T> compute_reference(TensorShape shape, PoolingLayerInfo info, DataType data_type, DataLayout data_layout, 144*c217d954SCole Faust QuantizationInfo input_qinfo, QuantizationInfo output_qinfo, bool indices) 145*c217d954SCole Faust { 146*c217d954SCole Faust // Create reference 147*c217d954SCole Faust SimpleTensor<T> src(shape, data_type, 1, input_qinfo); 148*c217d954SCole Faust // Fill reference 149*c217d954SCole Faust fill(src); 150*c217d954SCole Faust return reference::pooling_layer<T>(src, info, output_qinfo, indices ? &_ref_indices : nullptr, data_layout); 151*c217d954SCole Faust } 152*c217d954SCole Faust 153*c217d954SCole Faust TensorType _target{}; 154*c217d954SCole Faust SimpleTensor<T> _reference{}; 155*c217d954SCole Faust PoolingLayerInfo _pool_info{}; 156*c217d954SCole Faust bool _mixed_layout{ false }; 157*c217d954SCole Faust TensorType _target_indices{}; 158*c217d954SCole Faust SimpleTensor<uint32_t> _ref_indices{}; 159*c217d954SCole Faust }; 160*c217d954SCole Faust template <typename TensorType, typename AccessorType, typename FunctionType, typename T> 161*c217d954SCole Faust class PoolingLayerIndicesValidationFixture : public PoolingLayerValidationGenericFixture<TensorType, AccessorType, FunctionType, T> 162*c217d954SCole Faust { 163*c217d954SCole Faust public: 164*c217d954SCole Faust template <typename...> setup(TensorShape shape,PoolingType pool_type,Size2D pool_size,PadStrideInfo pad_stride_info,bool exclude_padding,DataType data_type,DataLayout data_layout)165*c217d954SCole Faust void setup(TensorShape shape, PoolingType pool_type, Size2D pool_size, PadStrideInfo pad_stride_info, bool exclude_padding, DataType data_type, DataLayout data_layout) 166*c217d954SCole Faust { 167*c217d954SCole Faust PoolingLayerValidationGenericFixture<TensorType, AccessorType, FunctionType, T>::setup(shape, PoolingLayerInfo(pool_type, pool_size, data_layout, pad_stride_info, exclude_padding), 168*c217d954SCole Faust data_type, data_layout, true); 169*c217d954SCole Faust } 170*c217d954SCole Faust }; 171*c217d954SCole Faust 172*c217d954SCole Faust template <typename TensorType, typename AccessorType, typename FunctionType, typename T, bool mixed_layout = false> 173*c217d954SCole Faust class PoolingLayerValidationFixture : public PoolingLayerValidationGenericFixture<TensorType, AccessorType, FunctionType, T> 174*c217d954SCole Faust { 175*c217d954SCole Faust public: 176*c217d954SCole Faust template <typename...> setup(TensorShape shape,PoolingType pool_type,Size2D pool_size,PadStrideInfo pad_stride_info,bool exclude_padding,DataType data_type,DataLayout data_layout)177*c217d954SCole Faust void setup(TensorShape shape, PoolingType pool_type, Size2D pool_size, PadStrideInfo pad_stride_info, bool exclude_padding, DataType data_type, DataLayout data_layout) 178*c217d954SCole Faust { 179*c217d954SCole Faust PoolingLayerValidationGenericFixture<TensorType, AccessorType, FunctionType, T>::setup(shape, PoolingLayerInfo(pool_type, pool_size, data_layout, pad_stride_info, exclude_padding), 180*c217d954SCole Faust data_type, data_layout, false, mixed_layout); 181*c217d954SCole Faust } 182*c217d954SCole Faust }; 183*c217d954SCole Faust 184*c217d954SCole Faust template <typename TensorType, typename AccessorType, typename FunctionType, typename T> 185*c217d954SCole Faust class PoolingLayerValidationMixedPrecisionFixture : public PoolingLayerValidationGenericFixture<TensorType, AccessorType, FunctionType, T> 186*c217d954SCole Faust { 187*c217d954SCole Faust public: 188*c217d954SCole Faust template <typename...> 189*c217d954SCole Faust void setup(TensorShape shape, PoolingType pool_type, Size2D pool_size, PadStrideInfo pad_stride_info, bool exclude_padding, DataType data_type, DataLayout data_layout, bool fp_mixed_precision = false) 190*c217d954SCole Faust { 191*c217d954SCole Faust PoolingLayerValidationGenericFixture<TensorType, AccessorType, FunctionType, T>::setup(shape, PoolingLayerInfo(pool_type, pool_size, data_layout, pad_stride_info, exclude_padding, fp_mixed_precision), 192*c217d954SCole Faust data_type, data_layout); 193*c217d954SCole Faust } 194*c217d954SCole Faust }; 195*c217d954SCole Faust 196*c217d954SCole Faust template <typename TensorType, typename AccessorType, typename FunctionType, typename T, bool mixed_layout = false> 197*c217d954SCole Faust class PoolingLayerValidationQuantizedFixture : public PoolingLayerValidationGenericFixture<TensorType, AccessorType, FunctionType, T> 198*c217d954SCole Faust { 199*c217d954SCole Faust public: 200*c217d954SCole Faust template <typename...> 201*c217d954SCole Faust void setup(TensorShape shape, PoolingType pool_type, Size2D pool_size, PadStrideInfo pad_stride_info, bool exclude_padding, DataType data_type, DataLayout data_layout = DataLayout::NCHW, 202*c217d954SCole Faust QuantizationInfo input_qinfo = QuantizationInfo(), QuantizationInfo output_qinfo = QuantizationInfo()) 203*c217d954SCole Faust { 204*c217d954SCole Faust PoolingLayerValidationGenericFixture<TensorType, AccessorType, FunctionType, T>::setup(shape, PoolingLayerInfo(pool_type, pool_size, data_layout, pad_stride_info, exclude_padding), 205*c217d954SCole Faust data_type, data_layout, false, input_qinfo, output_qinfo, mixed_layout); 206*c217d954SCole Faust } 207*c217d954SCole Faust }; 208*c217d954SCole Faust 209*c217d954SCole Faust template <typename TensorType, typename AccessorType, typename FunctionType, typename T> 210*c217d954SCole Faust class SpecialPoolingLayerValidationFixture : public PoolingLayerValidationGenericFixture<TensorType, AccessorType, FunctionType, T> 211*c217d954SCole Faust { 212*c217d954SCole Faust public: 213*c217d954SCole Faust template <typename...> setup(TensorShape src_shape,PoolingLayerInfo pool_info,DataType data_type)214*c217d954SCole Faust void setup(TensorShape src_shape, PoolingLayerInfo pool_info, DataType data_type) 215*c217d954SCole Faust { 216*c217d954SCole Faust PoolingLayerValidationGenericFixture<TensorType, AccessorType, FunctionType, T>::setup(src_shape, pool_info, data_type, pool_info.data_layout); 217*c217d954SCole Faust } 218*c217d954SCole Faust }; 219*c217d954SCole Faust 220*c217d954SCole Faust template <typename TensorType, typename AccessorType, typename FunctionType, typename T> 221*c217d954SCole Faust class GlobalPoolingLayerValidationFixture : public PoolingLayerValidationGenericFixture<TensorType, AccessorType, FunctionType, T> 222*c217d954SCole Faust { 223*c217d954SCole Faust public: 224*c217d954SCole Faust template <typename...> 225*c217d954SCole Faust void setup(TensorShape shape, PoolingType pool_type, DataType data_type, DataLayout data_layout = DataLayout::NCHW) 226*c217d954SCole Faust { 227*c217d954SCole Faust PoolingLayerValidationGenericFixture<TensorType, AccessorType, FunctionType, T>::setup(shape, PoolingLayerInfo(pool_type, data_layout), data_type, data_layout); 228*c217d954SCole Faust } 229*c217d954SCole Faust }; 230*c217d954SCole Faust 231*c217d954SCole Faust } // namespace validation 232*c217d954SCole Faust } // namespace test 233*c217d954SCole Faust } // namespace arm_compute 234*c217d954SCole Faust #endif /* ARM_COMPUTE_TEST_POOLING_LAYER_FIXTURE */ 235