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