Skip to content

Commit db51e8d

Browse files
authored
Print invoke status on unsuccessful invocation (#2762)
bug=#2761
1 parent ef64591 commit db51e8d

File tree

1 file changed

+8
-7
lines changed

1 file changed

+8
-7
lines changed

tensorflow/lite/micro/micro_interpreter_graph.cc

+8-7
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
/* Copyright 2021 The TensorFlow Authors. All Rights Reserved.
1+
/* Copyright 2025 The TensorFlow Authors. All Rights Reserved.
22
33
Licensed under the Apache License, Version 2.0 (the "License");
44
you 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

Comments
 (0)