diff options
Diffstat (limited to 'tests/validation/reference')
-rw-r--r-- | tests/validation/reference/Convolution.cpp | 8 | ||||
-rw-r--r-- | tests/validation/reference/Convolution.h | 4 |
2 files changed, 6 insertions, 6 deletions
diff --git a/tests/validation/reference/Convolution.cpp b/tests/validation/reference/Convolution.cpp index 84be858cfc..777e2df400 100644 --- a/tests/validation/reference/Convolution.cpp +++ b/tests/validation/reference/Convolution.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017 ARM Limited. + * Copyright (c) 2017, 2018 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -35,7 +35,7 @@ namespace validation namespace reference { template <typename T> -SimpleTensor<T> convolution(const SimpleTensor<T> &src, const int16_t *conv, uint32_t scale, BorderMode border_mode, T constant_border_value, const unsigned int filter_size) +SimpleTensor<T> convolution(const SimpleTensor<T> &src, const int16_t *conv, uint32_t scale, BorderMode border_mode, T constant_border_value, const unsigned int width, const unsigned int height) { SimpleTensor<T> dst(src.shape(), src.data_type()); SimpleTensor<int32_t> sum(src.shape(), src.data_type()); @@ -43,7 +43,7 @@ SimpleTensor<T> convolution(const SimpleTensor<T> &src, const int16_t *conv, uin 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(filter_size, filter_size), conv, 1, border_mode, constant_border_value); + apply_2d_spatial_filter(id, src, sum, TensorShape(width, height), conv, 1, border_mode, constant_border_value); if(tensor_elem_at<int32_t>(sum, id, border_mode, constant_border_value) < 0) { @@ -63,7 +63,7 @@ SimpleTensor<T> convolution(const SimpleTensor<T> &src, const int16_t *conv, uin } template SimpleTensor<uint8_t> convolution(const SimpleTensor<uint8_t> &src, const int16_t *conv, uint32_t scale, BorderMode border_mode, uint8_t constant_border_value, - const unsigned int filter_size); + const unsigned int widht, const unsigned int height); } // namespace reference } // namespace validation } // namespace test diff --git a/tests/validation/reference/Convolution.h b/tests/validation/reference/Convolution.h index bdaac28ae6..ea9f4e444b 100644 --- a/tests/validation/reference/Convolution.h +++ b/tests/validation/reference/Convolution.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017 ARM Limited. + * Copyright (c) 2017, 2018 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -35,7 +35,7 @@ namespace validation namespace reference { template <typename T> -SimpleTensor<T> convolution(const SimpleTensor<T> &src, const int16_t *conv, uint32_t scale, BorderMode border_mode, T constant_border_value, const unsigned int filter_size); +SimpleTensor<T> convolution(const SimpleTensor<T> &src, const int16_t *conv, uint32_t scale, BorderMode border_mode, T constant_border_value, const unsigned int width, const unsigned int height); } // namespace reference } // namespace validation } // namespace test |