ActivationDescriptor Struct Reference

An ActivationDescriptor for the ActivationLayer. More...

#include <Descriptors.hpp>

Inheritance diagram for ActivationDescriptor:

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 25 of file Descriptors.hpp.

Constructor & Destructor Documentation

ActivationDescriptor() [1/2]

ActivationDescriptor ( )

inline

Definition at line 27 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 33 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 41 of file Descriptors.hpp.

References 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 50 of file Descriptors.hpp.

Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), BoundedReLuTestCommon(), CompareActivationTestImpl(), CompareBoundedReLuTest(), armnnTfLiteParser::ComputeWrappedIndex(), ConstantLinearActivationTestCommon(), armnn::ConvertActivationDescriptorToAclActivationLayerInfo(), OnnxParserImpl::CreateNetworkFromString(), armnn::CreateStartOfLeakyReluNetwork(), RefActivationWorkload::Execute(), StaticRangeStrategy::ExecuteStrategy(), DynamicQuantizationStrategy::ExecuteStrategy(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetNetworkOutputBindingInfo(), ITfParser::TfParserImpl::IsSupportedLeakyReluPattern(), TfLiteParserImpl::OutputShapeOfSqueeze(), ITfParser::TfParserImpl::ParseRelu6(), ICaffeParser::CaffeParserImpl::ParseReluLayer(), ITfParser::TfParserImpl::ParseTanh(), StringifyLayerParameters< ActivationDescriptor >::Serialize(), and SimpleActivationTest().

m_B

float m_B

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

Definition at line 52 of file Descriptors.hpp.

Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), BoundedReLuTestCommon(), CompareActivationTestImpl(), CompareBoundedReLuTest(), armnnTfLiteParser::ComputeWrappedIndex(), ConstantLinearActivationTestCommon(), armnn::ConvertActivationDescriptorToAclActivationLayerInfo(), OnnxParserImpl::CreateNetworkFromString(), armnn::CreateStartOfLeakyReluNetwork(), RefActivationWorkload::Execute(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetNetworkOutputBindingInfo(), ActivationDescriptor::operator==(), TfLiteParserImpl::OutputShapeOfSqueeze(), ITfParser::TfParserImpl::ParseRelu6(), ITfParser::TfParserImpl::ParseTanh(), StringifyLayerParameters< ActivationDescriptor >::Serialize(), and SimpleActivationTest().

m_Function

ActivationFunction m_Function

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

Definition at line 48 of file Descriptors.hpp.

Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), BoundedReLuTestCommon(), CompareActivationTestImpl(), CompareBoundedReLuTest(), armnnTfLiteParser::ComputeWrappedIndex(), ConstantLinearActivationTestCommon(), armnn::ConvertActivationDescriptorToAclActivationLayerInfo(), OnnxParserImpl::CreateNetworkFromString(), armnn::CreateStartOfLeakyReluNetwork(), RefActivationWorkload::Execute(), StaticRangeStrategy::ExecuteStrategy(), DynamicQuantizationStrategy::ExecuteStrategy(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), armnn::GetInputTensorInfo(), IDeserializer::DeserializerImpl::GetNetworkOutputBindingInfo(), RefLayerSupport::IsActivationSupported(), ITfParser::TfParserImpl::IsSupportedLeakyReluPattern(), ActivationDescriptor::operator==(), TfLiteParserImpl::OutputShapeOfSqueeze(), ITfParser::TfParserImpl::ParseRelu(), ITfParser::TfParserImpl::ParseRelu6(), ICaffeParser::CaffeParserImpl::ParseReluLayer(), ITfParser::TfParserImpl::ParseSigmoid(), ITfParser::TfParserImpl::ParseSoftplus(), ITfParser::TfParserImpl::ParseTanh(), StringifyLayerParameters< ActivationDescriptor >::Serialize(), SimpleActivationTest(), and SqrtNNTest().

---

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

• include/armnn/Descriptors.hpp