code: def _at_least_x_are_equal(a, b, x):
      """At least `x` of `a` and `b` `Tensors` are equal."""
      match = tf.equal(a, b)
      match = tf.cast(match, tf.int32)
      return tf.greater_equal(tf.reduce_sum(match), x)
  
file_path: scenic/scenic/dataset_lib/imagenet_dataset.py
imports: import functools
  from typing import Optional
  from absl import logging
  from flax import jax_utils
  import jax
  import jax.numpy
  from scenic.dataset_lib import dataset_utils
  from scenic.dataset_lib import datasets
  import tensorflow
  import tensorflow_datasets
label_desc: Use image node_id 3 for calling a global function with example usage: _at_least_x_are_equal(a, b, x) and returns: tf