Add NNAPI delegation for LOGICAL_OR (#3904)
authorShubham Gupta/SNAP /SRI-Bangalore/Engineer/삼성전자 <shub98.gupta@samsung.com>
Mon, 10 Dec 2018 01:42:50 +0000 (07:12 +0530)
committer오형석/동작제어Lab(SR)/Staff Engineer/삼성전자 <hseok82.oh@samsung.com>
Mon, 10 Dec 2018 01:42:50 +0000 (10:42 +0900)
This patch adds NNAPI delegation for LOGICAL_OR.
Related issue: #3459

Signed-off-by: shubham <shub98.gupta@samsung.com>
libs/support/tflite/src/nnapi_delegate.cpp

index 9e9c583..55c5233 100644 (file)
@@ -838,6 +838,14 @@ TfLiteStatus AddOpsAndParams(
             static_cast<uint32_t>(node.outputs->size),
             reinterpret_cast<uint32_t*>(node.outputs->data)));
         continue;
+      case tflite::BuiltinOperator_LOGICAL_OR:
+        CHECK_NN(ANeuralNetworksModel_addOperationEx(
+            nn_model, ANEURALNETWORKS_LOGICAL_OR_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;
 #endif
       case tflite::BuiltinOperator_CONCAT_EMBEDDINGS:
       case tflite::BuiltinOperator_LSH_PROJECTION:
@@ -895,7 +903,7 @@ TfLiteStatus AddOpsAndParams(
       case tflite::BuiltinOperator_POW:
       case tflite::BuiltinOperator_FAKE_QUANT:
       //case tflite::BuiltinOperator_PACK:
-      case tflite::BuiltinOperator_LOGICAL_OR:
+      //case tflite::BuiltinOperator_LOGICAL_OR:
       case tflite::BuiltinOperator_ONE_HOT:
       //case tflite::BuiltinOperator_LOGICAL_AND:
       case tflite::BuiltinOperator_LOGICAL_NOT: