flax

The tf.data is working on changes which will make it no longer possible to set `tf.data.Options` by mutating `tf.data.Dataset.options()`. In preparation for this change, we plan to make the value returned by `tf.data.Dataset.options()` immutable. We have identified that the code in this CL would be affected and fix it beforehand.

#1044

Merged