def graph_fn():
    y_source = tf.constant([[1.0, 0.0], [2.0, 3.0]], dtype=tf.float32)
    x_source = tf.constant([[2.0, 4.0], [3.0, 3.0]], dtype=tf.float32)
    y_target = tf.constant([2.1, 0.1], dtype=tf.float32)
    x_target = tf.constant([1.2, 4.5], dtype=tf.float32)
    (offsets, indices) = ta_utils.compute_floor_offsets_with_indices(y_source, x_source, y_target, x_target)
    return (offsets, indices)
def graph_fn():
    y_source = tf.constant([[1.0, 0.0], [2.0, 3.0]], dtype=tf.float32)
    x_source = tf.constant([[2.0, 4.0], [3.0, 3.0]], dtype=tf.float32)
    y_target = tf.constant([2.1, 0.1], dtype=tf.float32)
    x_target = tf.constant([1.2, 4.5], dtype=tf.float32)
    (offsets, indices) = ta_utils.compute_floor_offsets_with_indices(y_source, x_source, y_target, x_target)
    return (offsets, indices)