Remove the deprecated tf.keras.dtensor.experimental.layout_map_scope API.

The deprecated warning was showing up when using the `LayoutMap.scope`, which is confusing to end user.

PiperOrigin-RevId: 526818762

Author: qlzh727

site/en/tutorials/distribute/dtensor_keras_tutorial.ipynb

@@ -634,7 +634,7 @@
         "layout_map['feature.*kernel'] = dtensor.Layout.batch_sharded(mesh, 'batch', rank=2)\n",
         "layout_map['feature.*bias'] = dtensor.Layout.batch_sharded(mesh, 'batch', rank=1)\n",
         "\n",
-        "with tf.keras.dtensor.experimental.layout_map_scope(layout_map):\n",
+        "with layout_map.scope():\n",
         "  subclassed_model = SubclassedModel()"
       ]
     },
@@ -713,7 +713,7 @@
       },
       "outputs": [],
       "source": [
-        "with tf.keras.dtensor.experimental.layout_map_scope(layout_map):\n",
+        "with layout_map.scope():\n",
         "  inputs = tf.keras.Input((16,), batch_size=16)\n",
         "  x = tf.keras.layers.Dense(16, name='feature')(inputs)\n",
         "  x = tf.keras.layers.Dropout(0.1)(x)\n",
@@ -731,7 +731,7 @@
       },
       "outputs": [],
       "source": [
-        "with tf.keras.dtensor.experimental.layout_map_scope(layout_map):\n",
+        "with layout_map.scope():\n",
         "  model = tf.keras.Sequential([\n",
         "      tf.keras.layers.Dense(16, name='feature', input_shape=(16,)),\n",
         "      tf.keras.layers.Dropout(0.1),\n",
@@ -744,8 +744,8 @@
   ],
   "metadata": {
     "colab": {
-      "collapsed_sections": [],
       "name": "dtensor_keras_tutorial.ipynb",
+      "provenance": [],
       "toc_visible": true
     },
     "kernelspec": {