Reshape does not support INT32

Signed-off-by: Justin Chu <justinchuby@users.noreply.github.com>

Author: justinchuby

onnxscript/optimizer/_constant_folding.py

@@ -196,10 +196,6 @@ def set_sym_value(self, value: ir.Value, sym_value: SymbolicValue) -> None:
 
     def get_shape_value(self, value: ir.Value | None) -> ir.Shape | None:
         const_value = _get_numpy_value(value, ir.DataType.INT64, size_limit=10)
-        if const_value is None:
-            # Reshape accepts shape input of INT32 type as well, so we also check for INT32 here
-            # This is common for tflite models
-            const_value = _get_numpy_value(value, ir.DataType.INT32, size_limit=10)
         if const_value is not None:
             if const_value.ndim == 1:
                 return ir.Shape(const_value.tolist())

onnxscript/optimizer/_constant_folding_test.py

@@ -614,22 +614,6 @@ def test_gather_symdim(self):
         optimized = self._fold(model)
         self.assertEqual(optimized.graph.node(-1).op_type, "Identity")
 
-    def test_reshape_identity_int32_shape(self):
-        """Reshape with a constant INT32 shape input should be recognized as identity."""
-        model_ir = ir.from_onnx_text(
-            """
-            <ir_version: 7, opset_import: [ "" : 17]>
-            agraph (float[3, 4] x) => (float[3, 4] z)
-            {
-                shape_i64 = Constant <value_ints=[3, 4]> ()
-                shape = Cast <to=6> (shape_i64)
-                z = Reshape (x, shape)
-            }
-        """
-        )
-        optimized = self._fold(model_ir)
-        self.assertEqual(optimized.graph.node(-1).op_type, "Identity")
-
     def test_input_size_limit(self):
         model_text = """
             <ir_version: 7, opset_import: [ "" : 17]>