20.02
|
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... | |
An ArgMinMaxDescriptor for ArgMinMaxLayer.
Definition at line 43 of file Descriptors.hpp.
|
inline |
Definition at line 45 of file Descriptors.hpp.
|
inline |
Definition at line 50 of file Descriptors.hpp.
References ArgMinMaxDescriptor::m_Axis, ActivationDescriptor::m_Function, and ArgMinMaxDescriptor::m_Function.
int m_Axis |
Axis to reduce across the input tensor.
Definition at line 58 of file Descriptors.hpp.
Referenced by ArgMinMaxInferOutputShape1dTest(), ArgMinMaxInferOutputShape2dTest(), ArgMinMaxInferOutputShape3dTest(), ArgMinMaxInferOutputShape4dTest(), ClArgMinMaxWorkload::ClArgMinMaxWorkload(), 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 56 of file Descriptors.hpp.
Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), ClArgMinMaxWorkload::ClArgMinMaxWorkload(), RefArgMinMaxWorkload::Execute(), Deserializer::GetNetworkOutputBindingInfo(), NeonArgMinMaxWorkload::NeonArgMinMaxWorkload(), ArgMinMaxDescriptor::operator==(), and SerializerVisitor::VisitArgMinMaxLayer().