def _parameter_properties(cls, dtype, num_classes=None):
return dict(
distribution=parameter_properties.BatchedComponentProperties(
# TODO(davmre): replace with `self.reinterpreted_batch_ndims` once
# support for `reinterpreted_batch_ndims=None` has been removed.
event_ndims=lambda self: self._get_reinterpreted_batch_ndims()), # pylint: disable=protected-access
reinterpreted_batch_ndims=(
parameter_properties.ShapeParameterProperties()))
def _parameter_properties(cls, dtype, num_classes=None):
return dict(
proposal_distribution=parameter_properties.
BatchedComponentProperties(),
importance_sample_size=parameter_properties.
ShapeParameterProperties(),
sample_size=parameter_properties.ShapeParameterProperties(),
stochastic_approximation_seed=parameter_properties.
ParameterProperties(event_ndims=None))
def _parameter_properties(cls, dtype):
return dict(
input_structure=parameter_properties.ShapeParameterProperties(),
output_structure=parameter_properties.ShapeParameterProperties())
def _parameter_properties(cls, dtype, num_classes=None):
return dict(
distribution=parameter_properties.BatchedComponentProperties(),
sample_shape=parameter_properties.ShapeParameterProperties())
def _parameter_properties(cls, dtype, num_classes=None):
return dict(initial_state_prior=parameter_properties.
BatchedComponentProperties(),
num_steps=parameter_properties.ShapeParameterProperties())
def _parameter_properties(cls, dtype):
return dict(
event_shape_out=parameter_properties.ShapeParameterProperties(),
event_shape_in=parameter_properties.ShapeParameterProperties())
def _parameter_properties(cls, dtype):
return dict(
num_or_size_splits=parameter_properties.ShapeParameterProperties(),
axis=parameter_properties.ShapeParameterProperties())
def _parameter_properties(cls, dtype):
return dict(paddings=parameter_properties.ShapeParameterProperties(),
constant_values=parameter_properties.ParameterProperties(),
axis=parameter_properties.ShapeParameterProperties())
def _parameter_properties(cls, dtype):
return dict(permutation=parameter_properties.ParameterProperties(
event_ndims=1,
default_constraining_bijector_fn=parameter_properties.
BIJECTOR_NOT_IMPLEMENTED),
axis=parameter_properties.ShapeParameterProperties())
def _parameter_properties(cls, dtype, num_classes=None):
return dict(
distributions=parameter_properties.BatchedComponentProperties(
event_ndims=lambda self: [0 for _ in self.distributions]),
axis=parameter_properties.ShapeParameterProperties())
def _parameter_properties(cls, dtype):
return dict(
rightmost_transposed_ndims=(
parameter_properties.ShapeParameterProperties(is_preferred=False)),
perm=parameter_properties.ShapeParameterProperties())
