aboutsummaryrefslogtreecommitdiff
path: root/include/armnn/Descriptors.hpp
diff options
context:
space:
mode:
authorJan Eilers <jan.eilers@arm.com>2020-03-11 12:52:46 +0000
committerJim Flynn <jim.flynn@arm.com>2020-03-18 17:50:48 +0000
commitbca73e1c82438f160364a113793d0a2195c665ac (patch)
tree0ebb17f7ce546af5bf1641e7a946d424f0f87b0a /include/armnn/Descriptors.hpp
parent3184c907b2420e6c66485529f336251b2b62aecf (diff)
downloadarmnn-bca73e1c82438f160364a113793d0a2195c665ac.tar.gz
IVGCVSW-4444 Adding Elu end to end test
* implemented activation layer end to end test * adds support for different tolerances in layer tests * added tests for Elu (Ref, Cl, Neon) Signed-off-by: Jan Eilers <jan.eilers@arm.com> Change-Id: I81e28cfb4456e815bae2fb31f5c345134ff2432f
Diffstat (limited to 'include/armnn/Descriptors.hpp')
-rw-r--r--include/armnn/Descriptors.hpp12
1 files changed, 10 insertions, 2 deletions
diff --git a/include/armnn/Descriptors.hpp b/include/armnn/Descriptors.hpp
index f1b29cc6c7..57917261d4 100644
--- a/include/armnn/Descriptors.hpp
+++ b/include/armnn/Descriptors.hpp
@@ -25,15 +25,23 @@ struct ActivationDescriptor
, m_B(0)
{}
+ ActivationDescriptor(armnn::ActivationFunction activation,
+ float a = 0,
+ float b = 0)
+ : m_Function(activation)
+ , m_A(a)
+ , m_B(b)
+ {}
+
bool operator ==(const ActivationDescriptor &rhs) const
{
return m_Function == rhs.m_Function && m_A == rhs.m_B && m_B == rhs.m_B;
}
/// @brief The activation function to use
- /// (Sigmoid, TanH, Linear, ReLu, BoundedReLu, SoftReLu, LeakyReLu, Abs, Sqrt, Square).
+ /// (Sigmoid, TanH, Linear, ReLu, BoundedReLu, SoftReLu, LeakyReLu, Abs, Sqrt, Square, Elu).
ActivationFunction m_Function;
- /// Alpha upper bound value used by the activation functions. (BoundedReLu, Linear, TanH).
+ /// Alpha upper bound value used by the activation functions. (BoundedReLu, Linear, TanH, Elu).
float m_A;
/// Beta lower bound value used by the activation functions. (BoundedReLu, Linear, TanH).
float m_B;