From 944c7463b8bc753027a80c0f3df8a410fd34e9aa Mon Sep 17 00:00:00 2001
From: Vishnuvardhan Janapati
 <46058173+jvishnuvardhan@users.noreply.github.com>
Date: Thu, 16 Sep 2021 20:44:12 -0700
Subject: [PATCH] Update keras_example.ipynb

Removed  `experimental` references in the guide.
---
 docs/keras_example.ipynb | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/docs/keras_example.ipynb b/docs/keras_example.ipynb
index 90d6dad1733..af5ca2b0178 100644
--- a/docs/keras_example.ipynb
+++ b/docs/keras_example.ipynb
@@ -130,11 +130,11 @@
         "  return tf.cast(image, tf.float32) / 255., label\n",
         "\n",
         "ds_train = ds_train.map(\n",
-        "    normalize_img, num_parallel_calls=tf.data.experimental.AUTOTUNE)\n",
+        "    normalize_img, num_parallel_calls=tf.data.AUTOTUNE)\n",
         "ds_train = ds_train.cache()\n",
         "ds_train = ds_train.shuffle(ds_info.splits['train'].num_examples)\n",
         "ds_train = ds_train.batch(128)\n",
-        "ds_train = ds_train.prefetch(tf.data.experimental.AUTOTUNE)"
+        "ds_train = ds_train.prefetch(tf.data.AUTOTUNE)"
       ]
     },
     {
@@ -160,10 +160,10 @@
       "outputs": [],
       "source": [
         "ds_test = ds_test.map(\n",
-        "    normalize_img, num_parallel_calls=tf.data.experimental.AUTOTUNE)\n",
+        "    normalize_img, num_parallel_calls=tf.data.AUTOTUNE)\n",
         "ds_test = ds_test.batch(128)\n",
         "ds_test = ds_test.cache()\n",
-        "ds_test = ds_test.prefetch(tf.data.experimental.AUTOTUNE)"
+        "ds_test = ds_test.prefetch(tf.data.AUTOTUNE)"
       ]
     },
     {