mindspore 2.2.11__cp37-cp37m-manylinux1_x86_64.whl → 2.3.0rc1__cp37-cp37m-manylinux1_x86_64.whl

mindspore/rewrite/__init__.py

@@ -16,25 +16,14 @@
 MindSpore Rewrite package.
 This is an experimental python package that is subject to change or deletion.
 """
-from mindspore.rewrite.parsers.attribute_parser import g_attribute_parser
-from mindspore.rewrite.parsers.constant_parser import g_name_parser, g_num_parser, g_str_parser
-from mindspore.rewrite.parsers.container_parser import g_list_parser, g_tuple_parser
-from .parsers.module_parser import g_module_parser
-from .parsers.class_def_parser import g_classdef_parser
-from .parsers.function_def_parser import g_functiondef_parser
-from .parsers.arguments_parser import g_arguments_parser
-from .parsers.assign_parser import g_assign_parser
-from .parsers.if_parser import g_if_parser
-from .parsers.return_parser import g_return_parser
-from .parsers.for_parser import g_for_parser
 from .api.scoped_value import ScopedValue, ValueType
 from .api.symbol_tree import SymbolTree
 from .api.node import Node
 from .api.node_type import NodeType
 from .api.pattern_engine import PatternEngine, PatternNode, VarNode, Replacement
-from .api.tree_node_helper import TreeNodeHelper
 from .sparsify.sparsify import sparsify
 from .sparsify.utils import ArgType, SparseFunc
 
+
 __all__ = ["SymbolTree", "Node", "NodeType", "ScopedValue", "ValueType", "PatternEngine", "PatternNode", "VarNode",
-           "Replacement", "TreeNodeHelper", "sparsify", "ArgType", "SparseFunc"]
+           "Replacement"]

mindspore/rewrite/api/node.py

@@ -58,9 +58,7 @@ class Node:
 
         A node is corresponding to source code like:
 
-        .. code-block::
-
-            `targets` = self.`name`(*`args`, **`kwargs`)
+        ``targets = self.name(*args, **kwargs)``
 
         Args:
             cell (Cell): Cell-operator of this forward-layer.
@@ -91,7 +89,7 @@ class Node:
             >>> from mindspore.rewrite import SymbolTree, ScopedValue
             >>> import mindspore.nn as nn
             >>> # Define the network structure of LeNet5. Refer to
-            >>> # https://gitee.com/mindspore/docs/blob/r2.2/docs/mindspore/code/lenet.py
+            >>> # https://gitee.com/mindspore/docs/blob/r2.3.q1/docs/mindspore/code/lenet.py
             >>> net = LeNet5()
             >>> stree = SymbolTree.create(net)
             >>> node = stree.get_node("conv1")
@@ -116,8 +114,10 @@ class Node:
     def create_call_function(function: FunctionType, targets: List[Union[ScopedValue, str]],
                              args: List[ScopedValue] = None, kwargs: Dict[str, ScopedValue] = None) -> 'Node':
         """
-        Create a node that corresponds to a function call. The `function` object is saved into network, and used via
-        getting object from `self.` .
+        Create a node that corresponds to a function call.
+
+        Note:
+            The codes inside the function will not be parsed.
 
         Args:
             function (FunctionType): The function to be called.
@@ -144,7 +144,7 @@ class Node:
             >>> import mindspore.nn as nn
             >>> import mindspore.ops as ops
             >>> # Define the network structure of LeNet5. Refer to
-            >>> # https://gitee.com/mindspore/docs/blob/r2.2/docs/mindspore/code/lenet.py
+            >>> # https://gitee.com/mindspore/docs/blob/r2.3.q1/docs/mindspore/code/lenet.py
             >>> net = LeNet5()
             >>> stree = SymbolTree.create(net)
             >>> node = stree.get_node("conv1")
