def METHOD_NAME(self):
    logger.info("Using TFRecords to load {} data.".format(self._split))
    dataset_key = self._dataset_name.replace("/", "")
    if dataset_key in self._tfrecord_datasets:
        self._dataset_loader = self._tfrecord_datasets[dataset_key](config=self._config, is_train=self._is_train)
    else:
        raise ValueError("Unknown dataset name: {}".format(self._dataset_name))
    dataset = self._dataset_loader.as_dataset()
    return dataset