code: def _create_mid_level(self, optimizer=None):
      # Create `TPUEmbedding` object.
      if optimizer is None:
          optimizer = tpu_embedding_v2_utils.SGD(learning_rate=0.1)
  
      return tpu_embedding_v1.TPUEmbeddingV0(
          feature_config=self.feature_config, optimizer=optimizer
      )
  
file_path: tensorflow/tensorflow/python/tpu/tests/tpu_embedding_v1_correctness_test.py
imports: from absl.testing import parameterized
  from tensorflow.python.compat import v2_compat
  from tensorflow.python.distribute import distribute_lib
  from tensorflow.python.eager import def_function
  from tensorflow.python.platform import test
  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
label_desc: Use image node_id 2 for calling the TPUEmbeddingV0CorrectnessTest obj's underlying member method code with example usage: obj._create_mid_level(optimizer) and returns: tpu_embedding_v1