Function post_order_recursive

python/singa/utils.py:234–267

return a list by the topological ordering (postorder of Depth-first search) Args: root: singa operator root_t: tensor Returns: deque[int]

(root, root_t)

def post_order_recursive(root, root_t):
    """
    return a list by the topological ordering (postorder of Depth-first search)
    Args:
        root: singa operator
        root_t: tensor
    Returns: 
        deque[int]
    """

    def recursive(root, yid, root_t):
        if root:
            # srcop: operator for a input of root
            # yid: id(output of this operator)
            # y: output of this operator
            for srcop, yid, y, _ in root.src:
                recursive(srcop, yid, y)

            if type(root).__name__ == 'Dummy':
                if root_t != None:
                    # constant within a node: weight
                    weights[root.name] = root_t
                else:
                    # constant outside a node: input
                    inputs[root.name] = root_t
            else:
                nodes[root.name] = root

    nodes = OrderedDict()
    weights = OrderedDict()
    inputs = OrderedDict()

    recursive(root, None, root_t)
    return nodes, weights, inputs