add save load to jit.all (#27131)

4c70e31a · Chen Weihang · GitHub · 8df5b4d6 · 4c70e31a
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

python/paddle/fluid/dygraph/jit.py python/paddle/fluid/dygraph/jit.py +1 -1

未找到文件。
--- a/python/paddle/fluid/dygraph/jit.py
+++ b/python/paddle/fluid/dygraph/jit.py
@@ -37,7 +37,7 @@ from paddle.fluid.wrapped_decorator import wrap_decorator

 __all__ = [
    'TracedLayer', 'declarative', 'dygraph_to_static_func', 'set_code_level',
-    'set_verbosity'
+    'set_verbosity', 'save', 'load', 'SaveLoadConfig'
 ]