description: TensorFlow Probability experimental python utilities.

Module: tfp.experimental.util

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

DEFAULT_METHODS_EXCLUDED_FROM_JIT