20.11
|
A BatchNormalizationDescriptor for the BatchNormalizationLayer. More...
#include <Descriptors.hpp>
Public Member Functions | |
BatchNormalizationDescriptor () | |
bool | operator== (const BatchNormalizationDescriptor &rhs) const |
Public Attributes | |
float | m_Eps |
Value to add to the variance. Used to avoid dividing by zero. More... | |
DataLayout | m_DataLayout |
The data layout to be used (NCHW, NHWC). More... | |
A BatchNormalizationDescriptor for the BatchNormalizationLayer.
Definition at line 621 of file Descriptors.hpp.
|
inline |
Definition at line 623 of file Descriptors.hpp.
|
inline |
Definition at line 628 of file Descriptors.hpp.
References BatchNormalizationDescriptor::m_DataLayout, and BatchNormalizationDescriptor::m_Eps.
DataLayout m_DataLayout |
The data layout to be used (NCHW, NHWC).
Definition at line 636 of file Descriptors.hpp.
Referenced by armnn::BatchNormImpl(), armnn::BOOST_AUTO_TEST_CASE(), BOOST_AUTO_TEST_CASE(), TestBatchNormalizationLayerVisitor::CheckDescriptor(), CreatNetwork(), Deserializer::GetNetworkOutputBindingInfo(), BatchNormalizationDescriptor::operator==(), armnnTfParser::OutputShapeOfExpandDims(), StringifyLayerParameters< BatchNormalizationDescriptor >::Serialize(), and SerializerVisitor::VisitBatchNormalizationLayer().
float m_Eps |
Value to add to the variance. Used to avoid dividing by zero.
Definition at line 634 of file Descriptors.hpp.
Referenced by armnn::BatchNormImpl(), BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), TestBatchNormalizationLayerVisitor::CheckDescriptor(), CompareBatchNormTest(), OnnxParser::CreateNetworkFromString(), Deserializer::GetNetworkOutputBindingInfo(), BatchNormalizationDescriptor::operator==(), armnnTfParser::OutputShapeOfExpandDims(), CaffeParserBase::ParseBatchNormLayer(), CaffeParserBase::ParseScaleLayer(), FuseBatchNorm< ConvLayer, ArmnnType, T >::Run(), StringifyLayerParameters< BatchNormalizationDescriptor >::Serialize(), and SerializerVisitor::VisitBatchNormalizationLayer().