Internal change

PiperOrigin-RevId: 704263127
google · Dec 10, 2024 · f7ad7c7 · f7ad7c7
1 parent b33fb98
commit f7ad7c7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/edward2/tensorflow/layers/recurrent.py b/edward2/tensorflow/layers/recurrent.py
@@ -151,6 +151,8 @@ def call_weights(self):
       self.recurrent_kernel = self.recurrent_initializer(
           self.recurrent_kernel.shape, self.dtype)
     if isinstance(self.bias_initializer, tf.keras.layers.Layer):
+      if self.bias is None:
+        raise ValueError('self.bias is None.')
       self.bias = self.bias_initializer(self.bias.shape, self.dtype)
     self.called_weights = True