From b12e92b67f7a5cdd7d0c96582435d1e52b638d10 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EC=9C=A4=ED=98=84=EC=8B=9D/=EB=8F=99=EC=9E=91=EC=A0=9C?= =?utf8?q?=EC=96=B4Lab=28SR=29/Principal=20Engineer/=EC=82=BC=EC=84=B1?= =?utf8?q?=EC=A0=84=EC=9E=90?= Date: Thu, 6 Sep 2018 18:23:52 +0900 Subject: [PATCH] Add Transpose into nnapi_delegate (#2624) This commit enables TF Lite to call Transpose operation of NNAPI. Signed-off-by: Hyun Sik Yoon --- libs/support/tflite/src/nnapi_delegate.cpp | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) 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(node.outputs->size), reinterpret_cast(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: -- 2.7.4