code: def test_check_checkpoint_variable_names_are_same_on_cpu_and_tpu(
      self, optimizer
  ):
      # Reinitialize the TPU so that we can re-initialize the embeddings with the
      # given optimizer.
      if optimizer != tpu_embedding_v2_utils.SGD:
          self.skip_if_oss()
      strategy = self._get_strategy()
  
      with strategy.scope():
          first_mid_level_contents = np.ones((4, 4))
          first_mid_level_optimizer = optimizer(learning_rate=0.1)
          initializer = init_ops_v2.Constant(first_mid_level_contents)
  
          table = tpu_embedding_v2_utils.TableConfig(
              vocabulary_size=4,
              dim=4,
              initializer=initializer,
              combiner="sum",
              name="table",
          )
          feature_config = (
              tpu_embedding_v2_utils.FeatureConfig(
                  table=table, name="feature"
              ),
          )
  
          first_mid_level = tpu_embedding_v1.TPUEmbeddingV0(
              feature_config, first_mid_level_optimizer
          )
  
          first_mid_level.build()
  
      cpu_mid_level_optimizer = optimizer(learning_rate=0.1)
      cpu_mid_level = tpu_embedding_for_serving.TPUEmbeddingForServing(
          feature_config, cpu_mid_level_optimizer
      )
      cpu_mid_level.build()
  
      tpu_checkpoint = util.Checkpoint(model=first_mid_level)
      tpu_checkpoint.save(self._get_tmpdir("save-tpu", "save"))
      tpu_variables = checkpoint_utils.list_variables(
          self._get_tmpdir("save-tpu")
      )
  
      cpu_checkpoint = util.Checkpoint(model=cpu_mid_level)
      cpu_checkpoint.save(self._get_tmpdir("save-cpu", "save"))
      cpu_variables = checkpoint_utils.list_variables(
          self._get_tmpdir("save-cpu")
      )
  
      self.assertAllEqual(tpu_variables, cpu_variables)
  
imports: from absl.testing import parameterized
  import numpy
  from tensorflow.python.checkpoint import checkpoint
  from tensorflow.python.compat import v2_compat
  from tensorflow.python.eager import def_function
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import tensor_spec
  from tensorflow.python.ops import init_ops_v2
  from tensorflow.python.platform import test
  from tensorflow.python.saved_model import load
  from tensorflow.python.saved_model import save
  from tensorflow.python.tpu import tpu_embedding_for_serving
  from tensorflow.python.tpu import tpu_embedding_v1
  from tensorflow.python.tpu import tpu_embedding_v2_utils
  from tensorflow.python.tpu.tests import tpu_embedding_base_test
  from tensorflow.python.training import checkpoint_utils
label_desc: Use image node_id 5 for calling the TPUEmbeddingCheckpointTest obj's underlying member method code with example usage: obj.test_check_checkpoint_variable_names_are_same_on_cpu_and_tpu(optimizer) without return types