commit 1aea22052d93f79f7ae468970a50220ef6b6fbcc
author Slava Shklyaev <slavash@google.com> Thu Mar 28 17:56:08 2019 +0000
committer Slava Shklyaev <slavash@google.com> Mon Apr 08 14:36:51 2019 +0100
tree 8757b4edc7025270c41ee040f60fdc9319b46fee
parent 7ff851a9363fbc7f3d17a5e22662d6c0d9f5aa60

Allow NNAPI delegate to report compliation failure

Bug: 123041316
Test: atest VtsHalNeuralnetworksV1_2Benchmark (from change I4a0b83c85775acae2b8e502aae60830ff4f28507)
Change-Id: Ib3229e20348eaee2cb968a213ac1f68bfce7418c
Merged-In: Ib3229e20348eaee2cb968a213ac1f68bfce7418c
(cherry picked from commit d739c1016333e9ed0b282fcc87ae8772ebd77e73)

tensorflow/lite/delegates/nnapi/nnapi_delegate.cc

diff --git a/tensorflow/lite/delegates/nnapi/nnapi_delegate.cc b/tensorflow/lite/delegates/nnapi/nnapi_delegate.cc
index c30a806..b656116 100644
--- a/tensorflow/lite/delegates/nnapi/nnapi_delegate.cc
+++ b/tensorflow/lite/delegates/nnapi/nnapi_delegate.cc
@@ -1327,9 +1327,9 @@
       },
 
       .prepare = [](TfLiteContext* context, TfLiteNode* node) -> TfLiteStatus {
-        // Since the underlying resize happened ahead of delegation
-        // worked. This does nothing.
-        return kTfLiteOk;
+        NNAPIDelegateKernel* state =
+            reinterpret_cast<NNAPIDelegateKernel*>(node->user_data);
+        return state->nn_compilation_ == nullptr ? kTfLiteError : kTfLiteOk;
       },
 
       .invoke = [](TfLiteContext* context, TfLiteNode* node) -> TfLiteStatus {