tf.distribute.InputOptions

Параметры запуска для experimental_distribute_dataset(s_from_function).

tf.distribute.InputOptions(
    experimental_prefetch_to_device=True
)

Это можно использовать для хранения некоторых параметров, специфичных для стратегии.

# Setup TPUStrategy
resolver = tf.distribute.cluster_resolver.TPUClusterResolver(tpu='')
tf.config.experimental_connect_to_cluster(resolver)
tf.tpu.experimental.initialize_tpu_system(resolver)
strategy = tf.distribute.TPUStrategy(resolver)

dataset = tf.data.Dataset.range(16)
distributed_dataset_on_host = (
    strategy.experimental_distribute_dataset(
        dataset,
        tf.distribute.InputOptions(
            experimental_prefetch_to_device=False)))