Gitiles
Code Review Sign In
review.mlplatform.org / tosa / tosa_checker / refs/tags/0.1.0-rc.1
commit52dacd6556d60815253d4e4938e218ea3d8084a2[log] [tgz]
authorThibaut Goetghebuer-Planchon <thibaut.goetghebuer-planchon@arm.com>Wed Jul 06 10:23:22 2022 +0100
committerThibaut Goetghebuer-Planchon <thibaut.goetghebuer-planchon@arm.com>Thu Aug 18 14:35:45 2022 +0100
tree4c470c567da6f70f65987d5af161bf4f950d107b
parentcc5d89eea4ff3dc398cac3b6025450f48ac20c1e [diff]
Initial commit

Change-Id: I2fb0933d595a6ede6417d09dd905ef72d6c60c9b
22 files changed
tree: 4c470c567da6f70f65987d5af161bf4f950d107b
  1. .bazelrc
  2. .bazelversion
  3. .bazelversion.license
  4. .gitignore
  5. LICENSES/
  6. MANIFEST.in
  7. README.md
  8. RELEASES.md
  9. WORKSPACE
  10. docker/
  11. examples/
  12. setup.py
  13. tests/
  14. tosa_checker/
README.md

TOSA Checker

Overview

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.

Prerequisites

To build the TOSA Checker the following are required:

  • Ubuntu® 20.04.03 LTS (the TOSA Checker has been tested on this OS, but should work on other Linux® OS)
  • Bazel™ (Bazelisk is an easy way to install the correct version of Bazel™)
  • Git™
  • Python® >=3.7 (python, python-dev and python-pip)

Building and installation

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

Docker™ image

A Docker™ image to build the TOSA Checker is also provided. More information can be found on docker/README.md.

Usage

>>> import tosa_checker as tc
>>> checker = tc.TOSAChecker(model_path="model.tflite")
>>> checker.is_tosa_compatible()
True

License

TOSA Checker is licensed under [Apache License 2.0].

Trademarks and Copyrights

Bazel™ is a trademark of Google® LLC. Git™ is a trademark of Software Freedom Conservancy. Linux® is the registered trademark of Linus Torvalds in the U.S. and elsewhere. Python® is a registered trademark of the PSF. Ubuntu® is a registered trademark of Canonical. Tensorflow™ is a trademark of Google® LLC. Docker™ is a trademark of Docker, Inc.