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

    name_to_features = {
        'x':
            tf.io.FixedLenFeature([config.model_config.features[0]],
                                  tf.float32),
    }

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