From 85db5d07db54b853484bfd358c3894d948c36baf Mon Sep 17 00:00:00 2001 From: Zhufeng Pan Date: Wed, 9 Nov 2022 10:49:10 -0800 Subject: [PATCH] Add a test case for keras.layers.Convolution3DTranspose to verify zero input shape does not raise errors PiperOrigin-RevId: 487289971 --- keras/layers/convolutional/conv_test.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/keras/layers/convolutional/conv_test.py b/keras/layers/convolutional/conv_test.py index fa0a04441f6..d8d7603142a 100644 --- a/keras/layers/convolutional/conv_test.py +++ b/keras/layers/convolutional/conv_test.py @@ -448,6 +448,12 @@ def test_conv3d_invalid_output_shapes(self): layer = keras.layers.Conv3D(**kwargs) layer.build((None, 5, 5, 5, 2)) + def test_conv3d_zero_dim_output(self): + conv = keras.layers.Convolution3DTranspose(2, [3, 3, 3], padding="same") + x = tf.random.uniform([1, 32, 32, 0, 3], dtype=tf.float32) + # The layer doesn't crash with 0 dim input + _ = conv(x) + @test_combinations.run_all_keras_modes(always_skip_v1=True) class GroupedConvTest(test_combinations.TestCase):