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);
39 int aclAxis = boost::numeric_cast<
int>(CalcAclAxis(numDims, unsignedAxis));
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);
63 int aclAxis = boost::numeric_cast<
int>(CalcAclAxis(numDims, unsignedAxis));
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);
void RunClFunction(arm_compute::IFunction &function, const CheckLocation &location)
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
arm_compute::Status ClArgMinMaxWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const ArgMinMaxDescriptor &descriptor)
virtual void Execute() const override
std::vector< TensorInfo > m_InputTensorInfos
ClArgMinMaxWorkload(const ArgMinMaxQueueDescriptor &descriptor, const WorkloadInfo &info)
std::vector< ITensorHandle * > m_Outputs
std::vector< ITensorHandle * > m_Inputs
An ArgMinMaxDescriptor for ArgMinMaxLayer.
#define ARMNN_SCOPED_PROFILING_EVENT_CL(name)