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
errmsg = "%s Input number: %d if inconsistent with the output number %d" % (unpack_cpp, \
len(node.inputs), len(list(last_cpp.inputs())))
assert len(node.inputs) == len(list(last_cpp.inputs())), errmsg
for _debug_input, _debug_output in zip(node.inputs, node.outputs):
// _debug_input = _input.debugName()
// _debug_output = _output.debugName()
if _debug_input in self.input_to_node and _debug_output in self.input_to_node: