diff --git a/python/tvm/driver/tvmc/transform.py b/python/tvm/driver/tvmc/transform.py index 08e7bdb93e858..253c624e6ed48 100644 --- a/python/tvm/driver/tvmc/transform.py +++ b/python/tvm/driver/tvmc/transform.py @@ -153,7 +153,6 @@ def layout_helper(layout): relay.transform.RemoveUnusedFunctions(), relay.transform.ConvertLayout(desired_layouts), relay.transform.FoldConstant(), - relay.transform.FoldScaleAxis(), ] )