21.08
|
A SoftmaxDescriptor for the SoftmaxLayer. More...
#include <Descriptors.hpp>
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... | |
A SoftmaxDescriptor for the SoftmaxLayer.
Definition at line 139 of file Descriptors.hpp.
|
inline |
Definition at line 141 of file Descriptors.hpp.
|
inline |
Definition at line 146 of file Descriptors.hpp.
References SoftmaxDescriptor::m_Axis, and SoftmaxDescriptor::m_Beta.
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().
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().