commit 18b73567d561fd5a5f0067ebbc54ca1dd2c2a1ab
author Miao Wang <miaowang@google.com> Sat Mar 03 00:39:57 2018 +0000
committer android-build-merger <android-build-merger@google.com> Sat Mar 03 00:39:57 2018 +0000
tree e01aee8bfa7bc10c4e248fad550955a99760bca2
parent 01fde62e34dee571d06acee33e59219285c7c5a1
parent a8d89afb68b79f65b6d422916c0e41f86824e90d

Don't delegate quantized tensors with non-positive scale. am: ace49253a6 am: cf8faa5b9e
am: a8d89afb68

Change-Id: I4173f45ca1abcb8f6cc5359501f7e3af840dd77f

tensorflow/contrib/lite/nnapi_delegate.cc

diff --git a/tensorflow/contrib/lite/nnapi_delegate.cc b/tensorflow/contrib/lite/nnapi_delegate.cc
index 02e8499..c1dcf0e 100644
--- a/tensorflow/contrib/lite/nnapi_delegate.cc
+++ b/tensorflow/contrib/lite/nnapi_delegate.cc
@@ -91,6 +91,10 @@
         nn_type = ANEURALNETWORKS_TENSOR_QUANT8_ASYMM;
         scale = tensor->params.scale;
         zeroPoint = tensor->params.zero_point;
+        if (scale <= 0.0f) {
+            // internal tensor, not valid for NNAPI
+            continue;
+        }
         break;
       case kTfLiteInt32:
         nn_type = ANEURALNETWORKS_TENSOR_INT32;