ArmNN
 25.11
Loading...
Searching...
No Matches
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.
bool m_KeepDims
 Enable/disable keep dimensions. If true, then the reduced dimensions that are of length 1 are kept.

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 {}

References m_Axis, and m_KeepDims.

Referenced by operator==().

◆ 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 {}

References m_Axis, and m_KeepDims.

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 m_Axis, m_KeepDims, and MeanDescriptor().

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 armnn::ClMeanValidate(), MeanLayer::CreateWorkload(), RefLayerSupport::IsMeanSupported(), MeanDescriptor(), MeanDescriptor(), armnn::NeonMeanWorkloadValidate(), operator==(), and StringifyLayerParameters< MeanDescriptor >::Serialize().


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