Do not print error message on kTfLiteAbort status

tensorflow · Nov 13, 2024 · 534a5d2 · 534a5d2
1 parent 2d52ab1
commit 534a5d2
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/tensorflow/lite/micro/micro_interpreter_graph.cc b/tensorflow/lite/micro/micro_interpreter_graph.cc
@@ -225,9 +225,11 @@ TfLiteStatus MicroInterpreterGraph::InvokeSubgraph(int subgraph_idx) {
     allocator_->ResetTempAllocations();
 
     if (invoke_status != kTfLiteOk) {
-      MicroPrintf("Node %s (number %d) failed to invoke with status %d",
-                  OpNameFromRegistration(registration), current_operator_index_,
-                  invoke_status);
+      if (invoke_status != kTfLiteAbort) {
+        MicroPrintf("Node %s (number %d) failed to invoke with status %d",
+                    OpNameFromRegistration(registration), current_operator_index_,
+                    invoke_status);
+      }
       return invoke_status;
     }
   }