code: def _trace(layer, x, i):
      name = _try_get_name(layer)
      z = tf.nest.map_structure(
          lambda x_: "{:14} {:<24} {:>10}".format(  # pylint: disable=g-long-lambda
              _try_get_name(x_),
              str(list(getattr(x_, "shape", "?"))),
              _try_get_name(getattr(x_, "dtype", x_), "?"),
          ),
          x,
      )
      print("--- TRACE{:02}:  {:<24} {}".format(i, name, z))
      sys.stdout.flush()
  
imports: import sys
  import tensorflow.compat.v2
  from tensorflow_probability.python.experimental.nn.util import utils
  from tensorflow_probability.python.internal import name_util
label_desc: Use image node_id 3 for calling a global function with example usage: _trace(layer, x, i) without return types