commit e52211e1544a30d24b29523c389116a9e4446e8c
author Ferran Balaguer <ferran.balaguer@arm.com> Mon Jun 17 12:23:52 2019 +0100
committer Ferran Balaguer Arm <ferran.balaguer@arm.com> Mon Jun 17 14:53:05 2019 +0000
tree 8d2cce39db4c89509a8b6c8569e85edb6d348b4b
parent 0b679f03ead8848edc225d44afbbcc54d112e433

IVGCVSW-3286 Add epsilon Reference workload support

Change-Id: I5cabbf9d1ef3858be68d6820d14845e512128c5b
Signed-off-by: Ferran Balaguer <ferran.balaguer@arm.com>

src/backends/reference/workloads/RefL2NormalizationWorkload.cpp

diff --git a/src/backends/reference/workloads/RefL2NormalizationWorkload.cpp b/src/backends/reference/workloads/RefL2NormalizationWorkload.cpp
index ce5699e..3b2ab50 100644
--- a/src/backends/reference/workloads/RefL2NormalizationWorkload.cpp
+++ b/src/backends/reference/workloads/RefL2NormalizationWorkload.cpp
@@ -61,7 +61,9 @@
 
                         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];