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

An ArgMinMaxDescriptor for ArgMinMaxLayer. More...

#include <Descriptors.hpp>

Inheritance diagram for ArgMinMaxDescriptor:
BaseDescriptor

Public Member Functions

 ArgMinMaxDescriptor ()
 
bool operator== (const ArgMinMaxDescriptor &rhs) const
 

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
 

Detailed Description

An ArgMinMaxDescriptor for ArgMinMaxLayer.

Definition at line 56 of file Descriptors.hpp.

Constructor & Destructor Documentation

◆ ArgMinMaxDescriptor()

ArgMinMaxDescriptor ( )
inline

Definition at line 58 of file Descriptors.hpp.

59  : m_Function(ArgMinMaxFunction::Min)
60  , m_Axis(-1)
61  , m_Output_Type(armnn::DataType::Signed32)
62  {}
armnn::ArgMinMaxDescriptor::m_Function
ArgMinMaxFunction m_Function
Specify if the function is to find Min or Max.
Definition: Descriptors.hpp:70
armnn::ArgMinMaxDescriptor::m_Axis
int m_Axis
Axis to reduce across the input tensor.
Definition: Descriptors.hpp:72
armnn::ArgMinMaxDescriptor::m_Output_Type
armnn::DataType m_Output_Type
Definition: Descriptors.hpp:74

Member Function Documentation

◆ operator==()

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

Definition at line 64 of file Descriptors.hpp.

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

65  {
66  return m_Function == rhs.m_Function && m_Axis == rhs.m_Axis && m_Output_Type == rhs.m_Output_Type;
67  }
armnn::ArgMinMaxDescriptor::m_Function
ArgMinMaxFunction m_Function
Specify if the function is to find Min or Max.
Definition: Descriptors.hpp:70
armnn::ArgMinMaxDescriptor::m_Axis
int m_Axis
Axis to reduce across the input tensor.
Definition: Descriptors.hpp:72
armnn::ArgMinMaxDescriptor::m_Output_Type
armnn::DataType m_Output_Type
Definition: Descriptors.hpp:74

Member Data Documentation

◆ m_Axis

int m_Axis

Axis to reduce across the input tensor.

Definition at line 72 of file Descriptors.hpp.

Referenced by ArgMinMaxInferOutputShape1dTest(), ArgMinMaxInferOutputShape2dTest(), ArgMinMaxInferOutputShape3dTest(), ArgMinMaxInferOutputShape4dTest(), ClArgMinMaxWorkload::ClArgMinMaxWorkload(), armnnTfLiteParser::ComputeWrappedIndex(), RefArgMinMaxWorkload::Execute(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetNetworkOutputBindingInfo(), ArgMinMaxLayer::InferOutputShapes(), NeonArgMinMaxWorkload::NeonArgMinMaxWorkload(), ArgMinMaxDescriptor::operator==(), and ICaffeParser::CaffeParserImpl::ParseArgmaxLayer().

◆ m_Function

ArgMinMaxFunction m_Function

Specify if the function is to find Min or Max.

Definition at line 70 of file Descriptors.hpp.

Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), ClArgMinMaxWorkload::ClArgMinMaxWorkload(), armnnTfLiteParser::ComputeWrappedIndex(), RefArgMinMaxWorkload::Execute(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), armnn::GetInputTensorInfo(), IDeserializer::DeserializerImpl::GetNetworkOutputBindingInfo(), NeonArgMinMaxWorkload::NeonArgMinMaxWorkload(), ArgMinMaxDescriptor::operator==(), and ICaffeParser::CaffeParserImpl::ParseArgmaxLayer().

◆ m_Output_Type

armnn::DataType m_Output_Type

Definition at line 74 of file Descriptors.hpp.

Referenced by armnnTfLiteParser::ComputeWrappedIndex(), RefArgMinMaxWorkload::Execute(), ArgMinMaxDescriptor::operator==(), and ICaffeParser::CaffeParserImpl::ParseArgmaxLayer().

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