21.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 626 of file Descriptors.hpp.
|
inline |
Definition at line 628 of file Descriptors.hpp.
|
inline |
Definition at line 633 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 641 of file Descriptors.hpp.
Referenced by armnn::BatchNormImpl(), armnn::BOOST_AUTO_TEST_CASE(), BOOST_AUTO_TEST_CASE(), TestBatchNormalizationLayerVisitor::CheckDescriptor(), CreatNetwork(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetNetworkOutputBindingInfo(), armnn::GetVector(), BatchNormalizationDescriptor::operator==(), ITfParser::TfParserImpl::ParseFusedBatchNorm(), and StringifyLayerParameters< BatchNormalizationDescriptor >::Serialize().
float m_Eps |
Value to add to the variance. Used to avoid dividing by zero.
Definition at line 639 of file Descriptors.hpp.
Referenced by armnn::BatchNormImpl(), BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), TestBatchNormalizationLayerVisitor::CheckDescriptor(), CompareBatchNormTest(), OnnxParserImpl::CreateNetworkFromString(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetNetworkOutputBindingInfo(), BatchNormalizationDescriptor::operator==(), ICaffeParser::CaffeParserImpl::ParseBatchNormLayer(), ITfParser::TfParserImpl::ParseFusedBatchNorm(), ICaffeParser::CaffeParserImpl::ParseScaleLayer(), FuseBatchNorm< ConvLayer, ArmnnType, T >::Run(), and StringifyLayerParameters< BatchNormalizationDescriptor >::Serialize().