20.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 123 of file Descriptors.hpp.
|
inline |
Definition at line 125 of file Descriptors.hpp.
|
inline |
Definition at line 130 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 138 of file Descriptors.hpp.
Referenced by armnn::ComputeSoftmaxAclAxis(), RefSoftmaxWorkload::Execute(), GetSoftmaxProfilerJson(), armnn::LogSoftmax(), LogSoftmaxTest1(), LogSoftmaxTest2(), LogSoftmaxTest3(), LogSoftmaxTest4(), SoftmaxDescriptor::operator==(), CaffeParserBase::ParseSoftmaxLayer(), StringifyLayerParameters< SoftmaxDescriptor >::Serialize(), and SerializerVisitor::VisitLogSoftmaxLayer().
float m_Beta |
Exponentiation value.
Definition at line 136 of file Descriptors.hpp.
Referenced by armnn::BOOST_AUTO_TEST_CASE(), BOOST_AUTO_TEST_CASE(), ClSoftmaxUint8Workload::ClSoftmaxUint8Workload(), TfLiteParser::CreateNetworkFromBinary(), RefSoftmaxWorkload::Execute(), Deserializer::GetNetworkOutputBindingInfo(), armnn::LogSoftmax(), LogSoftmaxTest1(), LogSoftmaxTest2(), LogSoftmaxTest3(), LogSoftmaxTest4(), NeonSoftmaxUint8Workload::NeonSoftmaxUint8Workload(), SoftmaxDescriptor::operator==(), Deserializer::OutputShapeOfReshape(), StringifyLayerParameters< SoftmaxDescriptor >::Serialize(), SerializerVisitor::VisitLogSoftmaxLayer(), and SerializerVisitor::VisitSoftmaxLayer().