1
- /* Copyright 2021 The TensorFlow Authors. All Rights Reserved.
1
+ /* Copyright 2025 The TensorFlow Authors. All Rights Reserved.
2
2
3
3
Licensed under the Apache License, Version 2.0 (the "License");
4
4
you may not use this file except in compliance with the License.
@@ -20,6 +20,7 @@ limitations under the License.
20
20
#include " tensorflow/lite/kernels/internal/compatibility.h"
21
21
#include " tensorflow/lite/micro/flatbuffer_utils.h"
22
22
#include " tensorflow/lite/micro/memory_helpers.h"
23
+ #include " tensorflow/lite/micro/micro_context.h"
23
24
#include " tensorflow/lite/micro/micro_log.h"
24
25
#include " tensorflow/lite/micro/micro_profiler.h"
25
26
#include " tensorflow/lite/schema/schema_generated.h"
@@ -236,12 +237,12 @@ TfLiteStatus MicroInterpreterGraph::InvokeSubgraph(int subgraph_idx) {
236
237
// prepare for the next call.
237
238
allocator_->ResetTempAllocations ();
238
239
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
+ }
245
246
return invoke_status;
246
247
}
247
248
}
0 commit comments