ArmNN
 21.02
Public Member Functions | Public Attributes | List of all members
MeanDescriptor Struct Reference

A MeanDescriptor for the MeanLayer. More...

#include <Descriptors.hpp>

Inheritance diagram for MeanDescriptor:
BaseDescriptor

Public Member Functions

 MeanDescriptor ()
 
 MeanDescriptor (const std::vector< unsigned int > &axis, bool keepDims)
 
bool operator== (const MeanDescriptor &rhs) const
 

Public Attributes

std::vector< unsigned int > m_Axis
 Values for the dimensions to reduce. More...
 
bool m_KeepDims
 Enable/disable keep dimensions. If true, then the reduced dimensions that are of length 1 are kept. More...
 

Detailed Description

A MeanDescriptor for the MeanLayer.

Definition at line 951 of file Descriptors.hpp.

Constructor & Destructor Documentation

◆ MeanDescriptor() [1/2]

MeanDescriptor ( )
inline

Definition at line 953 of file Descriptors.hpp.

954  : m_Axis()
955  , m_KeepDims(false)
956  {}
armnn::MeanDescriptor::m_Axis
std::vector< unsigned int > m_Axis
Values for the dimensions to reduce.
Definition: Descriptors.hpp:969
armnn::MeanDescriptor::m_KeepDims
bool m_KeepDims
Enable/disable keep dimensions. If true, then the reduced dimensions that are of length 1 are kept...
Definition: Descriptors.hpp:971

◆ MeanDescriptor() [2/2]

MeanDescriptor ( const std::vector< unsigned int > &  axis,
bool  keepDims 
)
inline

Definition at line 958 of file Descriptors.hpp.

959  : m_Axis(axis)
960  , m_KeepDims(keepDims)
961  {}
armnn::MeanDescriptor::m_Axis
std::vector< unsigned int > m_Axis
Values for the dimensions to reduce.
Definition: Descriptors.hpp:969
armnn::MeanDescriptor::m_KeepDims
bool m_KeepDims
Enable/disable keep dimensions. If true, then the reduced dimensions that are of length 1 are kept...
Definition: Descriptors.hpp:971

Member Function Documentation

◆ operator==()

bool operator== ( const MeanDescriptor &  rhs) const
inline

Definition at line 963 of file Descriptors.hpp.

References MeanDescriptor::m_Axis, and MeanDescriptor::m_KeepDims.

964  {
965  return m_Axis == rhs.m_Axis && m_KeepDims == rhs.m_KeepDims;
966  }
armnn::MeanDescriptor::m_Axis
std::vector< unsigned int > m_Axis
Values for the dimensions to reduce.
Definition: Descriptors.hpp:969
armnn::MeanDescriptor::m_KeepDims
bool m_KeepDims
Enable/disable keep dimensions. If true, then the reduced dimensions that are of length 1 are kept...
Definition: Descriptors.hpp:971

Member Data Documentation

◆ m_Axis

std::vector<unsigned int> m_Axis

Values for the dimensions to reduce.

Definition at line 969 of file Descriptors.hpp.

Referenced by BOOST_AUTO_TEST_CASE(), MeanLayer::Clone(), MeanLayer::CreateWorkload(), RefMeanWorkload::Execute(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetNormalizationDescriptor(), RefLayerSupport::IsMeanSupported(), MeanDescriptor::operator==(), TfLiteParserImpl::OutputShapeOfSqueeze(), ITfParser::TfParserImpl::ParseMean(), StringifyLayerParameters< MeanDescriptor >::Serialize(), and MeanLayer::ValidateTensorShapesFromInputs().

◆ m_KeepDims

bool m_KeepDims

Enable/disable keep dimensions. If true, then the reduced dimensions that are of length 1 are kept.

Definition at line 971 of file Descriptors.hpp.

Referenced by MeanLayer::CreateWorkload(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetNormalizationDescriptor(), RefLayerSupport::IsMeanSupported(), MeanDescriptor::operator==(), TfLiteParserImpl::OutputShapeOfSqueeze(), ITfParser::TfParserImpl::ParseMean(), StringifyLayerParameters< MeanDescriptor >::Serialize(), and MeanLayer::ValidateTensorShapesFromInputs().

The documentation for this struct was generated from the following file: