leandron commented on a change in pull request #6302:
URL: https://github.com/apache/incubator-tvm/pull/6302#discussion_r484429345
##########
File path: python/tvm/driver/tvmc/common.py
##########
@@ -17,6 +17,74 @@
"""
Common utility functions shared by TVMC modules.
"""
+import argparse
+import re
+
+from tvm import relay
+from tvm import transform
class TVMCException(Exception):
"""TVMC Exception"""
+
+
+def convert_graph_layout(mod, desired_layout):
Review comment:
Good catch, I didn't know that it didn't check for layouts, but
apparently it won't add anything if it is not needed. So I wrote a few tests
and even raised #6410 for and issue that was happening.
The new version includes these fixes and tests.
----------------------------------------------------------------
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]
