You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by GitBox <gi...@apache.org> on 2020/10/26 17:31:39 UTC

[GitHub] [incubator-tvm] csullivan commented on a change in pull request #6748: [Relay] A set of utilities that allows a model to be run efficiently on tensorcores.

csullivan commented on a change in pull request #6748:
URL: https://github.com/apache/incubator-tvm/pull/6748#discussion_r512140426



##########
File path: python/tvm/relay/transform/transform.py
##########
@@ -386,6 +386,33 @@ def AlterOpLayout():
     return _ffi_api.AlterOpLayout()
 
 
+class LayoutConfig(object):
+    """A structure for customizing the ConvertLayout pass."""
+
+    current = None
+
+    def __init__(self, skip_layers=None):
+        self.skip_counter = 0
+        self.skip_layers = skip_layers if skip_layers is not None else []

Review comment:
       When have you found it useful to skip a specific layer of a given operator type / how do you envision it being used? Mainly for debugging and performance 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:
users@infra.apache.org