Allow passing tf.distribute.InputOptions to Learner
Allow passing tf.distribute.InputOptions to Learner

This allows passing InputOptions which will be passed to strategy.distribute_datasets_from_function in order to control options for how the dataset is distributed.

PiperOrigin-RevId: 409476447
Change-Id: Ie73654f7a712952ab76f6d7c10efc600970a3193

tensorflow/agents,tensorflow/agents