aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorPablo Tello <pablo.tello@arm.com>2019-12-17 10:10:53 +0000
committermike.kelly <mike.kelly@arm.com>2019-12-17 13:00:30 +0000
commitc05798ca177d6a198d02425f6c57a35ec9ed6468 (patch)
tree8a0569cb75391155472edc90392bafe43bbe3299
parentd486e5289921f937fb7a9c2d84ff87229fa25d43 (diff)
downloadandroid-nn-driver-c05798ca177d6a198d02425f6c57a35ec9ed6468.tar.gz
IVGCVSW-4241 Enable L2_POOL_2D in Hal 1.2
Change-Id: Ib2a057d63a0722d90191ae69af2751cd03c33910 Signed-off-by: Pablo Tello <pablo.tello@arm.com>
-rw-r--r--1.2/HalPolicy.cpp8
1 files changed, 8 insertions, 0 deletions
diff --git a/1.2/HalPolicy.cpp b/1.2/HalPolicy.cpp
index cbd54e6f..5118ffb8 100644
--- a/1.2/HalPolicy.cpp
+++ b/1.2/HalPolicy.cpp
@@ -43,6 +43,8 @@ bool HalPolicy::ConvertOperation(const Operation& operation, const Model& model,
return ConvertFullyConnected(operation, model, data);
case V1_2::OperationType::L2_NORMALIZATION:
return ConvertL2Normalization(operation, model, data);
+ case V1_2::OperationType::L2_POOL_2D:
+ return ConvertL2Pool2d(operation, model, data);
case V1_2::OperationType::LOCAL_RESPONSE_NORMALIZATION:
return ConvertLocalResponseNormalization(operation, model, data);
case V1_2::OperationType::LOGISTIC:
@@ -492,6 +494,12 @@ bool HalPolicy::ConvertL2Normalization(const Operation& operation, const Model&
return ::ConvertL2Normalization<hal_1_2::HalPolicy>(operation, model, data);
}
+bool HalPolicy::ConvertL2Pool2d(const Operation& operation, const Model& model, ConversionData& data)
+{
+ ALOGV("hal_1_2::HalPolicy::ConvertL2Pool2d()");
+ return ConvertPooling2d<hal_1_2::HalPolicy>(operation, __func__, armnn::PoolingAlgorithm::L2, model, data);
+}
+
bool HalPolicy::ConvertLocalResponseNormalization(const Operation& operation,
const Model& model,
ConversionData& data)