@@ -155,7 +155,7 @@ class Node:
             >>> print(new_node.get_node_type())
             NodeType.CallFunction
         """
-        Validator.check_value_type("function", function, [FunctionType, type], "create_call_function")
+        Validator.check_value_type("function", function, [FunctionType, type, type(abs)], "create_call_function")
         Validator.check_element_type_of_iterable("targets", targets, [ScopedValue, str], "create_call_function")
         if args is not None:
             Validator.check_element_type_of_iterable("args", args, [ScopedValue], "create_call_function")
@@ -165,7 +165,7 @@ class Node:
 
     @staticmethod
     def create_input(param_name: str, default: Optional[ScopedValue] = None) -> 'Node':
-    # pylint: disable=missing-function-docstring
+        # pylint: disable=missing-function-docstring
         Validator.check_value_type("param_name", param_name, [str], "Node")
         if default is not None:
             Validator.check_value_type("default", default, [ScopedValue], "Node")
@@ -184,7 +184,7 @@ class Node:
         Examples:
             >>> from mindspore.rewrite import SymbolTree
             >>> # Define the network structure of LeNet5. Refer to
-            >>> # https://gitee.com/mindspore/docs/blob/r2.2/docs/mindspore/code/lenet.py
+            >>> # https://gitee.com/mindspore/docs/blob/r2.3.q1/docs/mindspore/code/lenet.py
             >>> net = LeNet5()
             >>> stree = SymbolTree.create(net)
             >>> node = stree.get_node("conv2")
@@ -204,7 +204,7 @@ class Node:
         Examples:
             >>> from mindspore.rewrite import SymbolTree
             >>> # Define the network structure of LeNet5. Refer to
-            >>> # https://gitee.com/mindspore/docs/blob/r2.2/docs/mindspore/code/lenet.py
+            >>> # https://gitee.com/mindspore/docs/blob/r2.3.q1/docs/mindspore/code/lenet.py
             >>> net = LeNet5()
             >>> stree = SymbolTree.create(net)
             >>> node = stree.get_node("conv1")
@@ -212,15 +212,7 @@ class Node:
             >>> print([user.get_name() for user in users])
             ['relu']
         """
-        belong_symbol_tree: SymbolTreeImpl = self._node.get_belong_symbol_tree()
-        if belong_symbol_tree is None:
-            return []
-        unique_results = []
-        for node_user in belong_symbol_tree.get_node_users(self._node):
-            node = node_user[0]
-            if node not in unique_results:
-                unique_results.append(node)
-        return [Node(node_impl) for node_impl in unique_results]
+        return [Node(node_impl) for node_impl in self._node.get_users()]
 
     def set_arg(self, index: int, arg: Union[ScopedValue, str]):
         """
@@ -237,7 +229,7 @@ class Node:
         Examples:
             >>> from mindspore.rewrite import SymbolTree
             >>> # Define the network structure of LeNet5. Refer to
-            >>> # https://gitee.com/mindspore/docs/blob/r2.2/docs/mindspore/code/lenet.py
+            >>> # https://gitee.com/mindspore/docs/blob/r2.3.q1/docs/mindspore/code/lenet.py
             >>> net = LeNet5()
             >>> stree = SymbolTree.create(net)
             >>> node = stree.get_node("relu_3")
@@ -265,7 +257,6 @@ class Node:
                 which means use first output of `src_node` as new input.
 
         Raises:
-            RuntimeError: If `src_node` is not belong to current `SymbolTree`.
             TypeError: If `arg_idx` is not a `int` number.
             ValueError: If `arg_idx` is out of range.
             TypeError: If `src_node` is not a `Node` instance.
@@ -276,14 +267,14 @@ class Node:
         Examples:
             >>> from mindspore.rewrite import SymbolTree
             >>> # Define the network structure of LeNet5. Refer to
-            >>> # https://gitee.com/mindspore/docs/blob/r2.2/docs/mindspore/code/lenet.py
+            >>> # https://gitee.com/mindspore/docs/blob/r2.3.q1/docs/mindspore/code/lenet.py
             >>> net = LeNet5()
             >>> stree = SymbolTree.create(net)
             >>> src_node = stree.get_node("fc1")
             >>> dst_node = stree.get_node("relu_3")
             >>> dst_node.set_arg_by_node(0, src_node, 0)
             >>> print(dst_node.get_args())
-            [fc1]
+            [fc1_var]
         """
         Validator.check_value_type("arg_idx", arg_idx, [int], "Node")
         Validator.check_value_type("src_node", src_node, [Node], "Node")
