code: def _get_grid_locations(
      image_height: TensorLike, image_width: TensorLike
  ) -> TensorLike:
      """Wrapper for `tf.meshgrid`."""
      y_range = tf.linspace(0, image_height - 1, image_height)
      x_range = tf.linspace(0, image_width - 1, image_width)
      y_grid, x_grid = tf.meshgrid(y_range, x_range, indexing="ij")
      return tf.stack((y_grid, x_grid), -1)
  
file_path: addons/tensorflow_addons/image/sparse_image_warp.py
imports: import tensorflow
  from tensorflow_addons.image import dense_image_warp
  from tensorflow_addons.image import interpolate_spline
  from tensorflow_addons.image import utils
  from tensorflow_addons.utils.types import TensorLike, FloatTensorLike
label_desc: Use image node_id 1 for calling a global function with example usage: _get_grid_locations(image_height, image_width) and returns: tf