Searched refs:_get_dict (Results 1 – 1 of 1) sorted by relevance
407 def _get_dict(name, dict_or_tensor, key): function430 prediction = _get_dict('prediction', predictions, self.prediction_key)433 label = _get_dict('label', labels, self.label_key)