Merge pull request #1906 from rabinadk1:patch-1

PiperOrigin-RevId: 378748819

Author: copybara-github

site/en/guide/tpu.ipynb

@@ -348,7 +348,7 @@
         "id": "8hSGBIYtUugJ"
       },
       "source": [
-        "To reduce Python overhead and maximize the performance of your TPU, pass in the **experimental** argument—`experimental_steps_per_execution`—to `Model.compile`. In this example, it increases throughput by about 50%:"
+        "To reduce Python overhead and maximize the performance of your TPU, pass in the argument—`steps_per_execution`—to `Model.compile`. In this example, it increases throughput by about 50%:"
       ]
     },
     {
@@ -363,7 +363,7 @@
         "  model = create_model()\n",
         "  model.compile(optimizer='adam',\n",
         "                # Anything between 2 and `steps_per_epoch` could help here.\n",
-        "                experimental_steps_per_execution = 50,\n",
+        "                steps_per_execution = 50,\n",
         "                loss=tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True),\n",
         "                metrics=['sparse_categorical_accuracy'])\n",
         "\n",