diff options
Diffstat (limited to 'tests/validation/reference/L2NormalizeLayer.cpp')
| -rw-r--r-- | tests/validation/reference/L2NormalizeLayer.cpp | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tests/validation/reference/L2NormalizeLayer.cpp b/tests/validation/reference/L2NormalizeLayer.cpp index 99f4e8a6e6..26677511e4 100644 --- a/tests/validation/reference/L2NormalizeLayer.cpp +++ b/tests/validation/reference/L2NormalizeLayer.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017 ARM Limited. + * Copyright (c) 2017-2018 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -66,7 +66,7 @@ SimpleTensor<T> l2_normalize(const SimpleTensor<T> &src, unsigned int axis, floa { const T *src_row_ptr = src.data() + du * elems; T *dst_row_ptr = dst.data() + du * elems; - const T normalization_value = std::sqrt(std::max(sum[du], epsilon)); + const T normalization_value = sqrt(std::max(sum[du], static_cast<T>(epsilon))); std::transform(src_row_ptr, src_row_ptr + elems, dst_row_ptr, [normalization_value](T val) { return val / normalization_value; @@ -82,6 +82,7 @@ SimpleTensor<T> l2_normalize(const SimpleTensor<T> &src, unsigned int axis, floa } template SimpleTensor<float> l2_normalize(const SimpleTensor<float> &src, unsigned int axis, float epsilon); +template SimpleTensor<half> l2_normalize(const SimpleTensor<half> &src, unsigned int axis, float epsilon); } // namespace reference } // namespace validation } // namespace test |