Include AllValuesQuantizer in default_8bit scheme

PiperOrigin-RevId: 320092094

Author: nutsiepully

tensorflow_model_optimization/python/core/quantization/keras/default_8bit/default_8bit_transforms.py

@@ -269,9 +269,8 @@ def pattern(self):
     return LayerPattern('InputLayer')
 
   def replacement(self, match_layer):
-    # TODO(pulkitb): Replace quantizer with InputLayer specific quantizer.
     quant_layer = quantize_layer.QuantizeLayer(
-        quantizers.MovingAverageQuantizer(
+        quantizers.AllValuesQuantizer(
             num_bits=8, per_axis=False, symmetric=False, narrow_range=False))
     layer_config = keras.layers.serialize(quant_layer)
     layer_config['name'] = quant_layer.name
@@ -285,7 +284,8 @@ def replacement(self, match_layer):
   def custom_objects(self):
     return {
         'QuantizeLayer': quantize_layer.QuantizeLayer,
-        'MovingAverageQuantizer': quantizers.MovingAverageQuantizer
+        'MovingAverageQuantizer': quantizers.MovingAverageQuantizer,
+        'AllValuesQuantizer': quantizers.AllValuesQuantizer
     }
 
 

tensorflow_model_optimization/python/core/quantization/keras/default_8bit/default_8bit_transforms_test.py

@@ -264,7 +264,7 @@ def testAddsQuantizeLayerAfterInputLayer(self):
           layer_after_input,
           quantize_layer.QuantizeLayer)
       self.assertIsInstance(
-          layer_after_input.quantizer, quantizers.MovingAverageQuantizer)
+          layer_after_input.quantizer, quantizers.AllValuesQuantizer)
 
   def testConcatTransform(self):
     r"""Tests the Concat Transform.