Removed unused part of the code. The code in the fix should deal exclusively with the case that is not properly handled by ops.reshape. This is when the batch dimension does not have a static shape (shape == None) but all other dimensions have static shape. In that case we can determine the static shape of the dimension containing -1 form all dimensions that are not the batch dimension.

Author: roebel

keras/src/layers/reshaping/reshape.py

@@ -70,11 +70,7 @@ def call(self, inputs):
         if self.need_explicit_shape_for_batch_size_None and (inputs.shape[0] is None):
             input_nonbatch_shape = tuple(inputs.shape[1:])
             if input_nonbatch_shape.count(None) == 0:
-                # If the input shape is fully defined, we can compute the desired target_shape
-                if True:
-                    inp_nonbatch_size = math.prod(inputs.shape[1:])
-                else:
-                    inp_nonbatch_size = ops.prod(ops.shape(inputs)[1:])
+                inp_nonbatch_size = math.prod(inputs.shape[1:])
                 target_shape = tuple(d if d != -1 else (inp_nonbatch_size // self.new_size_no_minus_one) for d in self.target_shape)
 
         return ops.reshape(