op { graph_op_name: "TakeDataset" visibility: HIDDEN }