@@ -316,7 +307,7 @@ class Node:
         Examples:
             >>> from mindspore.rewrite import SymbolTree
             >>> # Define the network structure of LeNet5. Refer to
-            >>> # https://gitee.com/mindspore/docs/blob/r2.2/docs/mindspore/code/lenet.py
+            >>> # https://gitee.com/mindspore/docs/blob/r2.3.q1/docs/mindspore/code/lenet.py
             >>> net = LeNet5()
             >>> stree = SymbolTree.create(net)
             >>> node = stree.get_node("conv1")
@@ -336,7 +327,7 @@ class Node:
         Examples:
             >>> from mindspore.rewrite import SymbolTree
             >>> # Define the network structure of LeNet5. Refer to
-            >>> # https://gitee.com/mindspore/docs/blob/r2.2/docs/mindspore/code/lenet.py
+            >>> # https://gitee.com/mindspore/docs/blob/r2.3.q1/docs/mindspore/code/lenet.py
             >>> net = LeNet5()
             >>> stree = SymbolTree.create(net)
             >>> node = stree.get_node("conv1")
@@ -363,7 +354,7 @@ class Node:
         Examples:
             >>> from mindspore.rewrite import SymbolTree
             >>> # Define the network structure of LeNet5. Refer to
-            >>> # https://gitee.com/mindspore/docs/blob/r2.2/docs/mindspore/code/lenet.py
+            >>> # https://gitee.com/mindspore/docs/blob/r2.3.q1/docs/mindspore/code/lenet.py
             >>> net = LeNet5()
             >>> stree = SymbolTree.create(net)
             >>> node = stree.get_node("conv1")
@@ -386,7 +377,7 @@ class Node:
         Examples:
             >>> from mindspore.rewrite import SymbolTree
             >>> # Define the network structure of LeNet5. Refer to
-            >>> # https://gitee.com/mindspore/docs/blob/r2.2/docs/mindspore/code/lenet.py
+            >>> # https://gitee.com/mindspore/docs/blob/r2.3.q1/docs/mindspore/code/lenet.py
             >>> net = LeNet5()
             >>> stree = SymbolTree.create(net)
             >>> node = stree.get_node("conv1")
@@ -403,14 +394,14 @@ class Node:
             SymbolTree, None if current node does not belong to any SymbolTree.
 
         Examples:
