Skip to content

Update gen micro mutable readme #1839

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 2 commits into from
Mar 24, 2023
Merged

Update gen micro mutable readme #1839

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
0040740
Update gen micro mutable readme
mansnils Mar 22, 2023
301a402
Merge branch 'main' into bugs
mergify[bot] Mar 24, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
22 changes: 13 additions & 9 deletions tensorflow/lite/micro/tools/gen_micro_mutable_op_resolver/README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@ bazel run tensorflow/lite/micro/tools/gen_micro_mutable_op_resolver:generate_mic

Note that if having only one tflite as input, the final output directory will be <output directory>/<base name of model>.

Example1:
Example:

```
bazel run tensorflow/lite/micro/tools/gen_micro_mutable_op_resolver:generate_micro_mutable_op_resolver_from_model -- \
Expand All @@ -22,7 +22,7 @@ bazel run tensorflow/lite/micro/tools/gen_micro_mutable_op_resolver:generate_mic

A header file called, gen_micro_mutable_op_resolver.h will be created in /tmp/gen_dir/person_detect.

Example2:
Example:

```
bazel run tensorflow/lite/micro/tools/gen_micro_mutable_op_resolver:generate_micro_mutable_op_resolver_from_model -- \
Expand Down Expand Up @@ -69,27 +69,31 @@ bazel run gen_dir/person_detect:micro_mutable_op_resolver_test

By default the model will run without any generated input or verifying the output. This can be done by adding the flag --verify_output=1.

Example:
Example assuming gen_dir and /tmp/my_model.tflite exists:

```
bazel run tensorflow/lite/micro/tools/gen_micro_mutable_op_resolver:generate_micro_mutable_op_resolver_from_model -- \
--input_tflite_files=/tmp/my_model.tflite --output_dir=$(realpath gen_dir)
--common_tflite_path=/tmp/ \
--input_tflite_files=my_model.tflite --output_dir=$(realpath gen_dir/my_model)
bazel run tensorflow/lite/micro/tools/gen_micro_mutable_op_resolver:generate_micro_mutable_op_resolver_from_model_test -- \
--input_tflite_files=/tmp/my_model.tflite --output_dir=$(realpath gen_dir) --verify_output=1
--input_tflite_file=/tmp/my_model.tflite --output_dir=$(realpath gen_dir) --verify_output=1
bazel run gen_dir/my_model:micro_mutable_op_resolver_test

```

Note that since test script appends the name of the model in the output directory, we add that to the output directory for the generated header (gen_dir/my_model) so that header and test files ends up in same directory.

Depending on the size of the input model the arena size may need to be increased. Arena size can be set with --arena_size=<size>.

Example:
Example assuming gen_dir and /tmp/big_model.tflite exists:

```
bazel run tensorflow/lite/micro/tools/gen_micro_mutable_op_resolver:generate_micro_mutable_op_resolver_from_model -- \
--input_tflite_files=/tmp/big_model.tflite --output_dir=gen_dir
--common_tflite_path=/tmp/ \
--input_tflite_files=big_model.tflite --output_dir=$(realpath gen_dir/big_model)
bazel run tensorflow/lite/micro/tools/gen_micro_mutable_op_resolver:generate_micro_mutable_op_resolver_from_model_test -- \
--input_tflite_files=/tmp/big_model.tflite --output_dir=gen_dir --verify_output=1 --arena_size=1000000
bazel run tensorflow/lite/micro/tools/gen_micro_mutable_op_resolver/generated/big_model:micro_mutable_op_resolver_test
--input_tflite_file=/tmp/big_model.tflite --output_dir=$(realpath gen_dir) --verify_output=1 --arena_size=1000000
bazel run gen_dir/big_model:micro_mutable_op_resolver_test

```

Expand Down