commit | 9fb745d92173abfa270e99bd5c9bd7cf85bfeb31 | [log] [tgz] |
---|---|---|
author | Thibaut Goetghebuer-Planchon <thibaut.goetghebuer-planchon@arm.com> | Wed Nov 23 11:42:43 2022 +0000 |
committer | Thibaut Goetghebuer-Planchon <thibaut.goetghebuer-planchon@arm.com> | Tue Dec 13 10:53:38 2022 +0000 |
tree | 3a430220145a184a940131d6f7b0465dacdbde3b | |
parent | a2bcf5f818699082adfd346eba216d96f14d6e6c [diff] |
Models with tfl.custom ops are considered to be non TOSA compliant Change-Id: Ib31b44b3819bbdf517b96d834879155eebb4f09c
The TOSA Checker is a tool that provides an easy way to ensure that a TensorFlow™ Lite model is compatible with the TOSA specification.
The project is currently in alpha, some stability issues may still be found.
To build the TOSA Checker the following are required:
pip install -U pip setuptools wheel pip install -U numpy python setup.py bdist_wheel pip install dist/tosa_checker-0.1.0-cp38-cp38-linux_x86_64.whl
A Docker™ image to build the TOSA Checker is also provided. More information can be found on docker/README.md.
>>> import tosa_checker as tc >>> checker = tc.TOSAChecker(model_path="model.tflite") >>> checker.is_tosa_compatible() True
TOSA Checker is licensed under [Apache License 2.0].