From c8d231626cc204034ab1053f9031ebbb631f1ba9 Mon Sep 17 00:00:00 2001 From: Sanghoon Lee Date: Fri, 12 Jan 2018 16:19:10 +0000 Subject: COMPMID-564: Implement reference and CL/NEON validation for CustomConvolution (output S16) Change-Id: Ic099336f558e994210a59e14ec0171fae68ccb80 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/116663 Reviewed-by: Pablo Tello Tested-by: Jenkins --- tests/validation/reference/Convolution.cpp | 55 +++++++++++++++++++++--------- 1 file changed, 39 insertions(+), 16 deletions(-) (limited to 'tests/validation/reference/Convolution.cpp') diff --git a/tests/validation/reference/Convolution.cpp b/tests/validation/reference/Convolution.cpp index 777e2df400..308c0b5a87 100644 --- a/tests/validation/reference/Convolution.cpp +++ b/tests/validation/reference/Convolution.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017, 2018 ARM Limited. + * Copyright (c) 2017-2018 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -35,34 +35,57 @@ namespace validation namespace reference { template -SimpleTensor convolution(const SimpleTensor &src, const int16_t *conv, uint32_t scale, BorderMode border_mode, T constant_border_value, const unsigned int width, const unsigned int height) +SimpleTensor convolution(const SimpleTensor &src, DataType output_data_type, const int16_t *conv, uint32_t scale, BorderMode border_mode, uint8_t constant_border_value, + const unsigned int width, + const unsigned int height) { - SimpleTensor dst(src.shape(), src.data_type()); - SimpleTensor sum(src.shape(), src.data_type()); + ARM_COMPUTE_ERROR_ON(0 == scale); + + SimpleTensor dst(src.shape(), output_data_type); for(int element_idx = 0; element_idx < src.num_elements(); ++element_idx) { const Coordinates id = index2coord(src.shape(), element_idx); - apply_2d_spatial_filter(id, src, sum, TensorShape(width, height), conv, 1, border_mode, constant_border_value); - if(tensor_elem_at(sum, id, border_mode, constant_border_value) < 0) - { - dst[element_idx] = 0; - } - else if((tensor_elem_at(sum, id, border_mode, constant_border_value) / scale) > 255) + switch(output_data_type) { - dst[element_idx] = 255; - } - else - { - dst[element_idx] = tensor_elem_at(sum, id, border_mode, constant_border_value) / scale; + case DataType::S16: + { + SimpleTensor sum(src.shape(), output_data_type); + apply_2d_spatial_filter(id, src, sum, TensorShape(width, height), conv, 1 / static_cast(scale), border_mode, constant_border_value); + dst[element_idx] = tensor_elem_at(sum, id, border_mode, constant_border_value); + } + break; + case DataType::U8: + { + SimpleTensor sum(src.shape(), output_data_type); + apply_2d_spatial_filter(id, src, sum, TensorShape(width, height), conv, 1, border_mode, constant_border_value); + if(tensor_elem_at(sum, id, border_mode, constant_border_value) < 0) + { + dst[element_idx] = 0; + } + else if((tensor_elem_at(sum, id, border_mode, constant_border_value) / scale) > 255) + { + dst[element_idx] = 255; + } + else + { + dst[element_idx] = tensor_elem_at(sum, id, border_mode, constant_border_value) / scale; + } + } + break; + default: + ARM_COMPUTE_ERROR("Not supported DataType"); + break; } } return dst; } -template SimpleTensor convolution(const SimpleTensor &src, const int16_t *conv, uint32_t scale, BorderMode border_mode, uint8_t constant_border_value, +template SimpleTensor convolution(const SimpleTensor &src, DataType output_data_type, const int16_t *conv, uint32_t scale, BorderMode border_mode, uint8_t constant_border_value, + const unsigned int widht, const unsigned int height); +template SimpleTensor convolution(const SimpleTensor &src, DataType output_data_type, const int16_t *conv, uint32_t scale, BorderMode border_mode, uint8_t constant_border_value, const unsigned int widht, const unsigned int height); } // namespace reference } // namespace validation -- cgit v1.2.1