|---| | [ View source on GitHub](https://github.com/tensorflow/probability/blob/v0.25.0/tensorflow_probability/python/experimental/nn/util/utils.py#L322-L336) | Returns a function which prints info related to input. tfp.experimental.nn.util.trace( name=None )