5c8f92c5d865bb8d8a90740506e4a7ef9f62398c,python/src/nnabla/utils/converter/tensorflow/importer.py,,_find_out_terminal_node,#Any#,32

Before Change

    unlike_output_types = ["Const", "Assign", "Noop", "Placeholder"]
    terminal_inputs = []
    inputs = set()
    outputs = set()
    need_add_postfix = kwargs.get("postfix", False)
    for node in graph_def.node:
        strip_name = _strip_node_name(node.name)
        if node.op == "Placeholder":
            terminal_inputs.append(_strip_node_name(node.name))
        outputs.add(strip_name)
        inputs.update(set(node.input))
    terminal_outputs = list(filter(lambda x: x not in unlike_output_types,
                                   outputs - inputs))
    if need_add_postfix:
        terminal_inputs = add_postfix(terminal_inputs)
        terminal_outputs = add_postfix(terminal_outputs)

After Change

            strip_name = _strip_node_name(node.name)
            terminal_inputs.append(strip_name)

    for node in graph_def.node:
        if input_cnt[node.name] == 0 and node.op not in unlike_output_types:
            terminal_outputs.append(node.name)

    if need_add_postfix:
        terminal_inputs = add_postfix(terminal_inputs)
        terminal_outputs = add_postfix(terminal_outputs)

Instances