aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorTeresa Charlin <teresa.charlinreyes@arm.com>2023-05-12 14:04:41 +0100
committerTeresa Charlin <teresa.charlinreyes@arm.com>2023-05-18 17:04:42 +0100
commit9a29dacfe40f076786770cbadef4c471e750a4d2 (patch)
tree9c6c373be7fc90233ba71cd03f835941e8d81a1b
parente9fd94148271a7c61323251319c2cbeeffc8e24c (diff)
downloadandroid-nn-driver-9a29dacfe40f076786770cbadef4c471e750a4d2.tar.gz
IVGCVSW-7400 Add support for Pow
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com> Change-Id: Ida6d93d203d2430befbbd2deaa02108078de73f8
-rw-r--r--1.2/HalPolicy.cpp2
-rw-r--r--1.3/HalPolicy.cpp2
2 files changed, 4 insertions, 0 deletions
diff --git a/1.2/HalPolicy.cpp b/1.2/HalPolicy.cpp
index bfc467cf..4e53c744 100644
--- a/1.2/HalPolicy.cpp
+++ b/1.2/HalPolicy.cpp
@@ -133,6 +133,8 @@ bool HalPolicy::ConvertOperation(const Operation& operation, const Model& model,
return ConvertPad(operation, model, data);
case V1_2::OperationType::PAD_V2:
return ConvertPadV2(operation, model, data);
+ case V1_2::OperationType::POW:
+ return ConvertElementwiseBinary(operation, model, data, BinaryOperation::Power);
case V1_2::OperationType::PRELU:
return ConvertPrelu(operation, model, data);
case V1_2::OperationType::QUANTIZE:
diff --git a/1.3/HalPolicy.cpp b/1.3/HalPolicy.cpp
index 16b325ba..aaf0813c 100644
--- a/1.3/HalPolicy.cpp
+++ b/1.3/HalPolicy.cpp
@@ -117,6 +117,8 @@ bool HalPolicy::ConvertOperation(const Operation& operation, const Model& model,
return ConvertPad(operation, model, data);
case V1_3::OperationType::PAD_V2:
return ConvertPadV2(operation, model, data);
+ case V1_3::OperationType::POW:
+ return ConvertElementwiseBinary(operation, model, data, BinaryOperation::Power);
case V1_3::OperationType::PRELU:
return ConvertPrelu(operation, model, data);
case V1_3::OperationType::QUANTIZE: