Searched refs:selected_id (Results 1 – 1 of 1) sorted by relevance
2235 def _select_class_id(ids, selected_id): argument2249 selected_id))2263 math_ops.cast(selected_id, dtypes.int64))2269 def _maybe_select_class_id(labels, predictions_idx, selected_id=None): argument2286 if selected_id is None:2288 return (_select_class_id(labels, selected_id),2289 _select_class_id(predictions_idx, selected_id))