def graph_fn():
    calibration_fn = calibration_builder.build(calibration_config)
    class_predictions_with_background = tf.constant([[[0.1, 0.2], [0.9, 0.1]], [[0.6, 0.4], [0.08, 0.92]]], dtype=tf.float32)
    calibrated_scores = calibration_fn(class_predictions_with_background)
    return calibrated_scores
def graph_fn():
    calibration_fn = calibration_builder.build(calibration_config)
    class_predictions_with_background = tf.constant([[[0.1, 0.2], [0.9, 0.1]], [[0.6, 0.4], [0.08, 0.92]]], dtype=tf.float32)
    calibrated_scores = calibration_fn(class_predictions_with_background)
    return calibrated_scores