Adding Constant Folding for Reshape Nodes to Optimizer

Author: phager90

tf2onnx/optimizer/const_fold_optimizer.py

@@ -110,6 +110,14 @@ def _fold_transpose(node, graph) -> list:
         const_val_after_trans = const_val.transpose(perm)
         return [const_val_after_trans]
 
+    @staticmethod
+    @_register_func("Reshape")
+    def _fold_reshape(node, graph):
+        const_val_data = node.inputs[0].get_tensor_value(as_list=False) 
+        const_val_shape = node.inputs[1].get_tensor_value(as_list=False)
+        const_val_after_trans = const_val_data.reshape(const_val_shape)
+        return [const_val_after_trans]
+
     @staticmethod
     @_register_func("Unsqueeze")
     def _fold_unsqueeze(node, graph):