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 --- include/armnn/Descriptors.hpp | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) (limited to 'include/armnn') diff --git a/include/armnn/Descriptors.hpp b/include/armnn/Descriptors.hpp index 9479db3571..dd1991d569 100644 --- a/include/armnn/Descriptors.hpp +++ b/include/armnn/Descriptors.hpp @@ -424,9 +424,12 @@ struct NormalizationDescriptor struct L2NormalizationDescriptor { L2NormalizationDescriptor() - : m_DataLayout(DataLayout::NCHW) + : m_Eps(1e-12f) + , m_DataLayout(DataLayout::NCHW) {} + /// Used to avoid dividing by zero. + float m_Eps; /// The data layout to be used (NCHW, NHWC). DataLayout m_DataLayout; }; -- cgit v1.2.1