{
  "datasetType": "TF",
  "numRows": 5,
  "columns": {
    "input_ids": {
      "type": "int64",
      "rank": 1,
      "shape": [128]
    },
    "segment_ids" : {
      "type": "int64",
      "rank": 1,
      "shape": [128]
    }
  }
}
