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
diff --git a/ethosu/vela/vela.py b/ethosu/vela/vela.py
index 05bd9ec..6fc5c47 100644
--- a/ethosu/vela/vela.py
+++ b/ethosu/vela/vela.py
@@ -187,7 +187,7 @@
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 @@
# 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)"
diff --git a/setup.py b/setup.py
index 1ab8dd5..d3a1013 100644
--- a/setup.py
+++ b/setup.py
@@ -34,7 +34,7 @@
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/",