-        >>> from mindspore.rewrite import SymbolTree
-        >>> # Define the network structure of LeNet5. Refer to
-        >>> # https://gitee.com/mindspore/docs/blob/r2.2/docs/mindspore/code/lenet.py
-        >>> net = LeNet5()
-        >>> stree = SymbolTree.create(net)
-        >>> node = stree.get_node("conv1")
-        >>> print(type(node.get_symbol_tree()))
-        <class 'mindspore.rewrite.api.symbol_tree.SymbolTree'>
+            >>> from mindspore.rewrite import SymbolTree
+            >>> # Define the network structure of LeNet5. Refer to
+            >>> # https://gitee.com/mindspore/docs/blob/r2.3.q1/docs/mindspore/code/lenet.py
+            >>> net = LeNet5()
+            >>> stree = SymbolTree.create(net)
+            >>> node = stree.get_node("conv1")
+            >>> print(type(node.get_symbol_tree()))
+            <class 'mindspore.rewrite.api.symbol_tree.SymbolTree'>
         """
         from .symbol_tree import SymbolTree
         stree_impl = self._node.get_belong_symbol_tree()
@@ -418,7 +409,82 @@ class Node:
             return None
         return SymbolTree(stree_impl)
 
+    def get_sub_tree(self) -> 'SymbolTree':
+        """
+        Get the sub symbol tree stored in node with type of `NodeType.Tree` .
+        See :class:`mindspore.rewrite.NodeType` for details on node types.
+
+        Returns:
+            SymbolTree stored in Tree node.
+
+        Raises:
+            TypeError: If current node is not type of `NodeType.Tree` .
+            AttributeError: If no symbol tree is stored in Tree node.
+
+        Examples:
+        >>> import mindspore.nn as nn
+        >>> from mindspore.rewrite import SymbolTree
+        >>>
+        >>> class SubNet(nn.Cell):
+        ...     def __init__(self):
+        ...         super().__init__()
+        ...         self.relu = nn.ReLU()
+        ...
+        ...     def construct(self, x):
+        ...         x = self.relu(x)
+        ...         return x
+        ...
+        ... class Net(nn.Cell):
+        ...     def __init__(self):
+        ...         super().__init__()
+        ...         self.subnet = SubNet()
+        ...
+        ...     def construct(self, x):
+        ...         x = self.subnet(x)
+        ...         return x
+        >>>
+        >>> net = Net()
+        >>> stree = SymbolTree.create(net)
+        >>> node = stree.get_node("subnet")
+        >>> print(type(node.get_sub_tree()))
+        <class 'mindspore.rewrite.api.symbol_tree.SymbolTree'>
+        """
+        if self.get_node_type() != NodeType.Tree:
+            raise TypeError("For get_sub_tree, the type of node should be 'NodeType.Tree', "
+                            f"but got {self.get_node_type()}")
+        subtree: SymbolTreeImpl = self.get_handler().symbol_tree
+        if subtree is None:
+            raise AttributeError(
+                f"For get_sub_tree, no symbol tree is stroed in node {self.get_name()}.")
+        from .symbol_tree import SymbolTree
+        return SymbolTree(subtree)
+
     def get_kwargs(self) -> {str: ScopedValue}:
+        """
+        Get keyword arguments of current node.
+
+        Returns:
+            A dict of keyword arguments, where key is of type str, and value is of type ``ScopedValue`` .
+
+        Examples:
+            >>> from mindspore.rewrite import SymbolTree
+            >>> from mindspore import nn
+            >>>
+            >>> class ReLUNet(nn.Cell):
+            ...     def __init__(self):
+            ...         super().__init__()
+            ...         self.relu = nn.ReLU()
+            ...
+            ...     def construct(self, input):
+            ...         output = self.relu(x=input)
+            ...         return output
+            >>>
+            >>> net = ReLUNet()
+            >>> stree = SymbolTree.create(net)
+            >>> node = stree.get_node("relu")
+            >>> print(node.get_kwargs())
+            {'x': input}
+        """
         return self._node.get_kwargs()
 
     def set_attribute(self, key: str, value):
@@ -431,3 +497,23 @@ class Node:
     def get_attribute(self, key: str):
         Validator.check_value_type("key", key, [str], "Node attribute")
         return self._node.get_attribute(key)
+
+    # pylint: disable=missing-docstring
+    def get_arg_providers(self) -> dict:
+        arg_providers = {}
+        for arg_idx, providers in self._node.get_arg_providers().items():
+            arg_providers[arg_idx] = (Node(providers[0]), providers[1])
+        return arg_providers
+
+    # pylint: disable=missing-docstring
+    def get_target_users(self, index=-1) -> Union[dict, list]:
+        Validator.check_value_type("index", index, [int], "get_target_users")
+        if index == -1:
+            target_users = {}
+            for target_idx, users in self._node.get_target_users().items():
+                target_users[target_idx] = [(Node(user[0]), user[1]) for user in users]
+            return target_users
+        target_users = []
+        for users in self._node.get_target_users(index):
+            target_users.append((Node(users[0]), users[1]))
+        return target_users

mindspore/rewrite/api/pattern_engine.py

@@ -24,7 +24,6 @@ from mindspore import _checkparam as Validator
 from .node_type import NodeType
 from .node import Node
 from .symbol_tree import SymbolTree
-from .tree_node_helper import TreeNodeHelper
 
 
 class PatternNode:
@@ -363,8 +362,8 @@ class PatternEngine:
             if cur_node in visited:
                 continue
             if cur_node.get_node_type() == NodeType.Tree:
-                subtree = TreeNodeHelper.get_sub_tree(cur_node)
-                _ = self.apply(subtree)
+                subtree = cur_node.get_sub_tree()
+                self.apply(subtree)
                 visited.append(cur_node)
                 queue.extend(cur_node.get_users())
                 continue

mindspore/rewrite/api/scoped_value.py

@@ -27,15 +27,17 @@ class ValueType(Enum):
       and container-type of ValueType.
     """
 
