From e5a9b73ccf898014b6e1f7efbfd5c20ac7912dda Mon Sep 17 00:00:00 2001 From: Nico Mexis Date: Sat, 4 Mar 2023 10:17:44 +0100 Subject: [PATCH] Fix layer.add_variable being deprecated --- tensorflow_probability/python/layers/conv_variational.py | 8 ++++---- tensorflow_probability/python/layers/dense_variational.py | 8 ++++---- 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/tensorflow_probability/python/layers/conv_variational.py b/tensorflow_probability/python/layers/conv_variational.py index 88e9fc1c8e..8b62fa34ca 100644 --- a/tensorflow_probability/python/layers/conv_variational.py +++ b/tensorflow_probability/python/layers/conv_variational.py @@ -185,28 +185,28 @@ def build(self, input_shape): # Must have a posterior kernel. self.kernel_posterior = self.kernel_posterior_fn( dtype, kernel_shape, 'kernel_posterior', - self.trainable, self.add_variable) + self.trainable, self.add_weight) if self.kernel_prior_fn is None: self.kernel_prior = None else: self.kernel_prior = self.kernel_prior_fn( dtype, kernel_shape, 'kernel_prior', - self.trainable, self.add_variable) + self.trainable, self.add_weight) if self.bias_posterior_fn is None: self.bias_posterior = None else: self.bias_posterior = self.bias_posterior_fn( dtype, (self.filters,), 'bias_posterior', - self.trainable, self.add_variable) + self.trainable, self.add_weight) if self.bias_prior_fn is None: self.bias_prior = None else: self.bias_prior = self.bias_prior_fn( dtype, (self.filters,), 'bias_prior', - self.trainable, self.add_variable) + self.trainable, self.add_weight) self.input_spec = tf.keras.layers.InputSpec( ndim=self.rank + 2, axes={channel_axis: input_dim}) diff --git a/tensorflow_probability/python/layers/dense_variational.py b/tensorflow_probability/python/layers/dense_variational.py index 2f842016b9..f9b3dec1d7 100644 --- a/tensorflow_probability/python/layers/dense_variational.py +++ b/tensorflow_probability/python/layers/dense_variational.py @@ -140,28 +140,28 @@ def build(self, input_shape): # Must have a posterior kernel. self.kernel_posterior = self.kernel_posterior_fn( dtype, [in_size, self.units], 'kernel_posterior', - self.trainable, self.add_variable) + self.trainable, self.add_weight) if self.kernel_prior_fn is None: self.kernel_prior = None else: self.kernel_prior = self.kernel_prior_fn( dtype, [in_size, self.units], 'kernel_prior', - self.trainable, self.add_variable) + self.trainable, self.add_weight) if self.bias_posterior_fn is None: self.bias_posterior = None else: self.bias_posterior = self.bias_posterior_fn( dtype, [self.units], 'bias_posterior', - self.trainable, self.add_variable) + self.trainable, self.add_weight) if self.bias_prior_fn is None: self.bias_prior = None else: self.bias_prior = self.bias_prior_fn( dtype, [self.units], 'bias_prior', - self.trainable, self.add_variable) + self.trainable, self.add_weight) self.built = True