Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tfp.experimental.nn.util.trace

View source on GitHub

Returns a function which prints info related to input.

tfp.experimental.nn.util.trace(
    name=None
)