From: 윤지영/동작제어Lab(SR)/Engineer/삼성전자 <jy910.yun@samsung.com>
Date: Fri, 24 Aug 2018 07:00:05 +0000 (+0900)
Subject: Add nnapi delegation of Split (#2462)
X-Git-Tag: 0.2~180
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=27b2191ebb3467fd434e515be4bc8ef95f14a3a7;p=platform%2Fcore%2Fml%2Fnnfw.git

Add nnapi delegation of Split (#2462)

Add nnapi delegation of Split

Signed-off-by: Jiyoung Yun <jy910.yun@samsung.com>
---

diff --git a/libs/support/tflite/src/nnapi_delegate.cpp b/libs/support/tflite/src/nnapi_delegate.cpp
index ef28acd..36dcb16 100644
--- a/libs/support/tflite/src/nnapi_delegate.cpp
+++ b/libs/support/tflite/src/nnapi_delegate.cpp
@@ -506,6 +506,13 @@ void AddOpsAndParams(tflite::Interpreter* interpreter,
             augmented_inputs.data(), static_cast<uint32_t>(node.outputs->size),
             reinterpret_cast<uint32_t*>(node.outputs->data)));
         continue;
+      case tflite::BuiltinOperator_SPLIT:
+        CHECK_NN(ANeuralNetworksModel_addOperationEx(
+            nn_model, ANEURALNETWORKS_SPLIT_EX,
+            static_cast<uint32_t>(augmented_inputs.size()),
+            augmented_inputs.data(), static_cast<uint32_t>(node.outputs->size),
+            reinterpret_cast<uint32_t*>(node.outputs->data)));
+        continue;
       case tflite::BuiltinOperator_CONCAT_EMBEDDINGS:
       case tflite::BuiltinOperator_LSH_PROJECTION:
       case tflite::BuiltinOperator_HASHTABLE_LOOKUP:
@@ -522,7 +529,6 @@ void AddOpsAndParams(tflite::Interpreter* interpreter,
       case tflite::BuiltinOperator_SPACE_TO_BATCH_ND:
       case tflite::BuiltinOperator_BATCH_TO_SPACE_ND:
       case tflite::BuiltinOperator_TRANSPOSE:
-      case tflite::BuiltinOperator_SPLIT:
       case tflite::BuiltinOperator_SQUEEZE:
       case tflite::BuiltinOperator_EXP:
       case tflite::BuiltinOperator_LOG_SOFTMAX: