From 624fa032b23a5cdecd5f32ca2b479a21f279b0d2 Mon Sep 17 00:00:00 2001 From: iabdalkader Date: Sun, 25 Feb 2024 12:08:29 +0100 Subject: [PATCH] libtf: Add operators size to params. Signed-off-by: iabdalkader --- libtf.cc | 1 + libtf.h | 1 + 2 files changed, 2 insertions(+) diff --git a/libtf.cc b/libtf.cc index ef3f969..c513c6e 100644 --- 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 index 39ecef4..557c85a 100644 --- 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.