16474aee0c575e615c78d32f2a170c3d7f8f0082,autokeras/graph.py,Graph,produce_model,#Graph#,75

Before Change

        output_id = self.node_to_id[self.model.outputs[0]]

        id_to_tensor = {input_id: input_tensor}
        q = Queue()
        q.put(input_id)
        while not q.empty():
            u = q.get()
            for v, edge_id in self.adj_list[u]:
                layer = self.edge_list[edge_id]
                if edge_id in self.old_edge_ids:
                    copied_layer = copy_layer(layer, get_int_tuple(id_to_tensor[u].shape))
                    temp_tensor = copied_layer(id_to_tensor[u])
                else:
                    temp_tensor = layer(id_to_tensor[u])
                id_to_tensor[v] = temp_tensor

After Change

                layer = self.edge_list[edge_id]

                if isinstance(layer, WeightedAdd):
                    edge_input_tensor = list(map(lambda x: id_to_tensor[x], self.edge_id_to_input_ids[edge_id]))
                else:
                    edge_input_tensor = id_to_tensor[u]

                if edge_id in self.old_edge_ids:

Instances