add exception

Author: danielenricocahall

keras/src/backend/tensorflow/nn.py

@@ -310,14 +310,21 @@ def conv(
 ):
     def _conv():
         tf_data_format = _convert_data_format(data_format, len(inputs.shape))
-        return tf.nn.convolution(
+        result = tf.nn.convolution(
             inputs,
             kernel,
             strides,
             padding.upper(),
             data_format=tf_data_format,
             dilations=dilation_rate,
         )
+        if any(dim == 0 for dim in result.shape):
+            raise ValueError(
+                f"Convolution produced an output with size 0 dimension: "
+                f"{result.shape}. Kernel size "
+                f"cannot be greater than the padded input size."
+            )
+        return result
 
     # Certain ops are are broken in Tensorflow on CPU only.
     # We can work around by compiling the op with XLA.

keras/src/layers/convolutional/conv_test.py

@@ -1095,3 +1095,11 @@ def test_conv_constraints(self):
         )
         layer.build((None, 5, 5, 3))
         self.assertIsInstance(layer.bias.constraint, constraints.NonNeg)
+
+    def test_conv_raises_exception_on_zero_dims(self):
+        x = np.random.rand(3, 4, 4, 4)
+        l = layers.Conv2D(6, [5, 5], 1, "valid")
+        with self.assertRaisesRegex(
+            ValueError, "Convolution produced an output with size 0 dimension"
+        ):
+            l(x)