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