description: TensorFlow Probability experimental python utilities.
View source on GitHub |
TensorFlow Probability experimental python utilities.
class DeferredModule
: Wrapper to defer initialization of a tf.Module
instance.
class JitPublicMethods
: Wrapper to compile an object's public methods using XLA.
as_composite(...)
: Returns a CompositeTensor
equivalent to the given object.
make_trainable(...)
: Constructs a distribution or bijector instance with trainable parameters.
register_composite(...)
: A decorator that registers a TFP object as composite-friendly.
Other Members | |
---|---|
DEFAULT_METHODS_EXCLUDED_FROM_JIT |