diff options
Diffstat (limited to 'tests/validation/reference/Median3x3.cpp')
| -rw-r--r-- | tests/validation/reference/Median3x3.cpp | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tests/validation/reference/Median3x3.cpp b/tests/validation/reference/Median3x3.cpp index 91a787a0a3..314bbe38e6 100644 --- a/tests/validation/reference/Median3x3.cpp +++ b/tests/validation/reference/Median3x3.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017 ARM Limited. + * Copyright (c) 2017-2019 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -45,8 +45,9 @@ SimpleTensor<T> median3x3(const SimpleTensor<T> &src, BorderMode border_mode, T { SimpleTensor<T> dst(src.shape(), src.data_type()); const int size_tot_filter = filter_size * filter_size; + const uint32_t num_elements = src.num_elements(); - for(int src_idx = 0; src_idx < src.num_elements(); ++src_idx) + for(uint32_t src_idx = 0; src_idx < num_elements; ++src_idx) { std::array<T, size_tot_filter> filter_elems = { { 0 } }; Coordinates id = index2coord(src.shape(), src_idx); |