Add NNAPI delegation for SQRT_EX (#3422)

This patch adds NNAPI delegation for SQRT_EX

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 0813c58..294997e 100644 (file)
--- a/libs/support/tflite/src/nnapi_delegate.cpp
+++ b/libs/support/tflite/src/nnapi_delegate.cpp
@@ -812,6 +812,15 @@ TfLiteStatus AddOpsAndParams(
               reinterpret_cast<uint32_t*>(node.outputs->data)));
           continue;
         }
+        else if (custom_name.compare("SQRT") == 0) {
+          CHECK_NN(ANeuralNetworksModel_addOperationEx(
+              nn_model, ANEURALNETWORKS_SQRT_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;
+       }
         else if (custom_name.compare("SquaredDifference") == 0) {
           CHECK_NN(ANeuralNetworksModel_addOperationEx(
               nn_model, ANEURALNETWORKS_SQUARED_DIFFERENCE_EX,