From: 윤현식/동작제어Lab(SR)/Principal Engineer/삼성전자 <hyunsik.yoon@samsung.com>
Date: Thu, 6 Sep 2018 09:23:52 +0000 (+0900)
Subject: Add Transpose into nnapi_delegate (#2624)
X-Git-Tag: 0.2~65
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=b12e92b67f7a5cdd7d0c96582435d1e52b638d10;p=platform%2Fcore%2Fml%2Fnnfw.git

Add Transpose into nnapi_delegate (#2624)

This commit enables TF Lite to call Transpose operation of NNAPI.

Signed-off-by: Hyun Sik Yoon <hyunsik.yoon@samsung.com>
---

diff --git a/libs/support/tflite/src/nnapi_delegate.cpp b/libs/support/tflite/src/nnapi_delegate.cpp
index b97529e..1eada4b 100644
--- a/libs/support/tflite/src/nnapi_delegate.cpp
+++ b/libs/support/tflite/src/nnapi_delegate.cpp
@@ -513,6 +513,10 @@ 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_TRANSPOSE:
+        nn_op_type = ANEURALNETWORKS_TRANSPOSE;
+        // param is almost same as reshape
+        break;
       case tflite::BuiltinOperator_CONCAT_EMBEDDINGS:
       case tflite::BuiltinOperator_LSH_PROJECTION:
       case tflite::BuiltinOperator_HASHTABLE_LOOKUP:
@@ -528,7 +532,6 @@ void AddOpsAndParams(tflite::Interpreter* interpreter,
       case tflite::BuiltinOperator_SKIP_GRAM:
       case tflite::BuiltinOperator_SPACE_TO_BATCH_ND:
       case tflite::BuiltinOperator_BATCH_TO_SPACE_ND:
-      case tflite::BuiltinOperator_TRANSPOSE:
       case tflite::BuiltinOperator_SQUEEZE:
       case tflite::BuiltinOperator_EXP:
       case tflite::BuiltinOperator_LOG_SOFTMAX: