TensorFlow Probability experimental python utilities.
Classes
class DeferredModule
: Wrapper to defer initialization of a tf.Module
instance.
class JitPublicMethods
: Wrapper to compile an object's public methods using XLA.
Functions
as_composite(...)
make_trainable(...)
: Constructs a distribution or bijector instance with trainable parameters.
make_trainable_stateless(...)
: Constructs a distribution or bijector instance with trainable parameters.
register_composite(...)
Other Members |
DEFAULT_METHODS_EXCLUDED_FROM_JIT
|
('event_shape',
'event_shape_tensor',
'batch_shape',
'batch_shape_tensor',
'dtype',
'kl_divergence',
'experimental_default_event_space_bijector',
'experimental_local_measure',
'forward_event_shape',
'forward_event_shape_tensor',
'inverse_event_shape',
'inverse_event_shape_tensor',
'forward_dtype',
'inverse_dtype',
'forward_event_ndims',
'inverse_event_ndims',
'experimental_compute_density_correction')
|