diff options
| author | Michalis Spyrou <michalis.spyrou@arm.com> | 2018-10-11 17:33:32 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:55:45 +0000 |
| commit | 8aaf93e8c12ce93d3d0082d4f4b70376f15536da (patch) | |
| tree | 0922f3dde6fafae181e101df315ef36007801850 /tests/validation/reference/L2NormalizeLayer.cpp | |
| parent | c93691717a6e7ca67e32b4dedd233b8c63b6daf2 (diff) | |
| download | ComputeLibrary-8aaf93e8c12ce93d3d0082d4f4b70376f15536da.tar.gz | |
COMPMID-1632 Add CLL2NormalizationLayer for NHWC and FP32
Change-Id: Iae22554d5fe893fd22a000eab5bfd8275ea06eb3
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/154102
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Tested-by: bsgcomp <bsgcomp@arm.com>
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 |