RSVP for your your local TensorFlow Everywhere event today!

tfp.experimental.nn.util.trace

Returns a function which prints info related to input.