-    # base type
+    # constant type
     ConstantValue = 0
     # container type
     TupleValue = 20
     ListValue = 21
     DictValue = 22
-    # other type
+    # variable type
     NamingValue = 40
     CustomObjValue = 41
+    # unsupported type
+    UnsupportedValue = 50
 
 
 class ScopedValue:
@@ -47,8 +49,8 @@ class ScopedValue:
 
     Args:
         arg_type (ValueType): A `ValueType` represents type of current value.
-        scope (str): A string represents scope of current value. Take "self.var1" as an example, `scope` of this
-            var1 is "self". Default: ``""`` .
+        scope (str, optional): A string represents scope of current value. Take "self.var1" as an example,
+            `scope` of this var1 is "self". Default: ``""`` .
         value: A handler represents value of current value. The type of value is corresponding to `arg_type`.
             Default: ``None`` .
     """
@@ -85,13 +87,12 @@ class ScopedValue:
             return cls(ValueType.TupleValue, "",
                        tuple(cls.create_variable_value(single_value) for single_value in value))
         if isinstance(value, list):
-            return cls(ValueType.ListValue, "", list(cls.create_variable_value(single_value) for single_value in value))
+            return cls(ValueType.ListValue, "",
+                       list(cls.create_variable_value(single_value) for single_value in value))
         if isinstance(value, dict):
-            for key, _ in value.items():
-                if not isinstance(key, str):
-                    raise TypeError("key should be str, got: ", type(key))
             return cls(ValueType.DictValue, "",
-                       dict((key, cls.create_variable_value(single_value)) for key, single_value in value.items()))
+                       dict((cls.create_variable_value(key),
+                             cls.create_variable_value(single_value)) for key, single_value in value.items()))
         return cls(ValueType.CustomObjValue, "", value)
 
     @classmethod
@@ -100,8 +101,8 @@ class ScopedValue:
         Create a naming `ScopedValue`. A `NamingValue` represents a reference to another variable.
 
         Args:
-            name: (str): A string represents the identifier of another variable.
-            scope: (str): A string represents the scope of another variable. Default: ``""`` .
+            name (str): A string represents the identifier of another variable.
+            scope (str, optional): A string represents the scope of another variable. Default: ``""`` .
 
         Returns:
             An instance of `ScopedValue`.
@@ -128,16 +129,16 @@ class ScopedValue:
 
         Args:
             names (List[str] or Tuple[str]): List or tuple of `str` represents names of referenced variables.
-            scopes (List[str] or Tuple[str]): List or tuple of `str` represents scopes of referenced variables.
-                Default: ``None`` .
+            scopes (List[str] or Tuple[str], optional): List or tuple of `str` represents scopes of
+                referenced variables. Default: ``None`` .
 
         Returns:
             An list of instance of `ScopedValue`.
 
         Raises:
-            RuntimeError: If the length of names is not equal to the length of scopes when scopes are not None.
             TypeError: If `names` is not `list` or `tuple` and name in `names` is not `str`.
             TypeError: If `scopes` is not `list` or `tuple` and scope in `scopes` is not `str`.
+            ValueError: If the length of names is not equal to the length of scopes when scopes are not None.
 
         Examples:
             >>> from mindspore.rewrite import ScopedValue
@@ -149,7 +150,7 @@ class ScopedValue:
         if scopes is not None:
             Validator.check_element_type_of_iterable("scopes", scopes, [str], "ScopedValue")
             if len(names) != len(scopes):
-                raise RuntimeError("Length of names should be equal to length of scopes")
+                raise ValueError("Length of names should be equal to length of scopes")
         result = []
         for index, name in enumerate(names):
             if scopes is not None:

mindspore/rewrite/api/symbol_tree.py

@@ -19,8 +19,7 @@ import mindspore as ms
 from mindspore.nn import Cell
 from mindspore import _checkparam as Validator
 from .node import Node
-from ..symbol_tree_builder import SymbolTreeBuilder
-from ..symbol_tree import Position, SymbolTree as SymbolTreeImpl
+from ..symbol_tree import Position, SymbolTreeBuilder, SymbolTree as SymbolTreeImpl
 
 ParamTypes = (int, str, float, bool, Node)
 MsDtypes = (ms.float16, ms.float32, ms.float64)
@@ -74,7 +73,7 @@ class SymbolTree:
         will parse the internal statements in the statement and generate corresponding nodes:
 
         - :class:`mindspore.nn.SequentialCell`
-        - Functions within classes
+        - Functions(Excludes Python built-in functions and third-party library functions)
         - Control flow statements, such as `if` statements
 
         Note:
@@ -91,16 +90,16 @@ class SymbolTree:
         The current rewrite module has the following syntax limitations:
 
         - Only networks of type :class:`mindspore.nn.Cell` are supported as input to the rewrite module.
-        - Parsing assignment statements with multiple output values is not currently supported.
-        - Parsing loop statements is not currently supported.
+        - Parsing one-line control flow syntax(e.g. one-line if-else, one-line for loop) is not currently supported.
         - Parsing decorator syntax is not currently supported.
-        - Parsing class variable syntax is not currently supported. If class variable uses external data,
-            the network after rewrite may be missing data.
         - Parsing local classes and embedded classes is not currently supported, that is, the definition
-            of classes need to be placed on the outermost layer.
+          of classes need to be placed on the outermost layer.
         - Parsing closure syntax is not currently supported, that is, the definition of out-of-class
-            functions need to be placed at the outermost layer.
+          functions need to be placed at the outermost layer.
         - Parsing lambda expression syntax is not currently supported.
+        - Parsing global variables is not currently supported, that is, global variables need to be converted to
+          class variables or local variables before they can be used.
+        - Parsing methods in the parent classes is not currently supported.
 
         For statements that do not support parsing, rewrite will generate nodes of type `NodeType.Python`
         for corresponding statements to ensure that the network after rewrite can run normally.
@@ -120,7 +119,7 @@ class SymbolTree:
         Examples:
             >>> from mindspore.rewrite import SymbolTree
             >>> # Define the network structure of LeNet5. Refer to
-            >>> # https://gitee.com/mindspore/docs/blob/r2.2/docs/mindspore/code/lenet.py
+            >>> # https://gitee.com/mindspore/docs/blob/r2.3.q1/docs/mindspore/code/lenet.py
             >>> net = LeNet5()
             >>> stree = SymbolTree.create(net)
             >>> print(type(stree))
@@ -164,12 +163,12 @@ class SymbolTree:
         Examples:
             >>> from mindspore.rewrite import SymbolTree
             >>> # Define the network structure of LeNet5. Refer to
-            >>> # https://gitee.com/mindspore/docs/blob/r2.2/docs/mindspore/code/lenet.py
+            >>> # https://gitee.com/mindspore/docs/blob/r2.3.q1/docs/mindspore/code/lenet.py
             >>> net = LeNet5()
             >>> stree = SymbolTree.create(net)
             >>> print([node.get_name() for node in stree.nodes()])
-            ['input_x', 'Expr', 'conv1', 'relu', 'max_pool2d', 'conv2', 'relu_1', 'max_pool2d_1',
-             'flatten', 'fc1', 'relu_2', 'fc2', 'relu_3', 'fc3', 'return']
+            ['input_x', 'conv1', 'relu', 'max_pool2d', 'conv2', 'relu_1', 'max_pool2d_1',
+             'unaryop_not', 'if_node', 'flatten', 'fc1', 'relu_2', 'fc2', 'relu_3', 'fc3', 'return_1']
         """
         Validator.check_value_type("all_nodes", all_nodes, [bool], "nodes")
         nodes = self._symbol_tree.all_nodes() if all_nodes else self._symbol_tree.nodes()
