Avoid None shape entries in TimeDistributed

keras-team · Jun 25, 2024 · b038ce2 · b038ce2
1 parent 9cfe208
commit b038ce2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/keras/src/layers/rnn/time_distributed.py b/keras/src/layers/rnn/time_distributed.py
@@ -72,8 +72,8 @@ def build(self, input_shape):
         self.built = True
 
     def call(self, inputs, training=None, mask=None):
-        input_shape = inputs.shape
-        mask_shape = None if mask is None else tuple(mask.shape)
+        input_shape = ops.shape(inputs)
+        mask_shape = None if mask is None else ops.shape(mask)
         batch_size = input_shape[0]
         timesteps = input_shape[1]