libtf: Add operators size to params.

Signed-off-by: iabdalkader <[email protected]>
openmv · Feb 25, 2024 · 624fa03 · 624fa03
1 parent b14944d
commit 624fa03
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/libtf.cc b/libtf.cc
@@ -264,6 +264,7 @@ extern "C" {
         libtf_resolver_init(resolver);
 
         tflite::MicroInterpreter interpreter(model, resolver, tensor_arena, tensor_arena_size, error_reporter);
+        params->operators_size = interpreter.operators_size();
 
         if (interpreter.AllocateTensors() != kTfLiteOk) {
             error_reporter->Report("AllocateTensors() failed!");

diff --git a/libtf.h b/libtf.h
@@ -28,6 +28,7 @@ typedef struct libtf_parameters {
     libtf_datatype_t output_datatype;
     float output_scale;
     int output_zero_point;
+    size_t operators_size;
 } libtf_parameters_t;
 
 // Call this first to get the model parameters.