def _load_tf_record(self, tf_record_path: str) -> RepresentativeDataset:
    """Loads TFRecord containing samples of type`RepresentativeDataSample`."""
    samples = []
    with context.eager_mode():
      for sample_bytes in readers.TFRecordDatasetV2(filenames=[tf_record_path]):
        sample_proto = _RepresentativeDataSample.FromString(
            sample_bytes.numpy()
        )
        sample = {}