Add NNAPI delegation to PRELU op (#3807)

This patch adds NNAPI delegation to PRELU op.

Signed-off-by: prasannar <prasanna.r@samsung.com>

diff --git a/libs/support/tflite/src/nnapi_delegate.cpp b/libs/support/tflite/src/nnapi_delegate.cpp
index 6b882f6..5286ee8 100644 (file)
--- a/libs/support/tflite/src/nnapi_delegate.cpp
+++ b/libs/support/tflite/src/nnapi_delegate.cpp
@@ -744,6 +744,14 @@ TfLiteStatus AddOpsAndParams(
             augmented_inputs.data(), static_cast<uint32_t>(node.outputs->size),
             reinterpret_cast<uint32_t*>(node.outputs->data)));
         continue;
+      case tflite::BuiltinOperator_PRELU:
+        CHECK_NN(ANeuralNetworksModel_addOperationEx(
+            nn_model, ANEURALNETWORKS_PRELU_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_ARG_MAX:
         CHECK_NN(ANeuralNetworksModel_addOperationEx(
             nn_model, ANEURALNETWORKS_ARGMAX_EX,
@@ -807,7 +815,7 @@ TfLiteStatus AddOpsAndParams(
       //case tflite::BuiltinOperator_DEQUANTIZE:
       case tflite::BuiltinOperator_DELEGATE:
       //case tflite::BuiltinOperator_CAST:
-      case tflite::BuiltinOperator_PRELU:
+      //case tflite::BuiltinOperator_PRELU:
       case tflite::BuiltinOperator_MAXIMUM:
       case tflite::BuiltinOperator_MINIMUM:
 #ifndef OBS_BUILD