Add CUSTOM(Abs) op to nnapi_delegate (#4379)

This commit adds CUSTOM(Abs) op to nnapi_delegate.

Signed-off-by: jiseob.jang <jiseob.jang@samsung.com>

diff --git a/libs/tflite/src/ext/nnapi_delegate.cpp b/libs/tflite/src/ext/nnapi_delegate.cpp
index 27c9cc4..472b3fb 100644 (file)
--- a/libs/tflite/src/ext/nnapi_delegate.cpp
+++ b/libs/tflite/src/ext/nnapi_delegate.cpp
@@ -960,6 +960,15 @@ TfLiteStatus AddOpsAndParams(
               reinterpret_cast<uint32_t*>(node.outputs->data)));
           continue;
         }
+        else if (custom_name.compare("Abs") == 0) {
+          CHECK_NN(ANeuralNetworksModel_addOperationEx(
+              nn_model, ANEURALNETWORKS_ABS_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;
+        }
         logError("Custom operations are not supported when using NNAPI.");
         return kTfLiteError;
         break;