diff --git a/tensorflow/python/data/experimental/ops/optimization_options.py b/tensorflow/python/data/experimental/ops/optimization_options.py index 41a819d94bb..eb95484ac53 100644 --- a/tensorflow/python/data/experimental/ops/optimization_options.py +++ b/tensorflow/python/data/experimental/ops/optimization_options.py @@ -33,7 +33,7 @@ class OptimizationOptions(options.OptionsBase): ```python options = tf.data.Options() options.experimental_optimization.map_vectorization = True - options.apply_default_optimizations = False + options.experimental_optimization.apply_default_optimizations = False dataset = dataset.with_options(options) ``` """