orbit/utils/common.py · 13d1e402b755843ffa86c1c229af8b36ad39e435 · ModelZoo / ResNet50_tensorflow

Allow passing a `tf.distribute.InputOptions` to... · 13d1e402

Dinghua Li authored Oct 20, 2021

Allow passing a `tf.distribute.InputOptions` to `orbit.make_distributed_dataset` through the `input_options` key word argument.

PiperOrigin-RevId: 404597006

13d1e402

common.py 3.69 KB

Replace common.py