diff --git a/tensorflow_probability/python/bijectors/bijector.py b/tensorflow_probability/python/bijectors/bijector.py index 0380f28748..e0899d3cbb 100644 --- a/tensorflow_probability/python/bijectors/bijector.py +++ b/tensorflow_probability/python/bijectors/bijector.py @@ -636,6 +636,9 @@ def __init__(self, self._initial_parameter_control_dependencies = tuple( d for d in self._parameter_control_dependencies(is_init=True) if d is not None) + if self._initial_parameter_control_dependencies: + self._initial_parameter_control_dependencies = ( + tf.group(*self._initial_parameter_control_dependencies),) if forward_min_event_ndims is None and inverse_min_event_ndims is None: raise ValueError('Must specify at least one of `forward_min_event_ndims` ' diff --git a/tensorflow_probability/python/distributions/distribution.py b/tensorflow_probability/python/distributions/distribution.py index 5ae02fc14d..9045b03179 100644 --- a/tensorflow_probability/python/distributions/distribution.py +++ b/tensorflow_probability/python/distributions/distribution.py @@ -492,6 +492,9 @@ def __init__(self, self._initial_parameter_control_dependencies = tuple( d for d in self._parameter_control_dependencies(is_init=True) if d is not None) + if self._initial_parameter_control_dependencies: + self._initial_parameter_control_dependencies = ( + tf.group(*self._initial_parameter_control_dependencies),) @classmethod def param_shapes(cls, sample_shape, name='DistributionParamShapes'):