21.08
|
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... | |
An ActivationDescriptor for the ActivationLayer.
Definition at line 25 of file Descriptors.hpp.
|
inline |
Definition at line 27 of file Descriptors.hpp.
|
inline |
Definition at line 33 of file Descriptors.hpp.
|
inline |
Definition at line 41 of file Descriptors.hpp.
References ActivationDescriptor::m_B, and ActivationDescriptor::m_Function.
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 BoundedReLuTestCommon(), CompareActivationTestImpl(), CompareBoundedReLuTest(), armnnTfLiteParser::ComputeWrappedIndex(), ConstantLinearActivationTestCommon(), armnn::ConvertActivationDescriptorToAclActivationLayerInfo(), OnnxParserImpl::CreateNetworkFromString(), RefActivationWorkload::ExecuteAsync(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetNetworkOutputBindingInfo(), TfLiteParserImpl::OutputShapeOfSqueeze(), StringifyLayerParameters< ActivationDescriptor >::Serialize(), SimpleActivationTest(), and TEST_SUITE().
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 BoundedReLuTestCommon(), CompareActivationTestImpl(), CompareBoundedReLuTest(), armnnTfLiteParser::ComputeWrappedIndex(), ConstantLinearActivationTestCommon(), armnn::ConvertActivationDescriptorToAclActivationLayerInfo(), OnnxParserImpl::CreateNetworkFromString(), RefActivationWorkload::ExecuteAsync(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetNetworkOutputBindingInfo(), ActivationDescriptor::operator==(), TfLiteParserImpl::OutputShapeOfSqueeze(), StringifyLayerParameters< ActivationDescriptor >::Serialize(), SimpleActivationTest(), and TEST_SUITE().
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 BoundedReLuTestCommon(), CompareActivationTestImpl(), CompareBoundedReLuTest(), armnnTfLiteParser::ComputeWrappedIndex(), ConstantLinearActivationTestCommon(), armnn::ConvertActivationDescriptorToAclActivationLayerInfo(), OnnxParserImpl::CreateNetworkFromString(), RefActivationWorkload::ExecuteAsync(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetNetworkOutputBindingInfo(), RefLayerSupport::IsActivationSupported(), ActivationDescriptor::operator==(), TfLiteParserImpl::OutputShapeOfSqueeze(), StringifyLayerParameters< ActivationDescriptor >::Serialize(), SimpleActivationTest(), SqrtNNTest(), and TEST_SUITE().