You are viewing a plain text version of this content. The canonical link for it is here.

Posted to commits@tvm.apache.org by "PhilippvK (via GitHub)" <gi...@apache.org> on 2023/03/15 08:19:33 UTC

[GitHub] [tvm] PhilippvK commented on a diff in pull request #14272: [TVMC] Improve --desired-layouts functionality

PhilippvK commented on code in PR #14272:
URL: https://github.com/apache/tvm/pull/14272#discussion_r1136680106


##########
python/tvm/driver/tvmc/transform.py:
##########
@@ -210,9 +229,8 @@ def generate_transform_args(parser):
     # AlterLayout
     parser.add_argument(
         "--desired-layout",
-        choices=["NCHW", "NHWC"],
-        default=None,
-        help="Change the data layout of the whole graph.",
+        nargs="+",
+        help="Change the data/kernel layout of the graph. (i.e. NCHW or NHWC:HWIO)",

Review Comment:
   I have updated the help message.



-- 
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.

To unsubscribe, e-mail: commits-unsubscribe@tvm.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org