code: def resize(image, image_size):
      return tf.image.resize(
          [image],
          [image_size, image_size],
          method=tf.image.ResizeMethod.BICUBIC,
      )[0]
  
file_path: elegy/examples/need-fixing/imagenet/input_pipeline.py
imports: import jax
  import tensorflow
  import tensorflow_datasets
label_desc: Use image node_id 2 for calling a global function with example usage: resize(image, image_size) and returns: tf