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

keras/api/golden/v2/tensorflow.keras.dtensor.experimental.pbtxt

@@ -8,8 +8,4 @@ tf_module {
  name: "optimizers"
  mtype: "<type \'module\'>"
  }
- member_method {
- name: "layout_map_scope"
- argspec: "args=[], varargs=args, keywords=kwds, defaults=None"
- }
 }

keras/dtensor/layout_map.py

@@ -27,7 +27,6 @@
 from keras.engine import base_layer
 
 # isort: off
-from tensorflow.python.util.deprecation import deprecated
 from tensorflow.python.util.tf_export import keras_export
 
 
@@ -235,10 +234,6 @@ def call(self, inputs):
 LayoutMap.get.__doc__ = LayoutMap.__getitem__.__doc__
 
 
-@keras_export("keras.dtensor.experimental.layout_map_scope", v1=[])
-@deprecated(
- None, "use tf.keras.dtensor.experimental.LayoutMap.scope() instead."
-)
 @contextlib.contextmanager
 def layout_map_scope(layout_map):
  """Apply the layout to all the tf.Variables created under the scope.

keras/dtensor/layout_map_test.py

@@ -340,7 +340,7 @@ def test_init_model_with_empty_layout_map(self):
 
  def test_weight_regularization(self):
  layout_map = layout_map_lib.LayoutMap(mesh=self.mesh)
- with layout_map_lib.layout_map_scope(layout_map):
+ with layout_map.scope():
  model = models.Sequential(
  [
  layers.Dense(