Update nnapi_delegation for Space_To_BatchND (#2904)

This patch will add nnapi delegaion for Space_to_BatchND

Signed-off-by: shubham <shub98.gupta@samsung.com>

diff --git a/libs/support/tflite/src/nnapi_delegate.cpp b/libs/support/tflite/src/nnapi_delegate.cpp
index 9de80c1..4a4a416 100644 (file)
--- 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: