def get_name_to_features(config: ml_collections.ConfigDict) -> Dict[str, Any]:
    """Return feature dict for decoding purposes. See BaseTask."""
    name_to_features = (
        mention_encoder_task.MentionEncoderTask.get_name_to_features(config))

    if config.apply_answer_mask:
      name_to_features['dense_answer_mask'] = tf.io.FixedLenFeature(
          config.model_config.encoder_config.max_length, tf.int64)

    return name_to_features
Return feature dict for decoding purposes. See BaseTask.
language/mentionmemory/tasks/entity_qa_task.py