gromero commented on a change in pull request #8086:
URL: https://github.com/apache/tvm/pull/8086#discussion_r636602209
##########
File path: python/tvm/driver/tvmc/compiler.py
##########
@@ -40,35 +40,42 @@
def add_compile_parser(subparsers):
""" Include parser for 'compile' subcommand """
- parser = subparsers.add_parser("compile", help="compile a model")
+ parser = subparsers.add_parser("compile", help="compile a model.")
parser.set_defaults(func=drive_compile)
parser.add_argument(
"--cross-compiler",
default="",
- help="the cross compiler to generate target libraries, e.g.
'aarch64-linux-gnu-gcc'",
+ help="the cross compiler to generate target libraries, e.g.
'aarch64-linux-gnu-gcc'.",
)
parser.add_argument(
"--desired-layout",
choices=["NCHW", "NHWC"],
default=None,
- help="change the data layout of the whole graph",
+ help="change the data layout of the whole graph.",
)
parser.add_argument(
"--dump-code",
metavar="FORMAT",
default="",
- help="comma separarated list of formats to export, e.g. 'asm,ll,relay'
",
+ help="comma separated list of possible formats to export the input
model, e.g. 'asm,ll,relay'.",
)
parser.add_argument(
"--model-format",
choices=frontends.get_frontend_names(),
- help="specify input model format",
+ help="specify input model format.",
+ )
+ parser.add_argument(
+ "-f",
+ "--output-format",
+ choices=["so", "mlf"],
+ default="so",
+ help="output format. Use 'so' for shared object or 'mlf' for Model
Library Format (only for µTVM targets).",
Review comment:
Done.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
