ActivationDescriptor Struct Reference

An ActivationDescriptor for the ActivationLayer. More...

#include <Descriptors.hpp>

Public Member Functions
	ActivationDescriptor ()
	ActivationDescriptor (armnn::ActivationFunction activation, float a=0, float b=0)
bool	operator== (const ActivationDescriptor &rhs) const

Public Attributes
ActivationFunction	m_Function
	The activation function to use (Sigmoid, TanH, Linear, ReLu, BoundedReLu, SoftReLu, LeakyReLu, Abs, Sqrt, Square, Elu). More...
float	m_A
	Alpha upper bound value used by the activation functions. (BoundedReLu, Linear, TanH, Elu). More...
float	m_B
	Beta lower bound value used by the activation functions. (BoundedReLu, Linear, TanH). More...

Detailed Description

An ActivationDescriptor for the ActivationLayer.

Definition at line 20 of file Descriptors.hpp.

Constructor & Destructor Documentation

ActivationDescriptor() [1/2]

ActivationDescriptor ( )

inline

Definition at line 22 of file Descriptors.hpp.

        : m_Function(ActivationFunction::Sigmoid)
        , m_A(0)
        , m_B(0)
    {}

ActivationDescriptor() [2/2]

ActivationDescriptor ( armnn::ActivationFunction  activation, float  a = 0, float  b = 0  )

inline

Definition at line 28 of file Descriptors.hpp.

            : m_Function(activation)
            , m_A(a)
            , m_B(b)
    {}

Member Function Documentation

operator==()

bool operator== ( const ActivationDescriptor &  rhs ) const

inline

Definition at line 36 of file Descriptors.hpp.

References ActivationDescriptor::m_A, ActivationDescriptor::m_B, and ActivationDescriptor::m_Function.

    {
        return m_Function == rhs.m_Function && m_A == rhs.m_B && m_B == rhs.m_B;
    }

Member Data Documentation

m_A

float m_A

Alpha upper bound value used by the activation functions. (BoundedReLu, Linear, TanH, Elu).

Definition at line 45 of file Descriptors.hpp.

Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), CompareBoundedReLuTest(), armnnTfLiteParser::ComputeWrappedIndex(), armnn::ConvertActivationDescriptorToAclActivationLayerInfo(), OnnxParser::CreateNetworkFromString(), armnn::CreateStartOfLeakyReluNetwork(), RefActivationWorkload::Execute(), Deserializer::GetNetworkOutputBindingInfo(), ActivationDescriptor::operator==(), armnnTfParser::OutputShapeOfExpandDims(), TfLiteParser::OutputShapeOfSqueeze(), armnnTfParser::OutputShapeOfSqueeze(), CaffeParserBase::ParseReluLayer(), StringifyLayerParameters< ActivationDescriptor >::Serialize(), StaticRangeVisitor::VisitActivationLayer(), SerializerVisitor::VisitActivationLayer(), and DynamicQuantizationVisitor::VisitActivationLayer().

m_B

float m_B

Beta lower bound value used by the activation functions. (BoundedReLu, Linear, TanH).

Definition at line 47 of file Descriptors.hpp.

Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), CompareBoundedReLuTest(), armnnTfLiteParser::ComputeWrappedIndex(), armnn::ConvertActivationDescriptorToAclActivationLayerInfo(), OnnxParser::CreateNetworkFromString(), armnn::CreateStartOfLeakyReluNetwork(), RefActivationWorkload::Execute(), Deserializer::GetNetworkOutputBindingInfo(), ActivationDescriptor::operator==(), TfLiteParser::OutputShapeOfSqueeze(), armnnTfParser::OutputShapeOfSqueeze(), StringifyLayerParameters< ActivationDescriptor >::Serialize(), and SerializerVisitor::VisitActivationLayer().

m_Function

ActivationFunction m_Function

The activation function to use (Sigmoid, TanH, Linear, ReLu, BoundedReLu, SoftReLu, LeakyReLu, Abs, Sqrt, Square, Elu).

Definition at line 43 of file Descriptors.hpp.

Referenced by armnn::BOOST_AUTO_TEST_CASE(), BOOST_AUTO_TEST_CASE(), CompareBoundedReLuTest(), armnnTfLiteParser::ComputeWrappedIndex(), armnn::ConvertActivationDescriptorToAclActivationLayerInfo(), OnnxParser::CreateNetworkFromString(), armnn::CreateStartOfLeakyReluNetwork(), RefActivationWorkload::Execute(), Deserializer::GetNetworkOutputBindingInfo(), RefLayerSupport::IsActivationSupported(), ActivationDescriptor::operator==(), ArgMinMaxDescriptor::operator==(), armnnTfParser::OutputShapeOfExpandDims(), TfLiteParser::OutputShapeOfSqueeze(), armnnTfParser::OutputShapeOfSqueeze(), CaffeParserBase::ParseReluLayer(), StringifyLayerParameters< ActivationDescriptor >::Serialize(), StaticRangeVisitor::VisitActivationLayer(), SerializerVisitor::VisitActivationLayer(), and DynamicQuantizationVisitor::VisitActivationLayer().

---

The documentation for this struct was generated from the following file:

• include/armnn/Descriptors.hpp