code: def testSerializationQuantizeWrapper(self):
      input_shape = (2,)
      layer = keras.layers.Dense(3)
      wrapper = QuantizeWrapper(
          layer=layer,
          quantize_config=self.quantize_registry.get_quantize_config(
              layer
          ),
          input_shape=input_shape,
      )
  
      custom_objects = {
          "QuantizeAwareActivation": QuantizeAwareActivation,
          "QuantizeWrapper": QuantizeWrapper,
      }
      custom_objects.update(
          default_8bit_quantize_registry._types_dict()
      )
  
      serialized_wrapper = serialize_layer(wrapper)
      with custom_object_scope(custom_objects):
          wrapper_from_config = deserialize_layer(serialized_wrapper)
  
      self.assertEqual(
          wrapper_from_config.get_config(), wrapper.get_config()
      )
  
file_path: model-optimization/tensorflow_model_optimization/python/core/quantization/keras/quantize_wrapper_test.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from absl.testing import parameterized
  import numpy
  import tensorflow
  from tensorflow_model_optimization.python.core.quantization.keras import quantize_aware_activation
  from tensorflow_model_optimization.python.core.quantization.keras import quantize_wrapper
  from tensorflow_model_optimization.python.core.quantization.keras.default_8bit import default_8bit_quantize_registry
label_desc: Use image node_id 5 for calling the QuantizeWrapperTest obj's underlying member method code with example usage: obj.testSerializationQuantizeWrapper() without return types