Fix the parse of output_nodes for submodule when tmp output is used in node's kwargs. (#482)

Author: Xreki

graph_net/torch/decompose_util.py

@@ -291,6 +291,8 @@ def get_args_node(arg):
     def get_args_node_and_self_node(node):
         for arg in node.args:
             yield from get_args_node(arg)
+        for name, values in node.kwargs.items():
+            yield from get_args_node(values)
         yield node
 
     for node in node_list[0:start_node_idx]: