aboutsummaryrefslogtreecommitdiff
path: root/1.2
diff options
context:
space:
mode:
authorMike Kelly <mike.kelly@arm.com>2023-08-04 13:17:23 +0100
committerTeresaARM <teresa.charlinreyes@arm.com>2023-09-26 13:21:50 +0000
commitd261ffe7211dca5727e3472f7e29a3d16630c1ec (patch)
treefa2fe09f70368f54e8d140cb4e182e6f08277525 /1.2
parentf762b8391cdb6064fcca4c269fb582e5534bbc7f (diff)
downloadandroid-nn-driver-d261ffe7211dca5727e3472f7e29a3d16630c1ec.tar.gz
IVGCVSW-6294 Added Split support to Android-NN-Driverv23.11branches/android-nn-driver_23_11
* Added Split support to Android-NN-Driver Signed-off-by: Mike Kelly <mike.kelly@arm.com> Change-Id: I4b0bc146597dae89b68402aa78d1f1b1c894a7fb
Diffstat (limited to '1.2')
-rw-r--r--1.2/HalPolicy.cpp8
-rw-r--r--1.2/HalPolicy.hpp2
2 files changed, 10 insertions, 0 deletions
diff --git a/1.2/HalPolicy.cpp b/1.2/HalPolicy.cpp
index 9a6266e5..9c44003f 100644
--- a/1.2/HalPolicy.cpp
+++ b/1.2/HalPolicy.cpp
@@ -173,6 +173,8 @@ bool HalPolicy::ConvertOperation(const Operation& operation, const Model& model,
return ConvertSpaceToBatchNd(operation, model, data);
case V1_2::OperationType::SPACE_TO_DEPTH:
return ConvertSpaceToDepth(operation, model, data);
+ case V1_2::OperationType::SPLIT:
+ return ConvertSplit(operation, model, data);
case V1_2::OperationType::SQRT:
return ConvertSqrt(operation, model, data);
case V1_2::OperationType::SQUEEZE:
@@ -469,6 +471,12 @@ bool HalPolicy::ConvertLstm(const Operation& operation, const Model& model, Conv
return ::ConvertLstm<hal_1_2::HalPolicy>(operation, model, data);
}
+bool HalPolicy::ConvertSplit(const Operation& operation, const Model& model, ConversionData& data)
+{
+ ALOGV("hal_1_2::HalPolicy::ConvertSplit()");
+ return ::ConvertSplit<hal_1_2::HalPolicy>(operation, model, data);
+}
+
bool HalPolicy::ConvertSqrt(const Operation& operation, const Model& model, ConversionData& data)
{
ALOGV("hal_1_2::HalPolicy::ConvertSqrt()");
diff --git a/1.2/HalPolicy.hpp b/1.2/HalPolicy.hpp
index 796f4de8..4d77dfe5 100644
--- a/1.2/HalPolicy.hpp
+++ b/1.2/HalPolicy.hpp
@@ -141,6 +141,8 @@ private:
static bool ConvertSpaceToDepth(const Operation& operation, const Model& model, ConversionData& data);
+ static bool ConvertSplit(const Operation& operation, const Model& model, ConversionData& data);
+
static bool ConvertSqrt(const Operation& operation, const Model& model, ConversionData& data);
static bool ConvertSqueeze(const Operation& operation, const Model& model, ConversionData& data);