From e52211e1544a30d24b29523c389116a9e4446e8c Mon Sep 17 00:00:00 2001 From: Ferran Balaguer Date: Mon, 17 Jun 2019 12:23:52 +0100 Subject: IVGCVSW-3286 Add epsilon Reference workload support Change-Id: I5cabbf9d1ef3858be68d6820d14845e512128c5b Signed-off-by: Ferran Balaguer --- src/backends/reference/workloads/RefL2NormalizationWorkload.cpp | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'src/backends/reference/workloads') diff --git a/src/backends/reference/workloads/RefL2NormalizationWorkload.cpp b/src/backends/reference/workloads/RefL2NormalizationWorkload.cpp index ce5699ef0b..3b2ab50c8b 100644 --- a/src/backends/reference/workloads/RefL2NormalizationWorkload.cpp +++ b/src/backends/reference/workloads/RefL2NormalizationWorkload.cpp @@ -61,7 +61,9 @@ RefL2NormalizationWorkload::RefL2NormalizationWorkload( unsigned int index = dataLayout.GetIndex(inputInfo.GetShape(), n, c, h, w); - const float scale = 1.0f / sqrtf(reduction); + float maximum = reduction < m_Data.m_Parameters.m_Eps ? m_Data.m_Parameters.m_Eps : reduction; + + const float scale = 1.0f / sqrtf(maximum); (*inputDecoder)[index]; (*outputEncoder)[index]; -- cgit v1.2.1