IVGCVSW-3697 Add Support for ANEURALNETWORKS_ARG[MAX|MIN] to HAL 1.2 Driver
!armnn:2313
Signed-off-by: Francis Murtagh <francis.murtagh@arm.com>
Change-Id: I543136e4e2ef9aece1378d2642064cc585246645
diff --git a/1.2/HalPolicy.cpp b/1.2/HalPolicy.cpp
index c8e2968..e6f8acb 100644
--- a/1.2/HalPolicy.cpp
+++ b/1.2/HalPolicy.cpp
@@ -88,6 +88,10 @@
return ConvertAbs(operation, model, data);
case V1_2::OperationType::ADD:
return ConvertAdd(operation, model, data);
+ case V1_2::OperationType::ARGMAX:
+ return ConvertArgMinMax(operation, model, data, ArgMinMaxFunction::Max);
+ case V1_2::OperationType::ARGMIN:
+ return ConvertArgMinMax(operation, model, data, ArgMinMaxFunction::Min);
case V1_2::OperationType::AVERAGE_POOL_2D:
return ConvertAveragePool2d(operation, model, data);
case V1_2::OperationType::BATCH_TO_SPACE_ND:
@@ -210,6 +214,15 @@
return ::ConvertAdd<hal_1_2::HalPolicy>(operation, model, data);
}
+bool HalPolicy::ConvertArgMinMax(const V1_2::Operation& operation,
+ const V1_2::Model& model,
+ ConversionData& data,
+ armnn::ArgMinMaxFunction argMinMaxFunction)
+{
+ ALOGV("hal_1_2::HalPolicy::ConvertArgMinMax()");
+ return ::ConvertArgMinMax<hal_1_2::HalPolicy>(operation, model, data, argMinMaxFunction);
+}
+
bool HalPolicy::ConvertAveragePool2d(const Operation& operation, const Model& model, ConversionData& data)
{
ALOGV("hal_1_2::HalPolicy::ConvertAveragePool2d()");