def graph_fn():
    (y_grid, x_grid) = ta_utils.image_shape_to_grids(height=2, width=3)
    return (y_grid, x_grid)
def graph_fn():
    (y_grid, x_grid) = ta_utils.image_shape_to_grids(height=2, width=3)
    return (y_grid, x_grid)