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())