message: Remove incorrect sanity check for `add_update`.
diff: @@ -1229,14 +1229,6 @@ class Layer(module.Module):
  # ignored, following the default path for adding updates.
  not call_context.saving):
  # Updates don't need to be run in a cross-replica context.
  - # TODO(b/142574744): Relax this restriction so that metrics/variables
  - # created outside of a strategy scope can be updated in the cross-replica
  - # context.
  - if (ops.executing_eagerly_outside_functions() and
  - not base_layer_utils.is_in_keras_graph()):
  - raise RuntimeError( # pylint: disable=g-doc-exception
  - '`add_update` was called in a cross-replica context. This is not '
  - 'expected. If you require this feature, please file an issue.')
  return
  updates = generic_utils.to_list(updates)
  