Handle boolean for nnapi delegate (#3775)
author오형석/동작제어Lab(SR)/Staff Engineer/삼성전자 <hseok82.oh@samsung.com>
Fri, 30 Nov 2018 07:38:29 +0000 (16:38 +0900)
committerGitHub Enterprise <noreply-CODE@samsung.com>
Fri, 30 Nov 2018 07:38:29 +0000 (16:38 +0900)
To handle boolean for nnapi delegate, change type to quantized tensor and call NNAPI

Signed-off-by: Hyeongseok Oh <hseok82.oh@samsung.com>
libs/support/tflite/src/nnapi_delegate.cpp

index 95157c1..f73871e 100644 (file)
@@ -197,6 +197,13 @@ TfLiteStatus addTensorOperands(tflite::Interpreter* interpreter,
         scale = tensor->params.scale;
         zeroPoint = tensor->params.zero_point;
         break;
+      case kTfLiteBool:
+        // Workaround to pass bool type under NNAPI
+        // Use bool type using ANEURALNETWORKS_TENSOR_QUANT8_ASYMM with scale = 1.0f and zero_point = 0
+        nn_type = ANEURALNETWORKS_TENSOR_QUANT8_ASYMM;
+        scale = 1.0f;
+        zeroPoint = 0;
+        break;
       default:
         logError("Unsupported tensor type %d", tensor->type);
         return kTfLiteError;