@@ -189,7 +188,7 @@ class SymbolTree:
         Examples:
             >>> from mindspore.rewrite import SymbolTree
             >>> # Define the network structure of LeNet5. Refer to
-            >>> # https://gitee.com/mindspore/docs/blob/r2.2/docs/mindspore/code/lenet.py
+            >>> # https://gitee.com/mindspore/docs/blob/r2.3.q1/docs/mindspore/code/lenet.py
             >>> net = LeNet5()
             >>> stree = SymbolTree.create(net)
             >>> node = stree.get_node('conv1')
@@ -203,7 +202,7 @@ class SymbolTree:
         return Node(node_impl)
 
     def get_inputs(self) -> List[Node]:
-        return [Node(node_impl) for node_impl in self._symbol_tree.get_inputs()]
+        return [Node(node_impl) for node_impl in self._symbol_tree.get_input_nodes()]
 
     def before(self, node: Union[Node, str]):
         """
@@ -217,12 +216,12 @@ class SymbolTree:
             A `Position` to indicate where to insert node.
 
         Raises:
-            TypeError: if `node` is not a `Node`.
+            TypeError: if `node` is not a Node or str.
 
         Examples:
             >>> from mindspore.rewrite import SymbolTree
             >>> # Define the network structure of LeNet5. Refer to
-            >>> # https://gitee.com/mindspore/docs/blob/r2.2/docs/mindspore/code/lenet.py
+            >>> # https://gitee.com/mindspore/docs/blob/r2.3.q1/docs/mindspore/code/lenet.py
             >>> net = LeNet5()
             >>> stree = SymbolTree.create(net)
             >>> for node in stree.nodes():
@@ -246,12 +245,12 @@ class SymbolTree:
             A `Position` to indicate where to insert node.
 
         Raises:
-            TypeError: If `node` is not a `Node`.
+            TypeError: If `node` is not a Node or str.
 
         Examples:
             >>> from mindspore.rewrite import SymbolTree
             >>> # Define the network structure of LeNet5. Refer to
-            >>> # https://gitee.com/mindspore/docs/blob/r2.2/docs/mindspore/code/lenet.py
+            >>> # https://gitee.com/mindspore/docs/blob/r2.3.q1/docs/mindspore/code/lenet.py
             >>> net = LeNet5()
             >>> stree = SymbolTree.create(net)
             >>> for node in stree.nodes():
@@ -277,7 +276,7 @@ class SymbolTree:
             An instance of Node being inserted.
 
         Raises:
-            RuntimeError: If `position` is not belong to current `SymbolTree`.
+            ValueError: If `position` is not belong to current `SymbolTree`.
             TypeError: If `position` is not a `Position`.
             TypeError: If `node` is not a `Node`.
 
@@ -285,7 +284,7 @@ class SymbolTree:
             >>> from mindspore.rewrite import SymbolTree, ScopedValue
             >>> import mindspore.nn as nn
             >>> # Define the network structure of LeNet5. Refer to
-            >>> # https://gitee.com/mindspore/docs/blob/r2.2/docs/mindspore/code/lenet.py
+            >>> # https://gitee.com/mindspore/docs/blob/r2.3.q1/docs/mindspore/code/lenet.py
             >>> net = LeNet5()
             >>> stree = SymbolTree.create(net)
             >>> node = stree.get_node("conv1")
@@ -309,12 +308,12 @@ class SymbolTree:
             An instance of `Node` being erased if node is in `SymbolTree` else None.
 
         Raises:
-            TypeError: The type of `node` is not Node.
+            TypeError: The type of `node` is not Node or str.
 
         Examples:
             >>> from mindspore.rewrite import SymbolTree
             >>> # Define the network structure of LeNet5. Refer to
-            >>> # https://gitee.com/mindspore/docs/blob/r2.2/docs/mindspore/code/lenet.py
+            >>> # https://gitee.com/mindspore/docs/blob/r2.3.q1/docs/mindspore/code/lenet.py
             >>> net = LeNet5()
             >>> stree = SymbolTree.create(net)
             >>> node = stree.get_node("conv1")
@@ -345,7 +344,6 @@ class SymbolTree:
             An instance of Node represents root of node_tree been replaced in.
 
         Raises:
-            RuntimeError: Old node is not isolated.
             TypeError: If `old_node` is not a `Node`.
             TypeError: If `new_nodes` is not a `list` or node in `new_nodes` is not a `Node`.
 
@@ -353,7 +351,7 @@ class SymbolTree:
             >>> from mindspore.rewrite import SymbolTree, ScopedValue
             >>> import mindspore.nn as nn
             >>> # Define the network structure of LeNet5. Refer to
-            >>> # https://gitee.com/mindspore/docs/blob/r2.2/docs/mindspore/code/lenet.py
+            >>> # https://gitee.com/mindspore/docs/blob/r2.3.q1/docs/mindspore/code/lenet.py
             >>> net = LeNet5()
             >>> stree = SymbolTree.create(net)
             >>> node = stree.get_node("conv1")
@@ -383,7 +381,7 @@ class SymbolTree:
 
         - **node type** (str): The type of node, refer to class:`mindspore.rewrite.NodeType` .
         - **name** (str): The name of node.
-        - **codes** (str): The source code statement corresponding to the node.
+        - **codes** (str): The code statement in the SymbolTree corresponding to the node.
         - **arg providers** (Dict[int, Tuple[str, int]]): The format is `{[idx, (n, k)]}` , which means the
           `idx` th parameter of the node is provided by the `k` th output of node `n` .
         - **target users** (Dict[int, List[Tuple[str, int]]]): The format is '{[idx, [(n, k)]]}' , which means
@@ -399,13 +397,14 @@ class SymbolTree:
         Examples:
             >>> from mindspore.rewrite import SymbolTree
             >>> # Define the network structure of LeNet5. Refer to
-            >>> # https://gitee.com/mindspore/docs/blob/r2.2/docs/mindspore/code/lenet.py
+            >>> # https://gitee.com/mindspore/docs/blob/r2.3.q1/docs/mindspore/code/lenet.py
             >>> net = LeNet5()
             >>> stree = SymbolTree.create(net)
             >>> stree.print_node_tabulate()
         """
         Validator.check_value_type("all_nodes", all_nodes, [bool], "print_node_tabulate")
