20 unsigned int CalcAclAxis(
unsigned int numDimensions,
unsigned int axisIndex)
22 return (numDimensions - axisIndex) - 1;
34 const arm_compute::TensorInfo aclInput = armcomputetensorutils::BuildArmComputeTensorInfo(input);
35 const arm_compute::TensorInfo aclOutput = armcomputetensorutils::BuildArmComputeTensorInfo(output);
43 return arm_compute::CLArgMinMaxLayer::validate(&aclInput, aclAxis, &aclOutput,
44 arm_compute::ReductionOperation::ARG_IDX_MAX);
48 return arm_compute::CLArgMinMaxLayer::validate(&aclInput, aclAxis, &aclOutput,
49 arm_compute::ReductionOperation::ARG_IDX_MIN);
67 m_ArgMinMaxLayer.configure(&input, aclAxis, &output, arm_compute::ReductionOperation::ARG_IDX_MAX);
71 m_ArgMinMaxLayer.configure(&input, aclAxis, &output, arm_compute::ReductionOperation::ARG_IDX_MIN);
virtual void Execute() const override
#define ARMNN_SCOPED_PROFILING_EVENT_CL(name)
void RunClFunction(arm_compute::IFunction &function, const CheckLocation &location)
const ArgMinMaxQueueDescriptor m_Data
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)
ClArgMinMaxWorkload(const ArgMinMaxQueueDescriptor &descriptor, const WorkloadInfo &info)
std::vector< ITensorHandle * > m_Outputs
arm_compute::Status ClArgMinMaxWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const ArgMinMaxDescriptor &descriptor)
Contains information about inputs and outputs to a layer.
int m_Axis
Axis to reduce across the input tensor.
std::vector< ITensorHandle * > m_Inputs
unsigned int GetNumDimensions() const