From 34e96f017b79825d2cf379ff8877e568670e7d9c Mon Sep 17 00:00:00 2001 From: =?utf8?q?Shubham=20Gupta/System=20SW=20/SRI-Bangalore/Engineer/?= =?utf8?q?=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Fri, 5 Oct 2018 09:36:59 +0530 Subject: [PATCH] Update nnapi_delegation for Space_To_BatchND (#2904) This patch will add nnapi delegaion for Space_to_BatchND Signed-off-by: shubham --- 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 9de80c1..4a4a416 100644 --- a/libs/support/tflite/src/nnapi_delegate.cpp +++ b/libs/support/tflite/src/nnapi_delegate.cpp @@ -484,6 +484,10 @@ void AddOpsAndParams(tflite::Interpreter* interpreter, add_strided_slice_params(node.builtin_data); nn_op_type = ANEURALNETWORKS_STRIDED_SLICE; break; + case tflite::BuiltinOperator_SPACE_TO_BATCH_ND: + nnapi_version = 11; // require NNAPI 1.1 + nn_op_type = ANEURALNETWORKS_SPACE_TO_BATCH_ND; + break; case tflite::BuiltinOperator_CAST: CHECK_NN(ANeuralNetworksModel_addOperationEx( nn_model, ANEURALNETWORKS_CAST_EX, @@ -544,7 +548,6 @@ void AddOpsAndParams(tflite::Interpreter* interpreter, case tflite::BuiltinOperator_PADV2: case tflite::BuiltinOperator_CALL: case tflite::BuiltinOperator_SKIP_GRAM: - case tflite::BuiltinOperator_SPACE_TO_BATCH_ND: case tflite::BuiltinOperator_BATCH_TO_SPACE_ND: case tflite::BuiltinOperator_SQUEEZE: case tflite::BuiltinOperator_LOG_SOFTMAX: -- 2.7.4