def __call__(self, record):
        """Decodes a BERT TF record to a TF example."""
        example = tf.parse_single_example(record, self.feature_map)
        for k, v in list(example.items()):
            if v.dtype == tf.int64:
                example[k] = tf.to_int32(v)
        return example
Decodes a BERT TF record to a TF example.
megatron_lm/megatron/deprecated_data_utils/tf_dl.py