Google I/O returns May 18-20! Reserve space and build your schedule Register now

tfp.experimental.nn.util.trace

Returns a function which prints info related to input.