ArmNN
 25.11
Loading...
Searching...
No Matches
SoftmaxDescriptor Struct Reference

A SoftmaxDescriptor for the SoftmaxLayer. More...

#include <Descriptors.hpp>

Inheritance diagram for SoftmaxDescriptor:
[legend]
Collaboration diagram for SoftmaxDescriptor:
[legend]

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.
int m_Axis
 Scalar, defaulted to the last index (-1), specifying the dimension the activation will be performed on.

Detailed Description

A SoftmaxDescriptor for the SoftmaxLayer.

Definition at line 177 of file Descriptors.hpp.

Constructor & Destructor Documentation

◆ SoftmaxDescriptor()

SoftmaxDescriptor ( )
inline

Definition at line 179 of file Descriptors.hpp.

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

References m_Axis, and m_Beta.

Referenced by operator==().

Member Function Documentation

◆ operator==()

bool operator== ( const SoftmaxDescriptor & rhs) const
inline

Definition at line 184 of file Descriptors.hpp.

185 {
186 return m_Beta == rhs.m_Beta && m_Axis == rhs.m_Axis;
187 }

References m_Axis, m_Beta, and SoftmaxDescriptor().

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

Referenced by armnn::ClLogSoftmaxWorkloadValidate(), armnn::ClSoftmaxWorkloadValidate(), armnn::ComputeSoftmaxAclAxis(), armnn::LogSoftmax(), armnn::NeonLogSoftmaxWorkloadValidate(), armnn::NeonSoftmaxWorkloadValidate(), operator==(), StringifyLayerParameters< SoftmaxDescriptor >::Serialize(), and SoftmaxDescriptor().

◆ m_Beta

float m_Beta

Exponentiation value.

Definition at line 190 of file Descriptors.hpp.

Referenced by armnn::ClLogSoftmaxWorkloadValidate(), armnn::ClSoftmaxWorkloadValidate(), ConvertSoftmaxToTosaOperator(), armnn::LogSoftmax(), armnn::NeonLogSoftmaxWorkloadValidate(), armnn::NeonSoftmaxWorkloadValidate(), operator==(), StringifyLayerParameters< SoftmaxDescriptor >::Serialize(), and SoftmaxDescriptor().

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