21.02
|
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::Execute(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), GetSoftmaxProfilerJson(), armnn::LogSoftmax(), LogSoftmaxTest1(), LogSoftmaxTest2(), LogSoftmaxTest3(), LogSoftmaxTest4(), NeonLogSoftmaxWorkload::NeonLogSoftmaxWorkload(), NeonSoftmaxWorkload::NeonSoftmaxWorkload(), SoftmaxDescriptor::operator==(), ICaffeParser::CaffeParserImpl::ParseSoftmaxLayer(), and StringifyLayerParameters< SoftmaxDescriptor >::Serialize().
float m_Beta |
Exponentiation value.
Definition at line 152 of file Descriptors.hpp.
Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), ClLogSoftmaxWorkload::ClLogSoftmaxWorkload(), ClSoftmaxWorkload::ClSoftmaxWorkload(), TfLiteParserImpl::CreateNetworkFromBinary(), RefSoftmaxWorkload::Execute(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetNetworkOutputBindingInfo(), armnn::LogSoftmax(), LogSoftmaxTest1(), LogSoftmaxTest2(), LogSoftmaxTest3(), LogSoftmaxTest4(), NeonLogSoftmaxWorkload::NeonLogSoftmaxWorkload(), NeonSoftmaxWorkload::NeonSoftmaxWorkload(), SoftmaxDescriptor::operator==(), IDeserializer::DeserializerImpl::OutputShapeOfReshape(), and StringifyLayerParameters< SoftmaxDescriptor >::Serialize().