ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

tf.data.experimental.AutotuneOptions

Represents options for autotuning dataset performance.

options = tf.data.Options()
options.autotune.enabled = False
dataset = dataset.with_options(options)

autotune_algorithm When autotuning is enabled (through autotune), determines the algorithm to use.
cpu_budget When autotuning is enabled (through autotune), determines the CPU budget to use. Values greater than the number of schedulable CPU cores are allowed but may result in CPU contention. If None, defaults to the number of schedulable CPU cores.
enabled Whether to automatically tune performance knobs. If None, defaults to True.
ram_budget When autotuning is enabled (through autotune), determines the RAM budget to use. Values greater than the available RAM in bytes may result in OOM. If None, defaults to half of the available RAM in bytes.

Methods

__eq__

View source

Return self==value.

__ne__

View source

Return self!=value.