tensorflow_probability/python/layers/distribution_layer.py
def params_size(event_size, num_components, name=None):
    """The number of `params` needed to create a single distribution."""
    with tf.compat.v1.name_scope(
        name, 'CategoricalMixtureOfOneHotCategorical_params_size',
        [event_size, num_components]):
      return MixtureSameFamily.params_size(
          num_components,
          OneHotCategorical.params_size(event_size, name=name),
          name=name)
def params_size(event_size, num_components, name=None):
    """The number of `params` needed to create a single distribution."""
    with tf.compat.v1.name_scope(
        name, 'CategoricalMixtureOfOneHotCategorical_params_size',
        [event_size, num_components]):
      return MixtureSameFamily.params_size(
          num_components,
          OneHotCategorical.params_size(event_size, name=name),
          name=name)
The number of `params` needed to create a single distribution.