def METHOD_NAME(self, y_true, y_pred):
    return ops.vectorized_map(
        lambda y_true_and_pred: self.regression_loss(
            y_true_and_pred[0], y_true_and_pred[1]
        ),
        (y_true, y_pred),
    )