flax

b35b60f2 - 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.

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. The default value of max_intra_op_parallelism is now 1. So explicitly setting it to 1 is no longer necessary. PiperOrigin-RevId: 360062191