15 #include <arm_compute/runtime/NEON/functions/NEArgMinMaxLayer.h> 19 unsigned int CalcAclAxis(
unsigned int numDimensions,
unsigned int axisIndex)
21 return (numDimensions - axisIndex) - 1;
33 const arm_compute::TensorInfo aclInput = armcomputetensorutils::BuildArmComputeTensorInfo(input);
34 const arm_compute::TensorInfo aclOutput = armcomputetensorutils::BuildArmComputeTensorInfo(output);
42 return arm_compute::NEArgMinMaxLayer::validate(&aclInput, aclAxis, &aclOutput,
43 arm_compute::ReductionOperation::ARG_IDX_MAX);
47 return arm_compute::NEArgMinMaxLayer::validate(&aclInput, aclAxis, &aclOutput,
48 arm_compute::ReductionOperation::ARG_IDX_MIN);
57 arm_compute::ITensor& input = boost::polymorphic_downcast<IAclTensorHandle*>(
m_Data.
m_Inputs[0])->GetTensor();
58 arm_compute::ITensor& output = boost::polymorphic_downcast<IAclTensorHandle*>(
m_Data.
m_Outputs[0])->GetTensor();
64 auto layer = std::make_unique<arm_compute::NEArgMinMaxLayer>();
68 layer->configure(&input, aclAxis, &output, arm_compute::ReductionOperation::ARG_IDX_MAX);
72 layer->configure(&input, aclAxis, &output, arm_compute::ReductionOperation::ARG_IDX_MIN);
75 m_ArgMinMaxLayer.reset(layer.release());
81 m_ArgMinMaxLayer->run();
const ArgMinMaxQueueDescriptor m_Data
#define ARMNN_SCOPED_PROFILING_EVENT_NEON(name)
ArgMinMaxFunction m_Function
Specify if the function is to find Min or Max.
Copyright (c) 2020 ARM Limited.
LayerDescriptor m_Parameters
std::vector< TensorInfo > m_InputTensorInfos
An ArgMinMaxDescriptor for ArgMinMaxLayer.
unsigned int GetUnsignedAxis(const unsigned int inputDimension, const int axis)
std::enable_if_t< std::is_unsigned< Source >::value &&std::is_unsigned< Dest >::value, Dest > numeric_cast(Source source)
arm_compute::Status NeonArgMinMaxWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const ArgMinMaxDescriptor &descriptor)
std::vector< ITensorHandle * > m_Outputs
Contains information about inputs and outputs to a layer.
int m_Axis
Axis to reduce across the input tensor.
std::vector< ITensorHandle * > m_Inputs
NeonArgMinMaxWorkload(const ArgMinMaxQueueDescriptor &descriptor, const WorkloadInfo &info)
virtual void Execute() const override
unsigned int GetNumDimensions() const