From 9a429edbe04c69be7c15a166b6a71ef90e3bf7ca Mon Sep 17 00:00:00 2001
From: =?utf8?q?Prasanna=20R/System=20SW=20/SRI-Bangalore/Engineer/?=
 =?utf8?q?=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= <prasanna.r@samsung.com>
Date: Tue, 13 Nov 2018 15:13:02 +0530
Subject: [PATCH] Add NNAPI delegation for SQRT_EX (#3422)

This patch adds NNAPI delegation for SQRT_EX

Signed-off-by: prasannar <prasanna.r@samsung.com>
---
 libs/support/tflite/src/nnapi_delegate.cpp | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/libs/support/tflite/src/nnapi_delegate.cpp b/libs/support/tflite/src/nnapi_delegate.cpp
index 0813c58..294997e 100644
--- 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,
-- 
2.7.4