code: def test_output_shapes_priority_over_feature_config_and_build(self):
      _, mid_level_api, _ = self._create_strategy_and_mid_level("sgd")
      # The output shapes setting in the feature config has the first priority.
      mid_level_api._output_shapes = [
          TensorShape((2, 4)) for _ in range(3)
      ]
      mid_level_api.build(
          [TensorShape((2, None, None)) for _ in range(3)]
      )
      self.assertEqual(
          mid_level_api._output_shapes,
          [TensorShape((2, 4)) for _ in range(3)],
      )
  
file_path: tensorflow/tensorflow/python/tpu/tests/tpu_embedding_v2_hd_valid_input_test.py
imports: import numpy
  from tensorflow.python.compat import v2_compat
  from tensorflow.python.data.ops import dataset_ops
  from tensorflow.python.distribute import distribute_lib
  from tensorflow.python.eager import def_function
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework.tensor_shape import TensorShape
  from tensorflow.python.platform import test
  from tensorflow.python.tpu.tests import tpu_embedding_base_test
label_desc: Use image node_id 3 for calling the TPUEmbeddingTest obj's underlying member method code with example usage: obj.test_output_shapes_priority_over_feature_config_and_build() without return types