Add direct access to the choice attributes of OneOf
Add direct access to the choice attributes of OneOf

PiperOrigin-RevId: 223652382

tensorflow/datasets,tensorflow/datasets,tensorflow/datasets,tensorflow/datasets,tensorflow/datasets