[Doc] ConvertLayout - Call RemoveUnunsedFunctions.

4c961ddf · Animesh Jain · Wei Chen · e578777f · 4c961ddf
Commit 4c961ddf authored Feb 06, 2020 by Animesh Jain Committed by Wei Chen Feb 07, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 1 deletions

docs/dev/convert_layout.rst
+5 -1

No files found.
--- a/docs/dev/convert_layout.rst
+++ b/docs/dev/convert_layout.rst
@@ -226,7 +226,11 @@ ConvertLayout pass is extremely easy to use. The pass is not a part of default r
                                             dtype_dict=dtype_dict)

    # Convert the layout to NCHW
-    mod = relay.transform.ConvertLayout('NCHW')(mod)
+    # RemoveUnunsedFunctions is used to clean up the graph.
+    seq = relay.transform.Sequential([relay.transform.RemoveUnusedFunctions(),
+                                      relay.transform.ConvertLayout('NCHW')])
+    with relay.transform.PassContext(opt_level=3):
+        mod = seq(mod)

    # Call relay compilation
    with relay.build_config(opt_level=3):