ArmNN
 21.11
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 1037 of file Descriptors.hpp.

Constructor & Destructor Documentation

◆ MeanDescriptor() [1/2]

MeanDescriptor ( )
inline

Definition at line 1039 of file Descriptors.hpp.

1040  : m_Axis()
1041  , m_KeepDims(false)
1042  {}
armnn::MeanDescriptor::m_Axis
std::vector< unsigned int > m_Axis
Values for the dimensions to reduce.
Definition: Descriptors.hpp:1055
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:1057

◆ MeanDescriptor() [2/2]

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

Definition at line 1044 of file Descriptors.hpp.

1045  : m_Axis(axis)
1046  , m_KeepDims(keepDims)
1047  {}
armnn::MeanDescriptor::m_Axis
std::vector< unsigned int > m_Axis
Values for the dimensions to reduce.
Definition: Descriptors.hpp:1055
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:1057

Member Function Documentation

◆ operator==()

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

Definition at line 1049 of file Descriptors.hpp.

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

1050  {
1051  return m_Axis == rhs.m_Axis && m_KeepDims == rhs.m_KeepDims;
1052  }
armnn::MeanDescriptor::m_Axis
std::vector< unsigned int > m_Axis
Values for the dimensions to reduce.
Definition: Descriptors.hpp:1055
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:1057

Member Data Documentation

◆ m_Axis

std::vector<unsigned int> m_Axis

Values for the dimensions to reduce.

Definition at line 1055 of file Descriptors.hpp.

Referenced by MeanLayer::Clone(), MeanLayer::CreateWorkload(), RefMeanWorkload::ExecuteAsync(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetNormalizationDescriptor(), RefLayerSupport::IsMeanSupported(), MeanDescriptor::operator==(), TfLiteParserImpl::OutputShapeOfSqueeze(), StringifyLayerParameters< MeanDescriptor >::Serialize(), TEST_SUITE(), 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 1057 of file Descriptors.hpp.

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

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