mindspore 2.4.0__cp311-cp311-win_amd64.whl

mindspore/rewrite/node/node.py

@@ -0,0 +1,1428 @@
+# Copyright 2022 Huawei Technologies Co., Ltd
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ============================================================================
+"""Node class define of Rewrite. See detail in Node class docstring."""
+from typing import Optional, Union, List, Dict
+import ast
+import inspect
+from types import FunctionType
+import sys
+
+from mindspore.nn import Cell
+from mindspore.ops import Primitive
+from mindspore import log as logger
+from ..api.scoped_value import ScopedValue, ValueType
+from ..api.node_type import NodeType
+from ..common.namespace import is_subtree
+from ..common.error_log import error_str
+from ..ast_helpers import AstModifier, AstReplacer, AstConverter
+from ... import _checkparam as Validator
+
+
+if sys.version_info >= (3, 9):
+    import ast as astunparse # pylint: disable=reimported, ungrouped-imports
+else:
+    import astunparse
+
+
+class LocalPrim(Primitive):
+    """This class is used to indicate a local primitive instance"""
+    def __init__(self, prim_obj: type):
+        super().__init__("rewrite_local_prim")
+        self.prim_obj = prim_obj
+
+
+class Node:
+    """
+    Node is a data structure represents a source code line in network. For the most part, Node represents an operator
+    invoking in forward which could be an instance of Cell, an instance of Primitive or a callable method. Fields of
+    Node has different meaning in different type of node:
+
+    - CallCell: a call-cell node represents an assign statement whose value is a calling to cell in mindspore.
+      `targets` is corresponding to targets of ast.Assign which means return values of this cell-op. `args` and
+      `kwargs` are corresponding to args and keywords of ast.Call which mean arguments to invoke cell-op's forward
+      method. `func` is corresponding to func of call expression which means symbol of the cell-op.
+    - CallPrimitive: a call-primitive node represents an ast.Assign whose value is a calling to operator in mindspore.
+      `targets`, `args`, `kwargs` and `func_name` are as previous.
+    - CallMethod: a call-method node represents an ast.Assign whose value is a calling to python-method such as `len`.
+      `targets` is corresponding to targets of ast.Assign which means return values of this method. `func_name`
+      represents the string name of method. `args` and `kwargs` are corresponding to args and keywords to invoke the
+      method. When value of ast.Assign is an ast.Name or ast.Attribute, it means a simplest assign which would also be
+      mapped to CallMethod node whose `func_name` is "PassThrough".
+    - Python: a python node holds an ast-node which is not parsed. a python node means some python statement is not
+      supported by Rewrite or ignored by Rewrite. `targets`, `args`, `kwargs` and `func_name` are don't-care.
+    - Input: an input node represents an input of current network which also a parameter of forward method of Cell.
+      `targets` is corresponding to arg-name of parameter of forward function. `args` means default-value of parameter
+      of forward function. `kwargs` and `func_name` are don't-care.
+    - Output: an output node represents the output of current network which is corresponding to return statement of
+      forward method of Cell. `args` represents return values. `func_name` are always be "return". `targets` and
+      `kwargs` are don't-care.
+    - Tree: a tree node represents a sub-network call in current network. A sub-network is also a Cell in mindspore, so
+      `targets`, `args`, `kwargs` and `func_name` are same as a call-cell node. `symbol_tree` is a handler of a
+      SymbolTree instance.
+    """
+
+    def __init__(self, node_type: NodeType, ast_node: Optional[ast.AST], targets: [ScopedValue],
+                 func_name: Optional[ScopedValue], args: List[ScopedValue], kwargs: Dict[str, ScopedValue], name: str,
+                 instance):
+        """
+        Constructor of Node. Rewrite recommend invoking class method of Node to instantiate an instance of Node such
+        as `create_call_op`, `create_call_method`, `create_python_node`, `create_input_node` and
+        `create_output_node`, etc. rather than invoking constructor of Node directly.
+
+        Args:
+            node_type (NodeType): A NodeType as type of Node.
+            ast_node (ast.AST, optional): An instance of ast.AST represents corresponding node in ast. `ast_node` should
+                not be None except when node type is Unknown.
+            targets (list[ScopedValue]): A list of instance of ScopedValue. See detail in docstring of Node class.
+            func_name (ScopedValue, optional): An instance of ScopedValue. See detail in docstring of Node class.
+            args (list[ScopedValue]): A list of instance of ScopedValue. See detail in docstring of Node class.
+            kwargs (Dict[str, ScopedValue]): A list of instance of ScopedValue. See detail in docstring of Node class.
+            name (str): A string represents name of node. Name of node will be unique when inserted into SymbolTree.
+                Name of node also used as field name in network class.
+            instance: Object in network corresponding to this node.
+        """
+        self._node_type: NodeType = node_type
+        self._ast_node: Optional[ast.AST] = ast_node
+        self._attribute: {str, object} = {}
+        if node_type in (NodeType.CallModule, NodeType.CallCell, NodeType.CallPrimitive):
+            self._attribute = Node._get_cell_or_prim_op_attribute(instance)
+        self._instance = instance
+        self._name = name
+        self._func_name: Optional[ScopedValue] = func_name
+        self._targets: [ScopedValue] = targets if targets is not None else []
+        self._args_num = len(args) if args is not None else 0
+        self._kwargs_num = len(kwargs) if kwargs is not None else 0
+        self._normalized_args_keys = []  # for saving args' order
+        self._normalized_args = self._get_normalized_args(args, kwargs)
+        # position in graph nodes list
+        # it will affect code-order of python code
+        self._prev: Optional[Node] = None
+        self._next: Optional[Node] = None
+        # A handler of SymbolTree current node belonging to
+        self._belong_tree = None
+        # A handler of NodeManager current node belonging to
+        self._node_manager = None
+        # A dict that records which target of which Node current Node's argument come from
+        self._arg_providers: {int: (Node, int)} = {}
+        # A dict that records which argument of which Node uses current Node's target
+        self._target_users: {int: [(Node, int)]} = {}
+        # Indicate this node represent a class type object, e.g. abs_ops = _get_cache_prim(P.Abs)
+        self._type_cls = None
+        # Indicate this node represent the initialize of a class type, e.g. abs_inst = P.Abs()
+        self._init_cls = None
+
+    @classmethod
+    def create_call_method(cls, ast_node: Optional[ast.AST], targets: [Union[ScopedValue, str]],
+                           func_name: Union[ScopedValue, str], args: [ScopedValue] = None,
+                           kwargs: {str: ScopedValue}=None, name: str = ""):
+        """
+        Class method of Node. Instantiate an instance of node whose type is CallCell. A CallCell node represents an
+        invoking to cell-op.
+
+        Args:
+            ast_node ([ast.AST, optional]): An instance of ast.AST represents corresponding node in ast. `ast_node`
+                should not be None currently.
+            targets (list[ScopedValue]): A list of instance of ScopedValue. See detail in docstring of Node class.
+            func_name ([ScopedValue, optional]): An instance of ScopedValue. See detail in docstring of Node class.
+            args (list[ScopedValue]): A list of instance of ScopedValue. See detail in docstring of Node class.
+            kwargs (dict{str: ScopedValue}): A list of instance of ScopedValue. See detail in docstring of Node class.
+            name (str): A string represents name of node. Name of node will be unique when inserted into SymbolTree.
+                Name of node also used as field name in network class.
+        """
+        if args is None:
+            args = []
+        if kwargs is None:
+            kwargs = {}
+        if isinstance(func_name, str):
+            func_name = ScopedValue.create_naming_value(func_name)
+        new_targets = Node._handle_targets(targets)
+        if ast_node is None:
+            raise RuntimeError("Input ast_node is None")
+        return cls(NodeType.CallMethod, ast_node, new_targets, func_name, args, kwargs, name, None)
+
+    @classmethod
+    def create_python_node(cls, ast_node: ast.AST, name: str = "", instance=None):
+        """
+        Class method of Node. Instantiate an instance of node whose type is Python. A Python node represents some python
+        statement is not supported by Rewrite or ignored by Rewrite.
+
+        Args:
+            ast_node (ast.AST): An instance of ast.AST represents corresponding node in ast.
+            name (str): A string represents name of node. Name of node will be unique when inserted into SymbolTree.
+                Name of node also used as field name in network class.
+            instance: An object corresponding to this node in network.
+        """
+        return cls(NodeType.Python, ast_node, None, None, [], {}, name, instance)
+
+    @classmethod
+    def create_input_node(cls, ast_node: Optional[ast.AST], arg_name: str, default: Optional[ScopedValue] = None,
+                          name: str = ""):
+        """
+        Class method of Node. Instantiate an instance of node whose type is Input. An Input node represents input of
+        SymbolTree which is corresponding to parameters of forward function.
+
+        Args:
+            ast_node (ast.AST): An instance of ast.AST represents corresponding node in ast.
+            arg_name (str): A string represents name of parameter.
+            default ([ScopedValue, optional]): An instance of ScopedValue represents default value of parameter.
+            name (str): A string represents name of node. Name of node will be unique when inserted into SymbolTree.
+                Name of node also used as field name in network class.
+        """
+        target = ScopedValue.create_naming_value(arg_name)
+        if default is None:
+            args = []
+        else:
+            args = [default]
+        if ast_node is None:
+            ast_node = ast.arg(arg_name, annotation="")
+        return cls(NodeType.Input, ast_node, [target], None, args, {}, name, None)
+
+    @classmethod
+    def create_output_node(cls, ast_node: ast.AST, return_value: [ScopedValue], name: str = "return"):
+        """
+        Class method of Node. Instantiate an instance of node whose type is Output. An Output node represents output of
+        SymbolTree which is corresponding to return statement of forward function.
+
+        Args:
+            ast_node (ast.AST): An instance of ast.AST represents corresponding node in ast.
+            return_values (list[str]): A list of string represents name of return values.
+            name (ScopedValue): An instance of ScopedValue represents name of node.
+        """
+        return cls(NodeType.Output, ast_node, None, ScopedValue.create_naming_value("return"), return_value, {},
+                   name, None)
+
+    @classmethod
+    def create_mathops_node(cls, ast_node: ast.AST, targets: [ScopedValue],
+                            op_type: ScopedValue, args: [ScopedValue], name: str = ""):
+        """
+        Class method of Node. Instantiate an instance of node whose type is `MathOps` .
+        A mathops node is used to represent a node with mathematical operations, such as
+        `y = a + b` , `y = not a` , `y = 0 < a < 1`, `y = a or b` , etc.
+
+        Args:
+            ast_node ([ast.AST, optional]): An instance of ast.AST represents corresponding node in ast. The type of
+                node is ast.Assign, and the type of ast_node.value is one of ast.BinOp, ast.UnaryOp, ast.BoolOp and
+                ast.Compare.
+            targets (list[ScopedValue]): Targets of mathematical operations. A list of instance of `ScopedValue`.
+                See detail in docstring of Node class.
+            op_type (ScopedValue): The type of ast_node.value saved by string. A ScopedValue with NamingValue type.
+            args (list[ScopedValue]): Values participating in the mathematical operations. All values are saved
+                sequentially in the list.
+            name (str): A string represents name of node. Name of node will be unique when inserted into `SymbolTree`.
+                Name of node also used as field name in network class. The format of mathops node name
+                is 'AstNodeName_AstOpName_n'.
+        """
+        return cls(NodeType.MathOps, ast_node, targets, op_type, args, None, name, None)
+
+    @staticmethod
+    def _create_call_function(function: FunctionType, targets: [Union[ScopedValue, str]], args: [ScopedValue] = None,
+                              kwargs: {str: ScopedValue}=None):
+        """
+        Create a node that corresponds to a function call.
+
+       Args:
+            function (FunctionType): The function to be called.
+            targets (list[str]): indicates output names. Used as targets of an assign statement in source code.
+            args (list[ScopedValue]): Indicate input names. Used as args of a call expression of an assign statement in
+                source code. Default: ``None`` , which indicates the `function` has no args inputs.
+            kwargs (dict): Type of key must be `str` and type of value must be `ScopedValue`.
+                Indicate keyword input names. Used as kwargs of a call expression of an assign statement in source
+                code. Default: ``None`` , which indicates the `function` has no kwargs inputs.
+
+        Returns:
+            An instance of `Node`.
+        """
+        if args is None:
+            args = []
+        if kwargs is None:
+            kwargs = {}
+        targets = Node._handle_targets(targets)
+        func_name = function.__name__
+        func_scope_name = ScopedValue.create_naming_value(func_name)
+        node = Node.inner_create_call_function(func_name, None, func_scope_name, function, targets, args, kwargs)
+        return node
+
+    @classmethod
+    def inner_create_call_function(cls, node_name: str, ast_node: ast.Assign, func_name: ScopedValue, func_obj: object,
+                                   targets: List[ScopedValue], args: List[ScopedValue], kwargs: Dict[str, ScopedValue]):
+        '''
+        Instantiate an instance of node whose type is `CallFunction`.
+
+        Args:
+            node_name (str): Name of node.
+            func_name (ScopedValue): Name of function.
+            ast_node ([ast.AST, optional]): An instance of ast.AST represents corresponding node in ast.
+            func_obj (Object): An object of function. See detail in docstring of Node class.
+            targets (List[ScopedValue]): A list of instance of `ScopedValue`. See detail in docstring of Node class.
+            args (List[ScopedValue]): A list of instance of `ScopedValue`. See detail in docstring of Node class.
+            kwargs (Dict[str, ScopedValue]): A list of instance of `ScopedValue`. See detail in docstring of `Node`
+                class.
+        '''
+        from . import CallFunction
+        # create CallFunction node
+        return CallFunction(targets, func_name, args, kwargs, node_name, ast_node, None, None, func_obj, False)
+
+    @staticmethod
+    def create_call_op(op: Union[Cell, Primitive], ast_node: Optional[ast.AST], targets: [Union[ScopedValue, str]],
+                       args: [ScopedValue] = None, kwargs: {str: ScopedValue}=None, node_name: str = "",
+                       is_sub_net: bool = False):
+        """
+        Static method of Node. Instantiate an instance of node whose type is `CallCell` or `CallPrimitive`.
+        If op is custom defined, it is treated by TreeNode.
+        A `CallCell` node represents an invoking to cell-op.
+        A `CallPrimitive` node represents an invoking to primitive-op.
+
+        Args:
+            op (Union[Cell, Primitive]): An instance of `Cell` or `Primitive` corresponding to this node.
+            ast_node ([ast.AST, optional]): An instance of ast.AST represents corresponding node in ast.
+            targets (list[ScopedValue]): A list of instance of `ScopedValue`. See detail in docstring of Node class.
+            args (list[ScopedValue]): A list of instance of `ScopedValue`. See detail in docstring of Node class.
+            kwargs (dict{str: ScopedValue}): A list of instance of `ScopedValue`. See detail in docstring of `Node`
+                class.
+            node_name (str): A string represents name of node. Name of node will be unique when inserted into
+                `SymbolTree`. Name of node also used as field name in network class.
+            is_sub_net (bool): Indicate that is `cell` a network. If `is_sub_net` is true, Rewrite will try to parse the
+                `cell` to a TreeNode, else a CallCell Node. Default is a False.
+        """
+        Validator.check_value_type("op", op, [Cell, Primitive], "Node")
+        if ast_node is not None:
+            Validator.check_value_type("ast_node", ast_node, [ast.AST], "Node")
+        Validator.check_element_type_of_iterable("targets", targets, [ScopedValue, str], "Node")
+        if args is not None:
+            Validator.check_element_type_of_iterable("args", args, [ScopedValue], "Node")
+        if kwargs is not None:
+            Validator.check_element_type_of_dict("kwargs", kwargs, [str], [ScopedValue], "Node")
+        if args is None:
+            args = []
+        if kwargs is None:
+            kwargs = {}
+        Validator.check_value_type("node_name", node_name, [str], "Node")
+        new_targets = Node._handle_targets(targets)
+        if isinstance(node_name, str):
+            func_name = ScopedValue.create_naming_value(node_name)
+        else:
+            func_name = node_name
+        if is_sub_net and is_subtree(op):
+            from ..symbol_tree import SymbolTreeBuilder
+            stb = SymbolTreeBuilder(op)
+            stree = stb.build()
+            replacer = AstReplacer(stree.get_class_ast())
+            replacer.replace_all(stree.get_ori_cls_name(), stree.get_opt_cls_name())
+            return TreeNode.create_tree_node(stree, ast_node, new_targets, func_name, args, kwargs, node_name, op)
+
+        return Node.create_call_buildin_op(op, ast_node, new_targets, func_name, args, kwargs, node_name)
+
+    @classmethod
+    def create_call_buildin_op(cls, op: Union[Cell, Primitive], ast_node: Optional[ast.AST], targets: [ScopedValue],
+                               func_name: ScopedValue, args: [ScopedValue] = None, kwargs: {str: ScopedValue}=None,
+                               node_name: str = ""):
+        """
+        Class method of Node. Instantiate an instance of node whose type is `CallCell` or `CallPrimitive`.
+        A `CallCell` node represents an invoking to cell-op.
+        A `CallPrimitive` node represents an invoking to primitive-op.
+
+        Args:
+            op (Union[Cell, Primitive]): An instance of `Cell` or `Primitive` corresponding to this node.
+            ast_node ([ast.AST, optional]): An instance of ast.AST represents corresponding node in ast.
+            targets (list[ScopedValue]): A list of instance of `ScopedValue`. See detail in docstring of Node class.
+            func_name ([ScopedValue, optional]): An instance of `ScopedValue`. See detail in docstring of Node class.
+            args (list[ScopedValue]): A list of instance of `ScopedValue`. See detail in docstring of Node class.
+            kwargs (dict{str: ScopedValue}): A list of instance of `ScopedValue`. See detail in docstring of `Node`
+                class.
+            node_name (str): A string represents name of node. Name of node will be unique when inserted into
+                `SymbolTree`. Name of node also used as field name in network class.
+        """
+
+        if not isinstance(op, (Cell, Primitive)):
+            raise ValueError("Input op is not a buildin op(Cell or Primitive): ", type(op))
+        if isinstance(op, Cell):
+            node_type = NodeType.CallCell
+        else:
+            node_type = NodeType.CallPrimitive
+        return cls(node_type, ast_node, targets, func_name, args, kwargs, node_name, op)
+
+    @staticmethod
+    def _get_construct_arg_names(parameters):
+        """
+        Static method of `Node`. Get parameters' names of the construct function.
+
+        Args:
+            parameters (MappingProxyType): An ordered mapping of parameters' names to the corresponding Parameter
+                objects.
+
+        Raises:
+            RuntimeError: Invalid parameter kind.
+
+        Returns:
+            - arg_names, Parameters' names, contain parameters of types in [POSITIONAL_ONLY, POSITIONAL_OR_KEYWORD].
+            - var_positional_name, Name of VAR_POSITIONAL parameters.
+            - var_keyword_name, Name of VAR_KEYWORD parameters.
+        """
+        position_only_names: [str] = []
+        positional_or_keyword_names: [str] = []
+        var_positional_name = None
+        keyword_only_names: [str] = []
+        var_keyword_name = None
+        for name, para in parameters.items():
+            if para.kind == inspect.Parameter.POSITIONAL_ONLY:  # parameters which appear before a '/'
+                position_only_names.append(name)
+            elif para.kind == inspect.Parameter.POSITIONAL_OR_KEYWORD:  # parameters which appear before '*' or '*args'
+                positional_or_keyword_names.append(name)
+            elif para.kind == inspect.Parameter.VAR_POSITIONAL:  # corresponds to a '*args'
+                var_positional_name = name
+            elif para.kind == inspect.Parameter.KEYWORD_ONLY:  # parameters which appear after '*' and before '**'
+                keyword_only_names.append(name)
+            elif para.kind == inspect.Parameter.VAR_KEYWORD:  # corresponds to a '**kwargs'
+                var_keyword_name = name
+            else:
+                raise RuntimeError("invalid parameter kind:", para.kind)
+        if "self" in position_only_names:
+            position_only_names.remove("self")
+        if "self" in positional_or_keyword_names:
+            positional_or_keyword_names.remove("self")
+        names = (position_only_names, positional_or_keyword_names, var_positional_name, keyword_only_names,
+                 var_keyword_name)
+        return names
+
+    @staticmethod
+    def _map_args_names(names: tuple, args: [ScopedValue], kwargs: {str: ScopedValue},
+                        normalized_args_keys: [str], normalized_args: {str: ScopedValue}):
+        """
+        Fill in normalized_args according to the order of parameters of construct func.
+
+        Args:
+            names (tuple): Parameters' name got from construct func.
+            args (list[ScopedValue]): A list of instance of ScopedValue. See detail in docstring of Node class.
+            kwargs (dict{str: ScopedValue}): A list of instance of ScopedValue. See detail in docstring of Node class.
+            normalized_args (dict{str: ScopedValue}): The normalized args to be filled.
+
+        Raises:
+            RuntimeError: Input args are invalid.
+            RuntimeError: Arg name already exist in kwargs.
+            RuntimeError: Input kwargs invalid.
+        """
+        position_only_names, positional_or_keyword_names, var_positional_name, keyword_only_names, var_keyword_name = \
+            names
+        for arg_index, arg in enumerate(args):
+            if arg_index < len(position_only_names):
+                arg_key = position_only_names[arg_index]
+            elif arg_index < len(position_only_names) + len(positional_or_keyword_names):
+                arg_key = positional_or_keyword_names[arg_index - len(position_only_names)]
+            elif var_positional_name:
+                arg_key = "{}_{}".format(var_positional_name, arg_index)
+            else:
+                raise RuntimeError("Input args are invalid.")
+
+            if arg_key in kwargs.keys():
+                raise RuntimeError("Arg name already exist in kwargs.")
+            normalized_args[arg_key] = arg
+            normalized_args_keys.append(arg_key)
+
+        # add kwargs according to parameters' order
+        parameters_order: [str] = []
+        parameters_order.extend(position_only_names)
+        parameters_order.extend(positional_or_keyword_names)
+        parameters_order.append(var_keyword_name)
+        parameters_order.extend(keyword_only_names)
+        parameters_order.append(var_keyword_name)
+
+        sorted_kwargs = []
+        var_keyword_count = len(parameters_order)
+        for arg_key, value in kwargs.items():
+            if arg_key not in parameters_order and not var_keyword_name:
+                raise RuntimeError("Input kwargs invalid.")
+            if arg_key in parameters_order:
+                sorted_kwargs.append([arg_key, value, parameters_order.index(arg_key)])
+            else:
+                sorted_kwargs.append([arg_key, value, var_keyword_count])
+                var_keyword_count += 1
+
+        sorted_kwargs.sort(key=lambda x: x[2])
+        for sorted_kwarg in sorted_kwargs:
+            normalized_args[sorted_kwarg[0]] = sorted_kwarg[1]
+            normalized_args_keys.append(sorted_kwarg[0])
+
+    @staticmethod
+    def _handle_custom_obj_in_args(args: [ScopedValue]) -> [ScopedValue]:
+        """
+        Convert CustomObjValue type argument to NamingValue type argument.
+
+        Args:
+            args (list[ScopedValue]): A list of instance of ScopedValue to be converted.
+
+        Returns:
+            A list of instance of ScopedValue which have been converted.
+        """
+        result = []
+        for arg in args:
+            if not isinstance(arg, ScopedValue):
+                raise TypeError("arg should be ScopedValue, got: ", type(arg))
+            if arg.type == ValueType.CustomObjValue:
+                logger.info("custom-object exist in args, should be replace before compile")
+                result.append(ScopedValue.create_naming_value("custom-object", "self"))
+            else:
+                result.append(arg)
+        return result
+
+    @staticmethod
+    def _handle_custom_obj_in_kwargs(kwargs: {str: ScopedValue}) -> {str: ScopedValue}:
+        """
+        Convert CustomObjValue type argument to NamingValue type argument.
+
+        Args:
+            kwargs (dict{str: ScopedValue}): A str to instance of ScopedValue dict whose value to be converted.
+
+        Returns:
+            A str to instance of ScopedValue dict whose value has be converted.
+        """
+        result: {str, ScopedValue} = {}
+        for arg, value in kwargs.items():
+            if not isinstance(value, ScopedValue):
+                raise TypeError("value should be ScopedValue, got: ", type(value))
+            if value.type == ValueType.CustomObjValue:
+                result[arg] = ScopedValue.create_naming_value("custom-object", "self")
+            else:
+                result[arg] = value
+        return result
+
+    @staticmethod
+    def _handle_targets(targets: [Union[ScopedValue, str]]) -> [ScopedValue]:
+        """
+        Normalize targets to be a list of ScopedValue. If target is a str, it will be converted to NamingValue type
+        ScopedValue.
+
+        Args:
+            targets (Union[ScopedValue, str]]): A list whose element could be a ScopedValue or a str to be normalized.
+
+        Returns:
+            A list of instance of ScopedValue which have been converted.
+        """
+        if not isinstance(targets, list):
+            raise TypeError("targets should be list, got: ", type(targets))
+        results = []
+        for target in targets:
+            if isinstance(target, str):
+                scope = ""
+                name = target
+                if target.count('.') > 0:
+                    scope, name = target.rsplit('.', 1)
+                results.append(ScopedValue.create_naming_value(name, scope))
+            elif isinstance(target, ScopedValue):
+                results.append(target)
+            else:
+                raise RuntimeError("Invalid symbol type: ", target)
+        return results
+
+    @staticmethod
+    def _get_cell_or_prim_op_attribute(obj) -> dict:
+        """
+        Find attributes of cell-op or primitive-op.
+
+        Args:
+            obj: A cell-op or a primitive-op.
+
+        Returns:
+            A dict represents attributes of input 'obj'.
+        """
+        attributes = {}
+        if obj is None:
+            return attributes
+        for k, v in obj.__dict__.items():
+            if k.startswith("_"):
+                continue
+            attributes[k] = v
+        attributes["cls"] = obj.__class__
+        return attributes
+
+    def get_type_cls(self) -> object:
+        """Get the class type object this node represented, e.g. abs_ops = _get_cache_prim(P.Abs)"""
+        return self._type_cls
+
+    def set_type_cls(self, x):
+        """Set the class type object this node represented, e.g. abs_ops = _get_cache_prim(P.Abs)"""
+        self._type_cls = x
+
+    def get_init_cls(self) -> object:
+        """Get the class type object initialized by this node, e.g. abs_inst = P.Abs()"""
+        return self._init_cls
+
+    def set_init_cls(self, x):
+        """Set the class type object initialized by this node"""
+        self._init_cls = x
+
+    def get_prev(self) -> 'Node':
+        """
+        Get previous node of current node in source code order.
+
+        Returns:
+            An instance of Node as previous node.
+        """
+        return self._prev
+
+    def get_next(self) -> 'Node':
+        """
+        Get next node of current node in source code order.
+
+        Returns:
+            An instance of Node as next node.
+        """
+        return self._next
+
+    def set_prev(self, node: 'Node'):
+        """
+        Set previous node of current node.
+
+        Args:
+            node (Node): Node to be set as previous node of current node.
+        """
+        self._prev = node
+
+    def set_next(self, node: 'Node'):
+        """
+        Set next node of current node.
+
+        Args:
+            node (Node): Node to be set as next node of current node.
+        """
+        self._next = node
+
+    def get_ast(self) -> Optional[ast.AST]:
+        """
+        Getter of _ast_node.
+
+        Returns:
+            An instance of ast.AST if self._ast_node if not None else None.
+        """
+        return self._ast_node
+
+    def set_ast(self, ast_node: ast.AST):
+        """
+        Setter of _ast_node.
+
+        Args:
+            ast_node (ast.AST): An instance of ast.AST as new value for _ast_node.
+        """
+        if not isinstance(ast_node, ast.AST):
+            raise TypeError("ast_node should be ast.AST, got: ", type(ast_node))
+        self._ast_node = ast_node
+
+    def get_belong_symbol_tree(self):
+        """Get the symbol tree to which node belongs."""
+        return self._belong_tree
+
+    def set_belong_symbol_tree(self, symbol_tree):
+        """Set the symbol tree to which node belongs."""
+        self._belong_tree = symbol_tree
+
+    def get_node_manager(self):
+        """Get the NodeManager current node belongs to."""
+        return self._node_manager
+
+    def set_node_manager(self, node_manager):
+        """Set NodeManager current node belongs."""
+        self._node_manager = node_manager
+
+    def isolate(self):
+        """Link prev node to next node and isolate node from source code order list."""
+        origin_prev: Optional[Node] = self.get_prev()
+        origin_next: Optional[Node] = self.get_next()
+        if origin_prev is not None:
+            origin_prev.set_next(origin_next)
+        if origin_next is not None:
+            origin_next.set_prev(origin_prev)
+        self.set_prev(None)
+        self.set_next(None)
+
+    def insert_before(self, node: 'Node'):
+        """
+        Insert a node before current node in source code list. Note that topological order is not determined here.
+
+        Args:
+            node (Node): An instance of node to be inserted in.
+        """
+        node.isolate()
+        origin_prev: Optional[Node] = self.get_prev()
+        if origin_prev is not None:
+            origin_prev.set_next(node)
+        node.set_prev(origin_prev)
+        node.set_next(self)
+        self.set_prev(node)
+
+    def insert_after(self, node: 'Node'):
+        """
+        Insert a node after current node in source code list. Note that topological order is not determined here.
+
+        Args:
+            node (Node): An instance of node to be inserted in.
+        """
+        node.isolate()
+        origin_next: Optional[Node] = self.get_next()
+        self.set_next(node)
+        node.set_prev(self)
+        node.set_next(origin_next)
+        if origin_next is not None:
+            origin_next.set_prev(node)
+
+    def get_inputs(self) -> ['Node']:
+        """
+        Get input nodes of current node in topological order.
+
+        Returns:
+            A list of instances of Node as input nodes.
+        """
+        inputs = []
+        for arg_provider in self.get_arg_providers().values():
+            if not arg_provider:
+                continue
+            inputs.append(arg_provider[0])
+        return inputs
+
+    def get_users(self) -> ['Node']:
+        """
+        Get user nodes of current node in topological order.
+
+        Returns:
+            A list of instances of Node as user nodes.
+        """
+        users = []
+        for target_users in self.get_target_users().values():
+            if not target_users:
+                continue
+            for (user, _) in target_users:
+                if user not in users:
+                    users.append(user)
+        return users
+
+    def get_targets(self) -> [ScopedValue]:
+        """
+        Getter of _targets.
+
+        - When node_type of current node is CallCell or CallPrimitive or CallMethod or Tree, `targets` are strings
+          represents invoke result of the cell-op or primitive-op or function-call which are corresponding to targets of
+          ast.Assign.
+        - When node_type of current node is Input, `targets` should have only one element which is a string represents
+          name of parameter of function.
+        - When node_type of current node is Python or Output, `targets` are don't-care.
+
+        Returns:
+            A list of instances of ScopedValue as targets of node.
+        """
+        return self._targets
+
+    def set_targets(self, targets: [ScopedValue]):
+        """
+        Setter of _targets.
+
+        Note:
+            This interface can only be called before node been inserted into symbol-tree because target will be unique
+            while insert into symbol-tree, in other word, set_targets is not a user-interface.
+
+            When `_targets` is updated, corresponding ast node would be updated also.
+
+            When node_type of current node is CallCell or CallPrimitive or CallMethod or Tree, `targets` are strings
+            represents invoke result of the cell-op or primitive-op or function-call which are corresponding to targets
+            of ast.Assign.
+
+            When node_type of current node is Input, `targets` should have only one element which is a string represents
+            name of parameter of function.
+
+            When node_type of current node is Python or Output, `targets` are don't-care.
+
+        Args:
+            targets ([ScopedValue]): A list of instances of ScopedValue as new targets.
+        """
+        self._targets = targets
+        if self._node_type in (NodeType.CallCell, NodeType.CallMethod, NodeType.CallPrimitive,
+                               NodeType.Tree, NodeType.CallFunction, NodeType.CellContainer,
+                               NodeType.MathOps):
+            self._sync_assign_targets_to_ast()
+
+    def get_func_name(self) -> ScopedValue:
+        """
+        Getter of `_func_name`. See detail in docstring of Node class for meaning of func.
+
+        Returns:
+            An instance of ScopedValue.
+        """
+        return self._func_name
+
+    def set_func_name(self, func_name: ScopedValue):
+        """
+        Setter of `_func_name`. See detail in docstring of Node class for meaning of func.
+
+        Note:
+            When `_func_name` is updated, corresponding ast node would be updated also.
+
+        Args:
+            func (ScopedValue): An instance of ScopedValue as new func.
+        """
+        self._func_name = func_name
+        if self._node_type in (NodeType.CallCell, NodeType.CallPrimitive):
+            self._sync_assign_func_name_to_ast()
+
+    def get_name(self) -> str:
+        """
+        Getter of `_name`.
+
+        Returns:
+            A str represents name of node.
+        """
+        return self._name
+
+    def set_name(self, name: str):
+        """
+        Setter of `_name`.
+
+        Args:
+            name (str): A str as new name of node.
+        """
+        self._name = name
+
+    def get_node_type(self) -> NodeType:
+        """
+        Get the node_type of current node.
+
+        Returns:
+            A NodeType as node_type of node.
+        """
+        return self._node_type
+
+    def get_instance_type(self) -> type:
+        """
+        Get the instance_type of current node.
+
+        - When node_type of current node is CallCell, instance_type is type of cell-op.
+        - When node_type of current node is CallPrimitive, instance_type is type of primitive-op.
+        - When node_type of current node is Tree, instance_type is type of network-cell.
+        - When node_type of current node is Python, Input, Output or CallMethod, instance_type should be NoneType
+
+        Returns:
+            A type.
+        """
+        if isinstance(self._instance, LocalPrim):
+            return self._instance.prim_obj
+        if inspect.isfunction(self._instance):
+            return self._instance
+        return type(self._instance)
+
+    def get_instance(self):
+        """
+        Get the instance of current node.
+
+        - When node_type of current node is CallCell, instance is an instance of Cell.
+        - When node_type of current node is CallPrimitive, instance is an instance of primitive.
+        - When node_type of current node is Tree, instance is an instance of network-cell.
+        - When node_type of current node is Python, Input, Output or CallMethod, instance should be None
+
+        Returns:
+            A object.
+        """
+        return self._instance
+
+    def set_arg_by_node(self, arg_idx: int, node: 'Node', out_idx: Optional[int] = None):
+        """
+        Set argument by another Node.
+        Note that when _normalized_args is updated, corresponding ast node would be updated also.
+
+        Args:
+            arg_idx (int): Indicate which input being modified.
+            node (Node): Node as new input. Can be a node or name of node.
+            out_idx ([int, optional]): Indicate which output of `node` as new argument. Default is None which means use
+                first output of `node_to_link` as new input.
+
+        Raises:
+            ValueError: If `arg_idx` is out of range.
+            ValueError: If `node` has multi-outputs while `out_idx` is None or `out_idx` is not offered.
+        """
+        Validator.check_value_type("node", node, [Node], "Node")
+        Validator.check_int_range(arg_idx, 0, self._args_num, Validator.INC_LEFT, "arg_idx")
+        if out_idx is None:
+            if len(node.get_targets()) != 1:
+                raise ValueError("node should has one output when out_idx is not provided")
+            out_idx = 0
+        Validator.check_int_range(out_idx, 0, len(node.get_targets()), Validator.INC_LEFT, "arg_idx")
+        new_arg = node.get_targets()[out_idx]
+        self._normalized_args[self._normalized_args_keys[arg_idx]] = new_arg
+        self._sync_arg()
+
+    def set_arg(self, arg: Union[ScopedValue, str], index: int) -> (ScopedValue, ScopedValue):
+        """
+        Set argument of `node`.
+        Note that when _normalized_args is updated, corresponding ast node would be updated also.
+
+        Args:
+            index (int): Indicate which input being modified.
+            arg (Union[ScopedValue, str]): New argument to been set.
+
+        Raises:
+            ValueError: If `index` is out of range.
+        """
+        Validator.check_int_range(index, 0, self._args_num, Validator.INC_LEFT, "index")
+        Validator.check_value_type("arg", arg, [ScopedValue, str], "Node")
+        if isinstance(arg, str):
+            arg = ScopedValue.create_naming_value(arg)
+        old_arg = self._normalized_args.get(self._normalized_args_keys[index])
+        self._normalized_args[self._normalized_args_keys[index]] = arg
+        self._sync_arg()
+        return arg, old_arg
+
+    def set_args(self, args: [ScopedValue]):
+        """
+        Set arguments of `node`.
+        Note that when _normalized_args is updated, corresponding ast node would be updated also.
+
+        Args:
+            args (list[ScopedValue]): New arguments to been set.
+
+        Raises:
+            TypeError: Element of new argument is not an instance of ScopedValue.
+        """
+        Validator.check_int_range(len(args), 0, self._args_num, Validator.INC_LEFT, "Length of args")
+        Validator.check_element_type_of_iterable("args", args, [ScopedValue], "Node")
+        for arg_index, arg in enumerate(args):
+            if not isinstance(arg, ScopedValue):
+                raise TypeError("arg should be ScopedValue, got: ", type(arg))
+            self._normalized_args[self._normalized_args_keys[arg_index]] = arg
+        self._sync_arg()
+
+    def set_kwargs(self, kwargs: {str: ScopedValue}):
+        """
+        Set keywords arguments of 'node'.
+        Note that when _normalized_args is updated, corresponding ast node would be updated also.
+
+        Args:
+            kwargs (dict{str: ScopedValue}): New arguments to been set.
+
+        Raises:
+            TypeError: Value of new argument is not an instance of ScopedValue.
+            RuntimeError: Length of new arguments is not equal to length of old arguments.
+        """
+        Validator.check_int_range(len(kwargs), 0, self._kwargs_num, Validator.INC_LEFT, "Length of kwargs")
+        Validator.check_element_type_of_dict("kwargs", kwargs, [str], [ScopedValue], "Node")
+        for key, arg in kwargs.items():
+            if key not in self._normalized_args.keys() or key not in self._normalized_args_keys:
+                raise RuntimeError("Input key is not exist, ", key)
+            if not isinstance(arg, ScopedValue):
+                raise TypeError("arg should be ScopedValue, got: ", type(arg))
+            self._normalized_args[key] = arg
+        self._sync_arg()
+
+    def set_kwarg(self, key: str, arg: ScopedValue):
+        """
+        Set keyword argument of 'node'.
+        Note that when _normalized_args is updated, corresponding ast node would be updated also.
+
+        Args:
+            key (str): A str represents key of new argument.
+            arg (ScopedValue): An instance of ScopedValue represents argument.
+
+        Raises:
+            RuntimeError: If 'key' is not in original kwargs' keys.
+        """
+        if key not in self._normalized_args_keys[self._args_num:] or key not in self._normalized_args.keys():
+            raise RuntimeError("Input key is not exist, ", key)
+        self._normalized_args[key] = arg
+        self._sync_arg()
+
+    def get_args(self):
+        """
+        Get the arguments of current node.
+
+        - When node_type of current node is CallCell, CallPrimitive or Tree, arguments are corresponding to args of
+          ast.Call which represents arguments to invoke cell-op's forward method or primitive-op's `call()` method.
+        - When node_type of current node is Input, arguments represents default-value of argument of function.
+        - When node_type of current node is Output, arguments represents return values.
+        - When node_type of current node is Python, arguments are don't-care.
+
+        Returns:
+            A list of instances of ScopedValue.
+        """
+        args = []
+        for arg_index in range(self._args_num):
+            args.append(self._normalized_args.get(self._normalized_args_keys[arg_index]))
+        return args
+
+    def get_kwargs(self):
+        """
+        Get the keyword arguments of current node.
+
+        - When node_type of current node is CallCell, CallPrimitive or Tree, keyword arguments are corresponding to
+          kwargs of ast.Call which represents arguments to invoke cell-op's forward method or primitive-op's `call()`
+          method.
+        - When node_type of current node is Python, Input or Output, keyword arguments are don't-care.
+
+        Returns:
+            A dict of str to instance of ScopedValue.
+        """
+        kwargs: {str, ScopedValue} = {}
+        for arg_index in range(self._args_num, self._args_num + self._kwargs_num):
+            key = self._normalized_args_keys[arg_index]
+            kwargs[key] = self._normalized_args.get(key)
+        return kwargs
+
+    def get_normalized_args(self) -> {str: ScopedValue}:
+        """
+        Get the normalized keyword arguments of current node.
+        Normalized arguments combine arguments and keyword arguments into keyword arguments by using parameter name as
+        key of arguments.
+
+        Returns:
+            A dict of str to instance of ScopedValue.
+        """
+        output = {}
+        for key in self._normalized_args_keys:
+            output[key] = self._normalized_args.get(key)
+        return output
+
+    def set_normalized_args(self, args: {str, ScopedValue}):
+        """
+        Set the normalized keyword arguments of current node.
+        Normalized arguments combine arguments and keyword arguments into keyword arguments by using parameter name as
+        key of arguments.
+
+        Args:
+            args ({str, ScopedValue}): A dict of str to instance of ScopedValue represents new normalized_args.
+        """
+        if len(args.values()) != len(self._normalized_args_keys):
+            raise RuntimeError("Length of args.values() should be equal to length of _normalized_args_keys, ",
+                               len(args.values()), " vs ", len(self._normalized_args_keys))
+        for key, arg in args.items():
+            self._normalized_args[key] = arg
+        self._sync_arg()
+
+    def set_attribute(self, key: str, value):
+        """
+        Set attribute of current node.
+
+        Args:
+            key (str): Key of new attribute.
+            value (object): Value of new attribute.
+        """
+        self._attribute[key] = value
+
+    def set_attributes(self, attributes):
+        """
+        Set attributes of current node.
+
+        Args:
+            attributes (dict): A dict represents new attributes.
+        """
+        self._attribute = attributes
+
+    def get_attributes(self):
+        """
+        Get all attributes of current node.
+
+        Returns:
+            A dict of str to instance of object as attributes.
+        """
+        return self._attribute
+
+    def get_attribute(self, key: str):
+        """
+        Get attribute of current node by key.
+
+        Args:
+            key (str): A str represents key of attribute you want to get.
+
+        Returns:
+            A object as attribute.
+        """
+        return self._attribute.get(key)
+
+    def get_arg_providers(self) -> dict:
+        """
+        Getter of _arg_providers.
+
+        Return:
+            dict, key is type of int indicating the index of args, and value is type of tuple, which includes
+                the node and the index of node's targets who provides the argument.
+        """
+        return self._arg_providers
+
+    def set_arg_providers(self, index: int, provider: tuple):
+        """
+        Setter of _arg_providers.
+
+        Args:
+            index (int): Indicating provider of which argument need to be set.
+            provider (tuple): A tuple includes the node and the index of node's targets who provides the argument.
+        """
+        self._arg_providers[index] = provider
+
+    def get_target_users(self, index=-1) -> Union[dict, list]:
+        """
+        Getter of _target_users.
+
+        Args:
+            index (int): Indicating users of which target need to be got. Default: -1, means all targets's users will
+                be returned.
+
+        Return:
+            Union[dict, list]. When index is not -1, a list of users of specified target will be returned.
+                The type of elements in list is tuple, which includes the user node and the index of node's arguments
+                who uses the target. When index is -1, a dict will be returned. The key is index of targets, and the
+                value is list of users of corresponding target.
+        """
+        if index == -1:
+            return self._target_users
+        if index not in self._target_users.keys():
+            self._target_users[index] = []
+        return self._target_users.get(index, None)
+
+    def append_target_users(self, index: int, provider: tuple):
+        """
+        Setter of _target_users.
+
+        Args:
+            index (int): Indicating users of which target need to be append.
+            provider (tuple): A tuple includes the node and the index of node's argument who uses the target.
+
+        """
+        if index not in self._target_users.keys():
+            self._target_users[index] = []
+        self._target_users.get(index).append(provider)
+
+    def update_ast_node(self) -> ast.AST:
+        """Update node's ast_node by current targets, func_name, args and kwargs."""
+        ast_assign = AstModifier.create_call_assign(self.get_targets(), self.get_func_name(),
+                                                    self.get_args(), self.get_kwargs())
+        self.set_ast(ast_assign)
+        return ast_assign
+
+    def get_source_code(self) -> str:
+        """Get source code of node from ast of node."""
+        return astunparse.unparse(self._ast_node).strip()
+
+    def append_kwarg(self, kwarg: Dict[str, ScopedValue]):
+        """
+        Append a new keyword arg to node.
+
+        Args:
+            kwarg (Dict[str, ScopedValue]): The new keyword arg.
+
+        """
+        if self.get_node_type() not in [NodeType.Tree, NodeType.CallFunction]:
+            raise TypeError(f"For append_new_kwarg, the type of node can only be one of [Tree, CallFunction], "
+                            f"but got {self.get_node_type()}")
+        Validator.check_element_type_of_dict("kwarg", kwarg, [str], [ScopedValue], "append_new_kwarg")
+        for arg_key, value in kwarg.items():
+            # add keyword into _normalized_args
+            self._normalized_args[arg_key] = value
+            self._normalized_args_keys.append(arg_key)
+            self._kwargs_num += 1
+            # add keyword ast into ast.Call
+            ast_assign: ast.Assign = self._ast_node
+            ast_call: ast.Call = ast_assign.value
+            new_keyword = ast.keyword(arg=arg_key, value=AstModifier.get_ast_by_value(value, None))
+            ast_call.keywords.append(new_keyword)
+
+    def _get_normalized_args(self, args: [ScopedValue], kwargs: {str: ScopedValue}) -> dict:
+        """
+        Merge args and kwargs to normalized args.
+        The keys of args are obtained from the construct function of type(self._instance).
+
+        Args:
+            args (list[ScopedValue]): A list of instance of ScopedValue. See detail in docstring of Node class.
+            kwargs (dict{str: ScopedValue}): A list of instance of ScopedValue. See detail in docstring of Node class.
+
+        Raises:
+            RuntimeError: Input args are invalid.
+            RuntimeError: Arg name already exist in kwargs.
+
+        Returns:
+            The normalized args.
+        """
+        if not args:
+            args = []
+        if not kwargs:
+            kwargs = {}
+        normalized_args: dict = dict()
+        if (args or kwargs) and self._instance and hasattr(type(self._instance), "construct"):
+            parameters = inspect.signature(type(self._instance).construct).parameters
+            names = Node._get_construct_arg_names(parameters)
+            Node._map_args_names(names, args, kwargs, self._normalized_args_keys, normalized_args)
+        else:
+            logger.debug("fail to get arg name from op, using arg_xx for args' name")
+            arg_temp_name, suffix = "arg", 0
+            for arg in args:
+                arg_key = "{}_{}".format(arg_temp_name, suffix)
+                while arg_key in kwargs.keys() or arg_key in normalized_args.keys():
+                    suffix += 1
+                    arg_key = "{}_{}".format(arg_temp_name, suffix)
+                normalized_args[arg_key] = arg
+                self._normalized_args_keys.append(arg_key)
+            for arg_key, value in kwargs.items():
+                normalized_args[arg_key] = value
+                self._normalized_args_keys.append(arg_key)
+        return normalized_args
+
+    # Synchronize rewrite node args to ast node
+    def _sync_assign_func_name_to_ast(self):
+        """Sync func_name of ast.Call of ast.Assign from self._name when NodeType is CallCell or CallPrimitive."""
+        if self._ast_node is None:
+            return
+        assign_ast = self._ast_node
+        if not isinstance(assign_ast, ast.Assign):
+            raise TypeError("assign_ast should be ast.Assign, got: ", type(assign_ast))
+        call_ast = assign_ast.value
+        if not isinstance(call_ast, ast.Call):
+            raise TypeError("call_ast should be ast.Call, got: ", type(call_ast))
+        if self._func_name.type == ValueType.UnsupportedValue:
+            return
+        func_ast = call_ast.func
+        if not self._func_name.scope:
+            if isinstance(func_ast, ast.Name):
+                func_ast.id = self._func_name.value
+            else:
+                call_ast.func = ast.Name(self._func_name.value, ast.Store())
+        else:
+            if isinstance(func_ast, ast.Attribute):
+                if not isinstance(func_ast.value, ast.Name):
+                    func_ast.value = ast.Name(self._func_name.scope, ast.Load())
+                else:
+                    func_ast.value.id = self._func_name.scope
+                func_ast.attr = self._func_name.value
+            else:
+                call_ast.func = ast.Attribute(ast.Name(self._func_name.scope, ast.Load()),
+                                              self._func_name.value, ast.Store())
+        ast.fix_missing_locations(assign_ast)
+
+    def _sync_assign_targets_to_ast(self):
+        """Sync targets of ast.Assign from self._targets when NodeType is CallCell, CallPrimitive or CallMethod."""
+        if self._ast_node is None:
+            return
+        assign_ast = self._ast_node
+        if not isinstance(assign_ast, ast.Assign):
+            raise TypeError(error_str(f"assign_ast should be ast.Assign, but got: {type(assign_ast)}",
+                                      father_node=assign_ast))
+        # update targets
+        target_ast_elems = AstConverter.get_ast_target_elems(assign_ast.targets[0])
+        if len(self._targets) != len(target_ast_elems):
+            raise ValueError(error_str(f"The number of targets should be {len(target_ast_elems)}, "
+                                       f"but got {len(self._targets)}", father_node=assign_ast))
+        for i, target_ast in enumerate(target_ast_elems):
+            target_ast_elems[i] = AstModifier.get_ast_by_value(self._targets[i], target_ast)
+
+    def _sync_call_args_to_ast(self):
+        """Sync args of ast.Call from self._normalized_args."""
+        if self._ast_node is None:
+            return
+        assign_ast = self._ast_node
+        if not isinstance(assign_ast, ast.Assign):
+            raise TypeError(f"When synchronizing args for '{self._name}'({self._node_type}), _ast_node should be "
+                            f"ast.Assign, but got: {type(assign_ast)}")
+        assign_value = assign_ast.value
+        if not isinstance(assign_value, ast.Call):
+            if isinstance(assign_value, ast.Attribute) and self._node_type in (NodeType.CellContainer,
+                                                                               NodeType.CallCell):
+                # CellContainers in control flow may be flatten to ast.Attribute: blocks_var = self.blocks
+                # In this case, no args exist in node, so we don't need to sync.
+                # CellContainers may be type of CallCell when share one implementation
+                return
+            raise TypeError(f"When synchronizing args for '{self._name}'({self._node_type}), _ast_node.value should "
+                            f"be ast.Call, but got: {type(assign_value)}")
+        keywords_ast = assign_value.keywords
+        args_ast = assign_value.args
+        if len(self._normalized_args_keys) != (len(keywords_ast) + len(args_ast)):
+            raise ValueError("ast keywords plus args len is not equal to self._normalized_args value")
+        for arg_index in range(self._args_num):
+            arg_ast = args_ast[arg_index]
+            args_ast[arg_index] = \
+                AstModifier.get_ast_by_value(self._normalized_args.get(self._normalized_args_keys[arg_index]), arg_ast)
+
+        # the order of kwargs may not the same as that in keywords_ast
+        keyword_map_index = {}
+        for index, keyword_ast in enumerate(keywords_ast):
+            keyword_map_index[keyword_ast.arg] = index
+        for keyword_index in range(self._kwargs_num):
+            key = self._normalized_args_keys[keyword_index + self._args_num]
+            keywords_ast[keyword_map_index.get(key)].value = \
+                AstModifier.get_ast_by_value(self._normalized_args.get(key),
+                                             keywords_ast[keyword_map_index.get(key)].value)
+
+    def _sync_call_method_args_to_ast(self):
+        """
+        Sync args to value of ast.Assign from self._normalized_args when NodeType is CallMethod.
+        For node with type of CallMethod, the value of ast.Assign is one of:
+        | func_name      | data_type   | value of ast.Assign     |
+        |:---------------|:------------|:------------------------|
+        | 'pass_through' | constants   | ast.Constant, ast.NameConstant, ast.Num, ast.Bytes, ast.Str |
+        | 'pass_through' | variables   | ast.Name, ast.Attribute |
+        | 'tuple'        | tuple       | ast.Tuple               |
+        | 'list'         | list        | ast.List                |
+        | 'dict'         | dict        | ast.Dict                |
+        """
+        if self._ast_node is None:
+            return
+        assign_ast = self._ast_node
+        if not isinstance(assign_ast, ast.Assign):
+            raise TypeError(f"For node '{self.get_name()}', assign_ast should be ast.Assign, got: ", type(assign_ast))
+        assign_value = assign_ast.value
+        if self._func_name.value == "pass_through":
+            # update constants/variables
+            assign_ast.value = \
+                AstModifier.get_ast_by_value(self._normalized_args.get(self._normalized_args_keys[0]), assign_value)
+        elif self._func_name.value in ("tuple", "list", "dict"):
+            # update tuple/list/dict
+            ast_elts = assign_value.values if isinstance(assign_value, ast.Dict) else assign_value.elts
+            if len(self._normalized_args_keys) != len(ast_elts):
+                raise ValueError(f"For node '{self.get_name()}', size of self._normalized_args_keys"
+                                 f"({len(self._normalized_args_keys)}) should be equal to size of elements of "
+                                 f"ast_elts({len(ast_elts)})")
+            for index, elt in enumerate(ast_elts):
+                scoped_value: ScopedValue = self._normalized_args.get(self._normalized_args_keys[index])
+                ast_elts[index] = AstModifier.get_ast_by_value(scoped_value, elt)
+        else:
+            raise TypeError(f"For node '{self.get_name()}', only support (pass_through, tuple or dict method) as "
+                            f"call_method, but got {self._func_name.value}")
+
+    def _sync_return_node_to_ast(self):
+        """
+        Sync args to value of ast.Return from self._normalized_args when NodeType is Output.
+
+        For node with type of CallMethod, the value of ast.Assign is one of:
+        (ast.Name, ast.Attribute)
+        """
+        if self._ast_node is None:
+            return
+        return_ast = self._ast_node
+        if not isinstance(return_ast, ast.Return):
+            raise TypeError(f"For node '{self.get_name()}', return_ast should be ast.Return, got: {type(return_ast)}")
+        return_value_ast = return_ast.value
+        return_ast.value = AstModifier.get_ast_by_value(self._normalized_args.get(self._normalized_args_keys[0]),
+                                                        return_value_ast)
+
+    def _sync_mathops_node_args_to_ast(self):
+        """
+        Sync values from self._normalized_args to the ast node for mathematical operations.
+        """
+        if self._ast_node is None:
+            return
+        if not isinstance(self._ast_node, ast.Assign):
+            raise TypeError(f"type of node should be ast.Assign, but got {type(self._ast_node)}")
+        mathops_node = self._ast_node.value
+        if isinstance(mathops_node, ast.BinOp):
+            left = mathops_node.left
+            right = mathops_node.right
+            mathops_node.left = AstModifier.get_ast_by_value(self._normalized_args.get(self._normalized_args_keys[0]),
+                                                             left)
+            mathops_node.right = AstModifier.get_ast_by_value(self._normalized_args.get(self._normalized_args_keys[1]),
+                                                              right)
+        elif isinstance(mathops_node, ast.UnaryOp):
+            operand = mathops_node.operand
+            mathops_node.operand = \
+                AstModifier.get_ast_by_value(self._normalized_args.get(self._normalized_args_keys[0]), operand)
+        elif isinstance(mathops_node, ast.BoolOp):
+            values = mathops_node.values
+            for arg_index in range(self._args_num):
+                arg_value = self._normalized_args.get(self._normalized_args_keys[arg_index])
+                values[arg_index] = AstModifier.get_ast_by_value(arg_value, values[arg_index])
+        elif isinstance(mathops_node, ast.Compare):
+            left = mathops_node.left
+            mathops_node.left = AstModifier.get_ast_by_value(self._normalized_args.get(self._normalized_args_keys[0]),
+                                                             left)
+            comparators = mathops_node.comparators
+            for arg_index in range(1, self._args_num):
+                arg_value = self._normalized_args.get(self._normalized_args_keys[arg_index])
+                comparators[arg_index - 1] = AstModifier.get_ast_by_value(arg_value, comparators[arg_index - 1])
+        else:
+            raise TypeError("The type of 'mathops_node' must be one of (ast.BinOp, ast.UnaryOp, "
+                            "ast.BoolOp, ast.Compare), but got ", type(mathops_node))
+
+    def _sync_control_flow_args_to_ast(self):
+        """
+        Sync values from self._normalized_args to the ast node of control flow.
+        """
+        if self._ast_node is None:
+            return
+        normalized_args_num = len(self._normalized_args_keys)
+        if normalized_args_num == 0:
+            return
+        if normalized_args_num > 1:
+            raise ValueError("self._normalized_args_keys should have less than 1 elements")
+        arg_value = self._normalized_args.get(self._normalized_args_keys[0])
+        if isinstance(self._ast_node, (ast.If, ast.IfExp, ast.While)):
+            self._ast_node.test = AstModifier.get_ast_by_value(arg_value, self._ast_node.test)
+        elif isinstance(self._ast_node, ast.For):
+            self._ast_node.iter = AstModifier.get_ast_by_value(arg_value, self._ast_node.iter)
+        else:
+            raise ValueError(f"For Control Flow, ast_node should be one of [ast.If, ast.IfExp, "
+                             f"ast.While, ast.For], but got {type(self._ast_node)}")
+
+    def _sync_arg(self):
+        """Sync _normalized_args to corresponding ast node when updated."""
+        if self._node_type in (NodeType.CallCell, NodeType.CallPrimitive, NodeType.Tree, \
+                               NodeType.CellContainer, NodeType.CallFunction):
+            self._sync_call_args_to_ast()
+        elif self._node_type == NodeType.Output:
+            self._sync_return_node_to_ast()
+        elif self._node_type == NodeType.CallMethod:
+            self._sync_call_method_args_to_ast()
+        elif self._node_type == NodeType.MathOps:
+            self._sync_mathops_node_args_to_ast()
+        elif self._node_type == NodeType.ControlFlow:
+            self._sync_control_flow_args_to_ast()
+
+
+# Child classes
+class TreeNode(Node):
+    """Tree type Node who holds a handler of SymbolTree."""
+
+    def __init__(self, tree, ast_node: ast.AST, targets: [ScopedValue], func: ScopedValue,
+                 args: [ScopedValue], kwargs: {str: ScopedValue}, name: str, instance):
+        """
+        Constructor of TreeNode. Rewrite recommend to invoking class method of Node to instantiate an instance of
+        TreeNode such as `create_tree_node` rather than invoking constructor of Node directly.
+
+        Args:
+            tree: An instance of SymbolTree represents a handler of sub-symbol-tree.
+            ast_node (ast.AST): An instance of ast.AST represents corresponding node in ast.
+            targets (list[ScopedValue]): A list of instance of ScopedValue. See detail in docstring of Node class.
+            func ([ScopedValue, optional]): An instance of ScopedValue. See detail in docstring of Node class.
+            args (list[ScopedValue]): A list of instance of ScopedValue. See detail in docstring of Node class.
+            kwargs (dict{str: ScopedValue}): A list of instance of ScopedValue. See detail in docstring of Node class.
+            name (str): A string represents name of node. Name of node will be unique when inserted into SymbolTree.
+                Name of node also used as field name in network class.
+            instance: Object in network corresponding to this node.
+        """
+        if isinstance(func, str):
+            func = ScopedValue.create_naming_value(func)
+        super().__init__(NodeType.Tree, ast_node, targets, func, args, kwargs, name, instance)
+        self.symbol_tree = tree
+
+    @classmethod
+    def create_tree_node(cls, tree, ast_node: ast.AST, targets: Union[ScopedValue, str],
+                         func_name: Union[ScopedValue, str], args: [ScopedValue], kwargs: {str: ScopedValue},
+                         name: str = "", instance=None):
+        """
+        Class method of TreeNode. Instantiate an instance of node whose type is Tree. A Tree node represents an invoking
+        to sub-network.
+
+        Args:
+            tree: An instance of SymbolTree represents a handler of sub-symbol-tree.
+            ast_node (ast.AST): An instance of ast.AST represents corresponding node in ast.
+            targets (list[ScopedValue]): A list of instance of ScopedValue. See detail in docstring of Node class.
+            func_name ([ScopedValue, optional]): An instance of ScopedValue. See detail in docstring of Node class.
+            args (list[ScopedValue]): A list of instance of ScopedValue. See detail in docstring of Node class.
+            kwargs (dict{str: ScopedValue}): A list of instance of ScopedValue. See detail in docstring of Node class.
+            name (str): A string represents name of node. Name of node will be unique when inserted into SymbolTree.
+                Name of node also used as field name in network class.
+            instance: Object in network corresponding to this node.
+        """
+        new_targets = Node._handle_targets(targets)
+        if isinstance(func_name, str):
+            func_name = ScopedValue.create_naming_value(func_name)
+        return cls(tree, ast_node, new_targets, func_name, args, kwargs, name, instance)