ArmNN
 24.02
MeanDescriptor Struct Reference

A MeanDescriptor for the MeanLayer. More...

#include <Descriptors.hpp>

Inheritance diagram for MeanDescriptor:
[legend]
Collaboration diagram for MeanDescriptor:
[legend]

Public Member Functions

 MeanDescriptor ()
 
 MeanDescriptor (const std::vector< unsigned int > &axis, bool keepDims)
 
bool operator== (const MeanDescriptor &rhs) const
 
- Public Member Functions inherited from BaseDescriptor
virtual bool IsNull () const
 
virtual ~BaseDescriptor ()=default
 

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 1172 of file Descriptors.hpp.

Constructor & Destructor Documentation

◆ MeanDescriptor() [1/2]

MeanDescriptor ( )
inline

Definition at line 1174 of file Descriptors.hpp.

1175  : m_Axis()
1176  , m_KeepDims(false)
1177  {}

◆ MeanDescriptor() [2/2]

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

Definition at line 1179 of file Descriptors.hpp.

1180  : m_Axis(axis)
1181  , m_KeepDims(keepDims)
1182  {}

Member Function Documentation

◆ operator==()

bool operator== ( const MeanDescriptor rhs) const
inline

Definition at line 1184 of file Descriptors.hpp.

1185  {
1186  return m_Axis == rhs.m_Axis && m_KeepDims == rhs.m_KeepDims;
1187  }

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

Member Data Documentation

◆ m_Axis

◆ 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 1192 of file Descriptors.hpp.

Referenced by MeanLayer::Clone(), MeanLayer::CreateWorkload(), MeanLayer::InferOutputShapes(), RefLayerSupport::IsMeanSupported(), MeanDescriptor::operator==(), StringifyLayerParameters< MeanDescriptor >::Serialize(), and MeanQueueDescriptor::Validate().


The documentation for this struct was generated from the following file:
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:1192
armnn::MeanDescriptor::m_Axis
std::vector< unsigned int > m_Axis
Values for the dimensions to reduce.
Definition: Descriptors.hpp:1190