21 unsigned int CalcAclAxis(
unsigned int numDimensions,
unsigned int axisIndex)
23 return (numDimensions - axisIndex) - 1;
35 const arm_compute::TensorInfo aclInput = armcomputetensorutils::BuildArmComputeTensorInfo(input);
36 const arm_compute::TensorInfo aclOutput = armcomputetensorutils::BuildArmComputeTensorInfo(output);
44 return arm_compute::CLArgMinMaxLayer::validate(&aclInput, aclAxis, &aclOutput,
45 arm_compute::ReductionOperation::ARG_IDX_MAX);
49 return arm_compute::CLArgMinMaxLayer::validate(&aclInput, aclAxis, &aclOutput,
50 arm_compute::ReductionOperation::ARG_IDX_MIN);
57 const arm_compute::CLCompileContext& clCompileContext)
77 m_ArgMinMaxLayer.configure(&input, aclAxis, &output, arm_compute::ReductionOperation::ARG_IDX_MAX);
81 m_ArgMinMaxLayer.configure(clCompileContext,
85 arm_compute::ReductionOperation::ARG_IDX_MIN);
virtual void Execute() const override
#define ARMNN_SCOPED_PROFILING_EVENT_CL_GUID(name, guid)
void RunClFunction(arm_compute::IFunction &function, const CheckLocation &location)
ArgMinMaxFunction m_Function
Specify if the function is to find Min or Max.
ClArgMinMaxWorkload(const ArgMinMaxQueueDescriptor &descriptor, const WorkloadInfo &info, const arm_compute::CLCompileContext &clCompileContext)
Copyright (c) 2021 ARM Limited and Contributors.
LayerDescriptor m_Parameters
#define ARMNN_SCOPED_PROFILING_EVENT(backendId, name)
std::vector< TensorInfo > m_InputTensorInfos
An ArgMinMaxDescriptor for ArgMinMaxLayer.
ArgMinMaxQueueDescriptor m_Data
unsigned int GetUnsignedAxis(const unsigned int inputDimension, const int axis)
profiling::ProfilingGuid GetGuid() const final
std::vector< ITensorHandle * > m_Outputs
#define ARMNN_REPORT_PROFILING_WORKLOAD_DESC(name, desc, infos, guid)
std::enable_if_t< std::is_unsigned< Source >::value &&std::is_unsigned< Dest >::value, Dest > numeric_cast(Source source)
arm_compute::Status ClArgMinMaxWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const ArgMinMaxDescriptor &descriptor)
Contains information about TensorInfos of a layer.
int m_Axis
Axis to reduce across the input tensor.
std::vector< ITensorHandle * > m_Inputs
unsigned int GetNumDimensions() const