Use tvm.transform.Sequential

apache · Apr 15, 2020 · 569dc57 · 569dc57
1 parent 148cacb
commit 569dc57
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/python/relay/test_pass_partition_graph.py b/tests/python/relay/test_pass_partition_graph.py
@@ -1083,7 +1083,7 @@ def expected():
     mod = tvm.IRModule()
     mod["main"] = create_graph()
 
-    seq = transform.Sequential([
+    seq = tvm.transform.Sequential([
         transform.AnnotateTarget(target),
         transform.MergeCompilerRegions(),
         transform.PartitionGraph(),
@@ -1160,7 +1160,7 @@ def expected():
     mod = tvm.IRModule()
     mod["main"] = create_graph()
 
-    seq = transform.Sequential([
+    seq = tvm.transform.Sequential([
         transform.AnnotateTarget(target),
         transform.MergeCompilerRegions(),
         transform.PartitionGraph(),