ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

tfp.experimental.nn.util.trace

Returns a function which prints info related to input.