code: def assert_replica_values_are_close(test_case_obj, replica_context):
      """Checks if all replica context tensors are near each other."""
      base_tensor = replica_context.values[0]
      for t in replica_context.values[1:]:
          test_case_obj.assertAllClose(base_tensor, t)
  
imports: from collections.abc import Callable, MutableSequence, Sequence
  from typing import Optional
  import numpy
  import tensorflow
  import tensorflow.compat.v2
  from tensorflow_privacy.privacy.fast_gradient_clipping import clip_grads
  from tensorflow_privacy.privacy.fast_gradient_clipping import layer_registry
  from tensorflow_privacy.privacy.fast_gradient_clipping import type_aliases
label_desc: Use image node_id 2 for calling a global function with example usage: assert_replica_values_are_close(test_case_obj, replica_context) without return types