SoftmaxDescriptor Struct Reference

A SoftmaxDescriptor for the SoftmaxLayer. More...

#include <Descriptors.hpp>

Inheritance diagram for SoftmaxDescriptor:

Public Member Functions
	SoftmaxDescriptor ()
bool	operator== (const SoftmaxDescriptor &rhs) const
Public Member Functions inherited from BaseDescriptor
virtual bool	IsNull () const
virtual	~BaseDescriptor ()=default

Public Attributes
float	m_Beta
	Exponentiation value. More...
int	m_Axis
	Scalar, defaulted to the last index (-1), specifying the dimension the activation will be performed on. More...

Detailed Description

A SoftmaxDescriptor for the SoftmaxLayer.

Definition at line 150 of file Descriptors.hpp.

Constructor & Destructor Documentation

SoftmaxDescriptor()

SoftmaxDescriptor ( )

inline

Definition at line 152 of file Descriptors.hpp.

        : m_Beta(1.0f)
        , m_Axis(-1)
    {}

Member Function Documentation

operator==()

bool operator== ( const SoftmaxDescriptor &  rhs ) const

inline

Definition at line 157 of file Descriptors.hpp.

References SoftmaxDescriptor::m_Axis, and SoftmaxDescriptor::m_Beta.

    {
        return m_Beta == rhs.m_Beta && m_Axis == rhs.m_Axis;
    }

Member Data Documentation

m_Axis

int m_Axis

Scalar, defaulted to the last index (-1), specifying the dimension the activation will be performed on.

Definition at line 165 of file Descriptors.hpp.

Referenced by ClLogSoftmaxWorkload::ClLogSoftmaxWorkload(), ClSoftmaxWorkload::ClSoftmaxWorkload(), armnn::ComputeSoftmaxAclAxis(), RefSoftmaxWorkload::ExecuteAsync(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), GetSoftmaxProfilerJson(), armnn::LogSoftmax(), LogSoftmaxTest1(), LogSoftmaxTest2(), LogSoftmaxTest3(), LogSoftmaxTest4(), NeonLogSoftmaxWorkload::NeonLogSoftmaxWorkload(), NeonSoftmaxWorkload::NeonSoftmaxWorkload(), SoftmaxDescriptor::operator==(), and StringifyLayerParameters< SoftmaxDescriptor >::Serialize().

m_Beta

float m_Beta

Exponentiation value.

Definition at line 163 of file Descriptors.hpp.

Referenced by armnnTfLiteParser::AsFloatArray(), ClLogSoftmaxWorkload::ClLogSoftmaxWorkload(), ClSoftmaxWorkload::ClSoftmaxWorkload(), RefSoftmaxWorkload::ExecuteAsync(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetNetworkOutputBindingInfo(), armnn::LogSoftmax(), LogSoftmaxTest1(), LogSoftmaxTest2(), LogSoftmaxTest3(), LogSoftmaxTest4(), NeonLogSoftmaxWorkload::NeonLogSoftmaxWorkload(), NeonSoftmaxWorkload::NeonSoftmaxWorkload(), SoftmaxDescriptor::operator==(), IDeserializer::DeserializerImpl::OutputShapeOfReshape(), StringifyLayerParameters< SoftmaxDescriptor >::Serialize(), and TEST_SUITE().

---

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

• include/armnn/Descriptors.hpp