ArmNN
NotReleased
|
An ActivationDescriptor for the ActivationLayer. More...
#include <Descriptors.hpp>
Public Member Functions | |
ActivationDescriptor () | |
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). More... | |
float | m_A |
Alpha upper bound value used by the activation functions. (BoundedReLu, Linear, TanH). 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 20 of file Descriptors.hpp.
|
inline |
Definition at line 22 of file Descriptors.hpp.
|
inline |
Definition at line 28 of file Descriptors.hpp.
References ActivationDescriptor::m_A, ActivationDescriptor::m_B, and ActivationDescriptor::m_Function.
float m_A |
Alpha upper bound value used by the activation functions. (BoundedReLu, Linear, TanH).
Definition at line 37 of file Descriptors.hpp.
Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), BoundedReLuTestCommon(), CompareActivationTestImpl(), CompareBoundedReLuTest(), ConstantLinearActivationTestCommon(), armnn::ConvertActivationDescriptorToAclActivationLayerInfo(), armnn::CreateStartOfLeakyReluNetwork(), RefActivationWorkload::Execute(), Deserializer::GetNetworkOutputBindingInfo(), ActivationDescriptor::operator==(), armnnTfParser::OutputShapeOfExpandDims(), TfLiteParser::OutputShapeOfReshape(), TfLiteParser::OutputShapeOfSqueeze(), armnnTfParser::OutputShapeOfSqueeze(), CaffeParserBase::ParseReluLayer(), StringifyLayerParameters< ActivationDescriptor >::Serialize(), SimpleActivationTest(), DynamicQuantizationVisitor::VisitActivationLayer(), StaticRangeVisitor::VisitActivationLayer(), and SerializerVisitor::VisitActivationLayer().
float m_B |
Beta lower bound value used by the activation functions. (BoundedReLu, Linear, TanH).
Definition at line 39 of file Descriptors.hpp.
Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), BoundedReLuTestCommon(), CompareActivationTestImpl(), CompareBoundedReLuTest(), ConstantLinearActivationTestCommon(), armnn::ConvertActivationDescriptorToAclActivationLayerInfo(), armnn::CreateStartOfLeakyReluNetwork(), RefActivationWorkload::Execute(), Deserializer::GetNetworkOutputBindingInfo(), ActivationDescriptor::operator==(), TfLiteParser::OutputShapeOfReshape(), TfLiteParser::OutputShapeOfSqueeze(), armnnTfParser::OutputShapeOfSqueeze(), StringifyLayerParameters< ActivationDescriptor >::Serialize(), SimpleActivationTest(), and SerializerVisitor::VisitActivationLayer().
ActivationFunction m_Function |
The activation function to use (Sigmoid, TanH, Linear, ReLu, BoundedReLu, SoftReLu, LeakyReLu, Abs, Sqrt, Square).
Definition at line 35 of file Descriptors.hpp.
Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), BoundedReLuTestCommon(), CompareActivationTestImpl(), CompareBoundedReLuTest(), ConstantLinearActivationTestCommon(), armnn::ConvertActivationDescriptorToAclActivationLayerInfo(), OnnxParser::CreateNetworkFromString(), armnn::CreateStartOfLeakyReluNetwork(), RefActivationWorkload::Execute(), Deserializer::GetNetworkOutputBindingInfo(), RefLayerSupport::IsActivationSupported(), ActivationDescriptor::operator==(), ArgMinMaxDescriptor::operator==(), armnnTfParser::OutputShapeOfExpandDims(), TfLiteParser::OutputShapeOfReshape(), TfLiteParser::OutputShapeOfSqueeze(), armnnTfParser::OutputShapeOfSqueeze(), CaffeParserBase::ParseReluLayer(), StringifyLayerParameters< ActivationDescriptor >::Serialize(), SimpleActivationTest(), SqrtNNTest(), StaticRangeVisitor::VisitActivationLayer(), DynamicQuantizationVisitor::VisitActivationLayer(), and SerializerVisitor::VisitActivationLayer().