code: def _assert_array_equal(x, y, message, name=None):
      """TF assertion similar to checking `np.array_equal`."""
      with tf.name_scope(name or "array_equal_check"):
          rank_check = assert_util.assert_equal(
              tf.rank(x), tf.rank(y), message=message, name="rank_check"
          )
          shape_check = assert_util.assert_equal(
              tf.shape(x),
              tf.shape(y),
              message=message,
              name="shape_check",
          )
          with tf.control_dependencies([rank_check]):
              with tf.control_dependencies([shape_check]):
                  return assert_util.assert_equal(
                      x, y, message=message, name="value_check"
                  )
  
imports: import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import assert_util
  from tensorflow_probability.python.internal import distribution_util
  from tensorflow_probability.python.internal import dtype_util
  from tensorflow_probability.python.internal import prefer_static
  from tensorflow_probability.python.internal import tensorshape_util
  from tensorflow_probability.python.math.generic import reduce_logmeanexp
label_desc: Use image node_id 16 for calling a global function with example usage: _assert_array_equal(x, y, message, name) and returns: assert_util