20.02
|
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 610 of file Descriptors.hpp.
|
inline |
Definition at line 612 of file Descriptors.hpp.
|
inline |
Definition at line 617 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 625 of file Descriptors.hpp.
Referenced by armnn::BatchNormImpl(), armnn::BOOST_AUTO_TEST_CASE(), TestBatchNormalizationLayerVisitor::CheckDescriptor(), 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 623 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(), StringifyLayerParameters< BatchNormalizationDescriptor >::Serialize(), and SerializerVisitor::VisitBatchNormalizationLayer().