ArmNN
 21.11
Public Member Functions | Public Attributes | List of all members
SoftmaxDescriptor Struct Reference

A SoftmaxDescriptor for the SoftmaxLayer. More...

#include <Descriptors.hpp>

Inheritance diagram for SoftmaxDescriptor:
BaseDescriptor

Public Member Functions

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

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

Constructor & Destructor Documentation

◆ SoftmaxDescriptor()

SoftmaxDescriptor ( )
inline

Definition at line 141 of file Descriptors.hpp.

142  : m_Beta(1.0f)
143  , m_Axis(-1)
144  {}
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:154
armnn::SoftmaxDescriptor::m_Beta
float m_Beta
Exponentiation value.
Definition: Descriptors.hpp:152

Member Function Documentation

◆ operator==()

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

Definition at line 146 of file Descriptors.hpp.

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

147  {
148  return m_Beta == rhs.m_Beta && m_Axis == rhs.m_Axis;
149  }
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:154
armnn::SoftmaxDescriptor::m_Beta
float m_Beta
Exponentiation value.
Definition: Descriptors.hpp:152

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

Referenced by ClLogSoftmaxWorkload::ClLogSoftmaxWorkload(), ClSoftmaxWorkload::ClSoftmaxWorkload(), RefSoftmaxWorkload::ExecuteAsync(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetNetworkOutputBindingInfo(), TfLiteParserImpl::LoadModel(), 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: