20.11
|
An ArgMinMaxDescriptor for ArgMinMaxLayer. More...
#include <Descriptors.hpp>
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 |
An ArgMinMaxDescriptor for ArgMinMaxLayer.
Definition at line 51 of file Descriptors.hpp.
|
inline |
Definition at line 53 of file Descriptors.hpp.
|
inline |
Definition at line 59 of file Descriptors.hpp.
References ArgMinMaxDescriptor::m_Axis, ActivationDescriptor::m_Function, ArgMinMaxDescriptor::m_Function, and ArgMinMaxDescriptor::m_Output_Type.
int m_Axis |
Axis to reduce across the input tensor.
Definition at line 67 of file Descriptors.hpp.
Referenced by ArgMinMaxInferOutputShape1dTest(), ArgMinMaxInferOutputShape2dTest(), ArgMinMaxInferOutputShape3dTest(), ArgMinMaxInferOutputShape4dTest(), ClArgMinMaxWorkload::ClArgMinMaxWorkload(), armnnTfLiteParser::ComputeWrappedIndex(), RefArgMinMaxWorkload::Execute(), Deserializer::GetNetworkOutputBindingInfo(), ArgMinMaxLayer::InferOutputShapes(), NeonArgMinMaxWorkload::NeonArgMinMaxWorkload(), ArgMinMaxDescriptor::operator==(), and SerializerVisitor::VisitArgMinMaxLayer().
ArgMinMaxFunction m_Function |
Specify if the function is to find Min or Max.
Definition at line 65 of file Descriptors.hpp.
Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), ClArgMinMaxWorkload::ClArgMinMaxWorkload(), armnnTfLiteParser::ComputeWrappedIndex(), RefArgMinMaxWorkload::Execute(), Deserializer::GetNetworkOutputBindingInfo(), NeonArgMinMaxWorkload::NeonArgMinMaxWorkload(), ArgMinMaxDescriptor::operator==(), and SerializerVisitor::VisitArgMinMaxLayer().
armnn::DataType m_Output_Type |
Definition at line 69 of file Descriptors.hpp.
Referenced by armnnTfLiteParser::ComputeWrappedIndex(), RefArgMinMaxWorkload::Execute(), and ArgMinMaxDescriptor::operator==().