Fix missing include

tensorflow · Nov 14, 2024 · 6b80c95 · 6b80c95
1 parent 534a5d2
commit 6b80c95
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tensorflow/lite/micro/micro_interpreter_graph.cc b/tensorflow/lite/micro/micro_interpreter_graph.cc
@@ -20,6 +20,7 @@ limitations under the License.
 #include "tensorflow/lite/kernels/internal/compatibility.h"
 #include "tensorflow/lite/micro/flatbuffer_utils.h"
 #include "tensorflow/lite/micro/memory_helpers.h"
+#include "tensorflow/lite/micro/micro_context.h"
 #include "tensorflow/lite/micro/micro_log.h"
 #include "tensorflow/lite/micro/micro_profiler.h"
 #include "tensorflow/lite/schema/schema_generated.h"
@@ -227,8 +228,8 @@ TfLiteStatus MicroInterpreterGraph::InvokeSubgraph(int subgraph_idx) {
     if (invoke_status != kTfLiteOk) {
       if (invoke_status != kTfLiteAbort) {
         MicroPrintf("Node %s (number %d) failed to invoke with status %d",
-                    OpNameFromRegistration(registration), current_operator_index_,
-                    invoke_status);
+                    OpNameFromRegistration(registration),
+                    current_operator_index_, invoke_status);
       }
       return invoke_status;
     }