add save load to jit.all (#27131)

PaddlePaddle · Sep 8, 2020 · 4c70e31 · 4c70e31
1 parent 8df5b4d
commit 4c70e31
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/paddle/fluid/dygraph/jit.py b/python/paddle/fluid/dygraph/jit.py
@@ -37,7 +37,7 @@
 
 __all__ = [
     'TracedLayer', 'declarative', 'dygraph_to_static_func', 'set_code_level',
-    'set_verbosity'
+    'set_verbosity', 'save', 'load', 'SaveLoadConfig'
 ]