-        self._symbol_tree.print_node_tabulate(all_nodes)
+        dump_str = self._symbol_tree.get_node_tabulate(all_nodes)
+        print(dump_str)
 
     def get_code(self) -> str:
         """
@@ -418,7 +417,7 @@ class SymbolTree:
         Examples:
             >>> from mindspore.rewrite import SymbolTree
             >>> # Define the network structure of LeNet5. Refer to
-            >>> # https://gitee.com/mindspore/docs/blob/r2.2/docs/mindspore/code/lenet.py
+            >>> # https://gitee.com/mindspore/docs/blob/r2.3.q1/docs/mindspore/code/lenet.py
             >>> net = LeNet5()
             >>> stree = SymbolTree.create(net)
             >>> codes = stree.get_code()
@@ -445,7 +444,7 @@ class SymbolTree:
         Examples:
             >>> from mindspore.rewrite import SymbolTree
             >>> # Define the network structure of LeNet5. Refer to
-            >>> # https://gitee.com/mindspore/docs/blob/r2.2/docs/mindspore/code/lenet.py
+            >>> # https://gitee.com/mindspore/docs/blob/r2.3.q1/docs/mindspore/code/lenet.py
             >>> net = LeNet5()
             >>> stree = SymbolTree.create(net)
             >>> new_net = stree.get_network()
@@ -479,3 +478,20 @@ class SymbolTree:
         """
         Validator.check_value_type("name", name, [str], "SymbolTree")
         return self._symbol_tree.unique_name(name)
+
+    def flatten_nodes(self, node, erase_nodes_after_return: bool = False):
+        Validator.check_value_type("node", node, [Node], "flatten_nodes")
+        Validator.check_value_type("erase_nodes_after_return", erase_nodes_after_return, [bool], "flatten_nodes")
+        return self._symbol_tree.flatten_nodes(node.get_handler(), erase_nodes_after_return)
+
+    def flatten_static_if_control_flow(self):
+        return self._symbol_tree.flatten_static_if_control_flow()
+
+    # pylint: disable=missing-docstring
+    def get_origin_network(self):
+        return self._symbol_tree.get_origin_network()
+
+    # pylint: disable=missing-docstring
+    def all_nodes(self, subtree_nodes: bool = True):
+        Validator.check_value_type("subtree_nodes", subtree_nodes, [bool], "all_nodes")
+        return [Node(n) for n in self._symbol_tree.all_nodes(subtree_nodes)]