{
  "datasetType": "TF",
  "numRows": 5,
  "columns": {
    "next_sentence_labels": {
      "type": "int64",
      "rank": 1,
      "shape": [1]
    },
    "masked_lm_positions": {
      "type": "int64",
      "rank": 1,
      "shape": [20]
    },
    "masked_lm_ids": {
      "type": "int64",
      "rank": 1,
      "shape": [20]
    },
    "masked_lm_weights": {
      "type": "float32",
      "rank": 1,
      "shape": [20]
    },
    "input_mask": {
      "type": "int64",
      "rank": 1,
      "shape": [128]
    },
    "input_ids": {
      "type": "int64",
      "rank": 1,
      "shape": [128]
    },
    "segment_ids" : {
      "type": "int64",
      "rank": 1,
      "shape": [128]
    }
  }
}