def graph_fn():
    y_coordinates = tf.constant([0.5], dtype=tf.float32)
    x_coordinates = tf.constant([4.5], dtype=tf.float32)
    output = ta_utils.get_surrounding_grids(height=3, width=5, y_coordinates=y_coordinates, x_coordinates=x_coordinates, radius=1)
    return output
def graph_fn():
    y_coordinates = tf.constant([0.5], dtype=tf.float32)
    x_coordinates = tf.constant([4.5], dtype=tf.float32)
    output = ta_utils.get_surrounding_grids(height=3, width=5, y_coordinates=y_coordinates, x_coordinates=x_coordinates, radius=1)
    return output