Skip to content
This repository was archived by the owner on Jun 23, 2024. It is now read-only.

Commit 3178052

Browse files
authored
Merge pull request #14 from iabdalkader/add_ops_size
2 parents b14944d + 624fa03 commit 3178052

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

libtf.cc

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -264,6 +264,7 @@ extern "C" {
264264
libtf_resolver_init(resolver);
265265

266266
tflite::MicroInterpreter interpreter(model, resolver, tensor_arena, tensor_arena_size, error_reporter);
267+
params->operators_size = interpreter.operators_size();
267268

268269
if (interpreter.AllocateTensors() != kTfLiteOk) {
269270
error_reporter->Report("AllocateTensors() failed!");

libtf.h

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@ typedef struct libtf_parameters {
2828
libtf_datatype_t output_datatype;
2929
float output_scale;
3030
int output_zero_point;
31+
size_t operators_size;
3132
} libtf_parameters_t;
3233

3334
// Call this first to get the model parameters.

0 commit comments

Comments
 (0)