diff --git a/compiler/dredd-rule-lib/README.md b/compiler/dredd-rule-lib/README.md index 348b0aefba9..2105e3c198a 100644 --- a/compiler/dredd-rule-lib/README.md +++ b/compiler/dredd-rule-lib/README.md @@ -21,6 +21,7 @@ Models (input of test) exist in *model repo*, where The following metric functions are provided: - `all_op_count` : the count of operations inside a compiled tflite file - `file_size` : the size of compiled tflite file +- `tensor_shape` : The shape of a specific node in a compiled tflite file. The format looks like `[1,-1,7,2]`. - In addition, `op_count`, `conv2d_weight_not_constant`, etc. - Please , refer to [`rule-lib.sh`](rule-lib.sh) for metric functions