Skip to content

Commit ea36733

Browse files
authored
1 parent 75f5290 commit ea36733

File tree

1 file changed

+13
-9
lines changed
  • tensorflow/lite/micro/tools/gen_micro_mutable_op_resolver

1 file changed

+13
-9
lines changed

tensorflow/lite/micro/tools/gen_micro_mutable_op_resolver/README.md

Lines changed: 13 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ bazel run tensorflow/lite/micro/tools/gen_micro_mutable_op_resolver:generate_mic
1212

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

15-
Example1:
15+
Example:
1616

1717
```
1818
bazel run tensorflow/lite/micro/tools/gen_micro_mutable_op_resolver:generate_micro_mutable_op_resolver_from_model -- \
@@ -22,7 +22,7 @@ bazel run tensorflow/lite/micro/tools/gen_micro_mutable_op_resolver:generate_mic
2222

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

25-
Example2:
25+
Example:
2626

2727
```
2828
bazel run tensorflow/lite/micro/tools/gen_micro_mutable_op_resolver:generate_micro_mutable_op_resolver_from_model -- \
@@ -69,27 +69,31 @@ bazel run gen_dir/person_detect:micro_mutable_op_resolver_test
6969

7070
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.
7171

72-
Example:
72+
Example assuming gen_dir and /tmp/my_model.tflite exists:
7373

7474
```
7575
bazel run tensorflow/lite/micro/tools/gen_micro_mutable_op_resolver:generate_micro_mutable_op_resolver_from_model -- \
76-
--input_tflite_files=/tmp/my_model.tflite --output_dir=$(realpath gen_dir)
76+
--common_tflite_path=/tmp/ \
77+
--input_tflite_files=my_model.tflite --output_dir=$(realpath gen_dir/my_model)
7778
bazel run tensorflow/lite/micro/tools/gen_micro_mutable_op_resolver:generate_micro_mutable_op_resolver_from_model_test -- \
78-
--input_tflite_files=/tmp/my_model.tflite --output_dir=$(realpath gen_dir) --verify_output=1
79+
--input_tflite_file=/tmp/my_model.tflite --output_dir=$(realpath gen_dir) --verify_output=1
7980
bazel run gen_dir/my_model:micro_mutable_op_resolver_test
8081
8182
```
8283

84+
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.
85+
8386
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>.
8487

85-
Example:
88+
Example assuming gen_dir and /tmp/big_model.tflite exists:
8689

8790
```
8891
bazel run tensorflow/lite/micro/tools/gen_micro_mutable_op_resolver:generate_micro_mutable_op_resolver_from_model -- \
89-
--input_tflite_files=/tmp/big_model.tflite --output_dir=gen_dir
92+
--common_tflite_path=/tmp/ \
93+
--input_tflite_files=big_model.tflite --output_dir=$(realpath gen_dir/big_model)
9094
bazel run tensorflow/lite/micro/tools/gen_micro_mutable_op_resolver:generate_micro_mutable_op_resolver_from_model_test -- \
91-
--input_tflite_files=/tmp/big_model.tflite --output_dir=gen_dir --verify_output=1 --arena_size=1000000
92-
bazel run tensorflow/lite/micro/tools/gen_micro_mutable_op_resolver/generated/big_model:micro_mutable_op_resolver_test
95+
--input_tflite_file=/tmp/big_model.tflite --output_dir=$(realpath gen_dir) --verify_output=1 --arena_size=1000000
96+
bazel run gen_dir/big_model:micro_mutable_op_resolver_test
9397
9498
```
9599

0 commit comments

Comments
 (0)