Print invoke status on unsuccessful invocation

tensorflow · Nov 12, 2024 · 2d52ab1 · 2d52ab1
1 parent 9245002
commit 2d52ab1
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/tensorflow/lite/micro/micro_interpreter_graph.cc b/tensorflow/lite/micro/micro_interpreter_graph.cc
@@ -224,12 +224,10 @@ TfLiteStatus MicroInterpreterGraph::InvokeSubgraph(int subgraph_idx) {
     // prepare for the next call.
     allocator_->ResetTempAllocations();
 
-    if (invoke_status == kTfLiteError) {
+    if (invoke_status != kTfLiteOk) {
       MicroPrintf("Node %s (number %d) failed to invoke with status %d",
                   OpNameFromRegistration(registration), current_operator_index_,
                   invoke_status);
-      return kTfLiteError;
-    } else if (invoke_status != kTfLiteOk) {
       return invoke_status;
     }
   }