diff options
Diffstat (limited to 'tests/validation/reference/Permute.cpp')
-rw-r--r-- | tests/validation/reference/Permute.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tests/validation/reference/Permute.cpp b/tests/validation/reference/Permute.cpp index 619a787a05..36b07dc277 100644 --- a/tests/validation/reference/Permute.cpp +++ b/tests/validation/reference/Permute.cpp @@ -45,7 +45,8 @@ SimpleTensor<T> permute(const SimpleTensor<T> &src, PermutationVector perm) SimpleTensor<T> dst{ dst_shape, src.data_type(), src.num_channels(), src.quantization_info() }; // Compute reference - for(int i = 0; i < src.num_elements(); ++i) + const uint32_t num_elements = src.num_elements(); + for(uint32_t i = 0; i < num_elements; ++i) { const Coordinates src_coords = index2coord(src.shape(), i); Coordinates dst_coords = src_coords; |