aboutsummaryrefslogtreecommitdiff
path: root/tests/validation/reference/Remap.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'tests/validation/reference/Remap.cpp')
-rw-r--r--tests/validation/reference/Remap.cpp9
1 files changed, 5 insertions, 4 deletions
diff --git a/tests/validation/reference/Remap.cpp b/tests/validation/reference/Remap.cpp
index f862c13700..a7352eb3a3 100644
--- a/tests/validation/reference/Remap.cpp
+++ b/tests/validation/reference/Remap.cpp
@@ -1,5 +1,5 @@
/*
- * Copyright (c) 2017-2018 ARM Limited.
+ * Copyright (c) 2017-2019 ARM Limited.
*
* SPDX-License-Identifier: MIT
*
@@ -44,9 +44,10 @@ SimpleTensor<T> remap(const SimpleTensor<T> &in, SimpleTensor<float> &map_x, Sim
ARM_COMPUTE_ERROR_ON_MSG(border_mode == BorderMode::REPLICATE, "BorderMode not supported");
SimpleTensor<T> out(in.shape(), in.data_type());
ARM_COMPUTE_ERROR_ON(out.num_elements() != map_x.num_elements());
- const int width = in.shape().x();
- const int height = in.shape().y();
- for(int idx = 0; idx < out.num_elements(); idx++)
+ const int width = in.shape().x();
+ const int height = in.shape().y();
+ const uint32_t num_elements = out.num_elements();
+ for(uint32_t idx = 0; idx < num_elements; idx++)
{
const Coordinates id_out = index2coord(out.shape(), idx);
valid_mask[idx] = 1;