code: def _assert_rank(
      x, rank, message=None, name=None
  ):  # pylint: disable=unused-argument
      return _assert_equal(x=len(np.shape(x)), y=rank, message=message)
  
imports: import numpy
  import six
  from tensorflow_probability.python.internal.backend.numpy import _utils
  from tensorflow_probability.python.internal.backend.numpy import dtype
  from tensorflow_probability.python.internal.backend.numpy import ops
label_desc: Use image node_id 14 for calling a global function with example usage: _assert_rank(x, rank, message, name) and returns: _assert_equal