10 #include <arm_compute/core/Error.h> 11 #include <arm_compute/runtime/IFunction.h> 17 const TensorInfo& output,
18 const ArgMinMaxDescriptor& descriptor);
24 virtual void Execute()
const override;
27 std::unique_ptr<arm_compute::IFunction> m_ArgMinMaxLayer;
arm_compute::Status NeonArgMinMaxWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const ArgMinMaxDescriptor &descriptor)
NeonArgMinMaxWorkload(const ArgMinMaxQueueDescriptor &descriptor, const WorkloadInfo &info)
virtual void Execute() const override