diff options
author | Tim Hall <tim.hall@arm.com> | 2020-11-12 23:29:25 +0000 |
---|---|---|
committer | Tim Hall <tim.hall@arm.com> | 2020-11-20 12:55:48 +0000 |
commit | d13bd061b28067d65c1565b6d7ba87e64b8cea63 (patch) | |
tree | 57a5fa18a5643429cfa53d7033bc0e67da1dc1b3 | |
parent | 907cb03bb73ce590b044947b0981ceb25e8c3430 (diff) | |
download | ethos-u-vela-d13bd061b28067d65c1565b6d7ba87e64b8cea63.tar.gz |
vela: Update tool description text
- Updated and aligned the --help and setup.py descriptions
Signed-off-by: Tim Hall <tim.hall@arm.com>
Change-Id: I78c11b1b3dd51284b34d57a6caca45cd222b4678
-rw-r--r-- | ethosu/vela/vela.py | 9 | ||||
-rw-r--r-- | setup.py | 2 |
2 files changed, 8 insertions, 3 deletions
diff --git a/ethosu/vela/vela.py b/ethosu/vela/vela.py index 05bd9ec1..6fc5c47f 100644 --- a/ethosu/vela/vela.py +++ b/ethosu/vela/vela.py @@ -187,7 +187,7 @@ def main(args=None): if args is None: args = sys.argv[1:] - parser = argparse.ArgumentParser(prog="vela", description="Neural network model compiler for Ethos-U") + parser = argparse.ArgumentParser(prog="vela", description="Neural network model compiler for Arm Ethos-U NPUs") parser.add_argument("--version", action="version", version=__version__) parser.add_argument( "--api-version", action="version", version=api_version, help="Displays the version of the external API." @@ -200,7 +200,12 @@ def main(args=None): # set network nargs to be optional to allow the support-ops-report CLI option to be used standalone parser.add_argument( - "network", metavar="NETWORK", type=str, default=None, nargs="?", help="Filename of network to process" + "network", + metavar="NETWORK", + type=str, + default=None, + nargs="?", + help="Filename of the input TensorFlow Lite for Microcontrollers network", ) parser.add_argument( "--output-dir", type=str, default="output", help="Output directory to write files to (default: %(default)s)" @@ -34,7 +34,7 @@ mlw_module = Extension( setup( name="ethos-u-vela", use_scm_version=True, - description="Optimise TensorFlow Lite models for Ethos-U NPU.", + description="Neural network model compiler for Arm Ethos-U NPUs", long_description=long_description, long_description_content_type="text/markdown", url="https://git.mlplatform.org/ml/ethos-u/ethos-u-vela.git/", |