ArmNN
 24.02
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. 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 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  {}

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 SoftmaxDescriptor::m_Axis, and SoftmaxDescriptor::m_Beta.

Member Data Documentation

◆ m_Axis

◆ m_Beta


The documentation for this struct was generated from the following file:
armnn::SoftmaxDescriptor::m_Beta
float m_Beta
Exponentiation value.
Definition: Descriptors.hpp:190
armnn::SoftmaxDescriptor::m_Axis
int m_Axis
Scalar, defaulted to the last index (-1), specifying the dimension the activation will be performed o...
Definition: Descriptors.hpp:192