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);
38 int aclAxis = boost::numeric_cast<
int>(CalcAclAxis(numDims, unsignedAxis));
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();
62 int aclAxis = boost::numeric_cast<
int>(CalcAclAxis(numDims, unsignedAxis));
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();
arm_compute::Status NeonArgMinMaxWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const ArgMinMaxDescriptor &descriptor)
#define ARMNN_SCOPED_PROFILING_EVENT_NEON(name)
unsigned int GetUnsignedAxis(const unsigned int inputDimension, const int axis)
unsigned int GetNumDimensions() const
LayerDescriptor m_Parameters
ArgMinMaxFunction m_Function
Specify if the function is to find Min or Max.
int m_Axis
Axis to reduce across the input tensor.
const ArgMinMaxQueueDescriptor m_Data
std::vector< TensorInfo > m_InputTensorInfos
std::vector< ITensorHandle * > m_Outputs
std::vector< ITensorHandle * > m_Inputs
NeonArgMinMaxWorkload(const ArgMinMaxQueueDescriptor &descriptor, const WorkloadInfo &info)
virtual void Execute() const override
An ArgMinMaxDescriptor for ArgMinMaxLayer.