An ArgMinMaxDescriptor for ArgMinMaxLayer. More...

#include <Descriptors.hpp>

Inheritance diagram for ArgMinMaxDescriptor:

Collaboration diagram for ArgMinMaxDescriptor:

Public Member Functions
	ArgMinMaxDescriptor ()

bool	operator== (const ArgMinMaxDescriptor &rhs) const

Public Member Functions inherited from BaseDescriptor
virtual bool	IsNull () const

virtual	~BaseDescriptor ()=default

Public Attributes
ArgMinMaxFunction	m_Function
	Specify if the function is to find Min or Max. More...

int	m_Axis
	Axis to reduce across the input tensor. More...

armnn::DataType	m_Output_Type
	Deprecated and will be removed in future release. More...

Detailed Description

An ArgMinMaxDescriptor for ArgMinMaxLayer.

Definition at line 67 of file Descriptors.hpp.

Constructor & Destructor Documentation

◆ ArgMinMaxDescriptor()

ArgMinMaxDescriptor ( )

inline

Definition at line 69 of file Descriptors.hpp.

         : m_Function(ArgMinMaxFunction::Min)
         , m_Axis(-1)
         , m_Output_Type(armnn::DataType::Signed32)
     {}

References armnn::Min, and armnn::Signed32.

Member Function Documentation

◆ operator==()

bool operator== ( const ArgMinMaxDescriptor & rhs ) const

inline

Definition at line 75 of file Descriptors.hpp.

     {
         return m_Function == rhs.m_Function && m_Axis == rhs.m_Axis && m_Output_Type == rhs.m_Output_Type;
     }

References ArgMinMaxDescriptor::m_Axis, ArgMinMaxDescriptor::m_Function, and ArgMinMaxDescriptor::m_Output_Type.

Member Data Documentation

◆ m_Axis

int m_Axis

Axis to reduce across the input tensor.

Definition at line 83 of file Descriptors.hpp.

Referenced by ClArgMinMaxWorkload::ClArgMinMaxWorkload(), ArgMinMaxLayer::InferOutputShapes(), NeonArgMinMaxWorkload::NeonArgMinMaxWorkload(), ArgMinMaxDescriptor::operator==(), and ArgMinMaxQueueDescriptor::Validate().

◆ m_Function

ArgMinMaxFunction m_Function

Specify if the function is to find Min or Max.

Definition at line 81 of file Descriptors.hpp.

Referenced by ClArgMinMaxWorkload::ClArgMinMaxWorkload(), NeonArgMinMaxWorkload::NeonArgMinMaxWorkload(), and ArgMinMaxDescriptor::operator==().

◆ m_Output_Type

armnn::DataType m_Output_Type

Deprecated and will be removed in future release.

Definition at line 85 of file Descriptors.hpp.

Referenced by ArgMinMaxDescriptor::operator==().

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

include/armnn/Descriptors.hpp

Public Member Functions

Public Attributes