tree: fe280d301cc70871e2c043fd1be4c9820420b1e4 [path history] [tgz]
  1. README.md
  2. report_model_ops.py
python/scripts/report-model-ops/README.md

Extract list of operators from a list of TfLite models

Introduction

Purpose of this script is to inspect a list of user-provided TfLite models and report the list of operators that are used as well as the data-types that the models operate on. The script can subsequently generate a configuration file that can be provided to the Compute Library build system and generate a library that contains only the operators required by the given model(s) to run.

Utilizing this script, use-case tailored Compute Library dynamic libraries can be created, helping reduce the overall binary size requirements.

Usage example

Assuming that the virtual environment is activated and the requirements are present, we can run the following command:

./report_model_ops.py -m modelA.tfile modelB.tflite -c build_config.json

Input arguments

models (required) : A list of comma separated model files.

Supported model formats are:

  • TfLite

config (optional) : The configuration file to be created on JSON format that can be provided to ComputeLibrary's build system and generate a library with the given list of operators and data-types

debug (optional) : Flag that enables debug information