logits = head_lib._check_logits(logits, self.logits_dimension)# pylint:disable=protected-access# Predict.pred_keys = prediction_keys.PredictionKeyswithops.name_scope(None,'predictions', (logits,)): probabilities = math_ops.sigmoid(logits, name=pred_keys.PROBABILITIES) predictions = { pred_keys....