|
|
@ -89,6 +89,8 @@ from .io import save, load, load_program_state, set_program_state
|
|
|
|
from .dygraph.checkpoint import save_dygraph, load_dygraph
|
|
|
|
from .dygraph.checkpoint import save_dygraph, load_dygraph
|
|
|
|
from .dygraph.varbase_patch_methods import monkey_patch_varbase
|
|
|
|
from .dygraph.varbase_patch_methods import monkey_patch_varbase
|
|
|
|
Tensor = LoDTensor
|
|
|
|
Tensor = LoDTensor
|
|
|
|
|
|
|
|
enable_imperative = enable_dygraph
|
|
|
|
|
|
|
|
disable_imperative = disable_dygraph
|
|
|
|
|
|
|
|
|
|
|
|
__all__ = framework.__all__ + executor.__all__ + \
|
|
|
|
__all__ = framework.__all__ + executor.__all__ + \
|
|
|
|
trainer_desc.__all__ + transpiler.__all__ + \
|
|
|
|
trainer_desc.__all__ + transpiler.__all__ + \
|
|
|
@ -104,6 +106,8 @@ __all__ = framework.__all__ + executor.__all__ + \
|
|
|
|
'dygraph',
|
|
|
|
'dygraph',
|
|
|
|
'enable_dygraph',
|
|
|
|
'enable_dygraph',
|
|
|
|
'disable_dygraph',
|
|
|
|
'disable_dygraph',
|
|
|
|
|
|
|
|
'enable_imperative',
|
|
|
|
|
|
|
|
'disable_imperative',
|
|
|
|
'transpiler',
|
|
|
|
'transpiler',
|
|
|
|
'nets',
|
|
|
|
'nets',
|
|
|
|
'optimizer',
|
|
|
|
'optimizer',
|
|
|
|