5d2a59fd4cf708d285d0db8ff3522c9156d2c4a9,src/sdk/pynni/nni/_graph_utils.py,TorchModuleGraph,unpack_manually,#TorchModuleGraph#,538

Before Change

                            self.input_to_node[_debug_input].remove(node)
                            // add the following nodes of _output into the input_to_node[_debug_input]
                            self.input_to_node[_debug_input].extend(self.input_to_node[_debug_output])
                        if _debug_input in self.output_to_node and _debug_output in self.output_to_node:
                            // output_to_node[_debug_output] is a NodePyGroup, because one output
                            // tensor only can be generated by one node.
                            self.output_to_node[_debug_output] = self.output_to_node[_debug_input]

        self.unpacked = True

    def _build_graph(self):
        

After Change

                            self.input_to_node[_debug_input].extend(self.input_to_node[_debug_output])
                        // just remove the _debug_output from the grapgh index. So that we can also skip
                        // the construct and tuple
                        if _debug_output in self.input_to_node:
                            for following_node in self.input_to_node[_debug_output]:
                                _tmp_index = following_node.inputs.index(_debug_output)
                                following_node.inputs[_tmp_index] = _debug_input


        self.unpacked = True

    def _build_graph(self):
        

Instances