tfma.model_util.find_input_name_in_features

Maps input name to an entry in features. Returns None if not found.