1- /* Copyright 2021 The TensorFlow Authors. All Rights Reserved.
1+ /* Copyright 2025 The TensorFlow Authors. All Rights Reserved.
22
33Licensed under the Apache License, Version 2.0 (the "License");
44you may not use this file except in compliance with the License.
@@ -20,6 +20,7 @@ limitations under the License.
2020#include " tensorflow/lite/kernels/internal/compatibility.h"
2121#include " tensorflow/lite/micro/flatbuffer_utils.h"
2222#include " tensorflow/lite/micro/memory_helpers.h"
23+ #include " tensorflow/lite/micro/micro_context.h"
2324#include " tensorflow/lite/micro/micro_log.h"
2425#include " tensorflow/lite/micro/micro_profiler.h"
2526#include " tensorflow/lite/schema/schema_generated.h"
@@ -236,12 +237,12 @@ TfLiteStatus MicroInterpreterGraph::InvokeSubgraph(int subgraph_idx) {
236237 // prepare for the next call.
237238 allocator_->ResetTempAllocations ();
238239
239- if (invoke_status == kTfLiteError ) {
240- MicroPrintf ( " Node %s (number %d) failed to invoke with status %d " ,
241- OpNameFromRegistration (registration), current_operator_index_ ,
242- invoke_status);
243- return kTfLiteError ;
244- } else if (invoke_status != kTfLiteOk ) {
240+ if (invoke_status != kTfLiteOk ) {
241+ if (invoke_status != kTfLiteAbort ) {
242+ MicroPrintf ( " Node %s (number %d) failed to invoke with status %d " ,
243+ OpNameFromRegistration (registration),
244+ current_operator_index_, invoke_status) ;
245+ }
245246 return invoke_status;
246247 }
247248 }
0 commit comments