Add default quantize config for layers.Concatenate

PiperOrigin-RevId: 406274393

Author: daverim

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

@@ -92,6 +92,7 @@ class Default8BitQuantizeRegistry(
 
       # _QuantizeInfo(layers.Conv3D, ['kernel'], ['activation']),
       # _QuantizeInfo(layers.Conv3DTranspose, ['kernel'], ['activation']),
+      _QuantizeInfo(layers.Concatenate, [], [], True),
       _no_quantize(layers.Cropping1D),
       _no_quantize(layers.Cropping2D),
       _no_quantize(layers.Cropping3D),

tensorflow_model_optimization/python/core/quantization/keras/experimental/default_n_bit/default_n_bit_quantize_registry.py

@@ -101,6 +101,7 @@ class DefaultNBitQuantizeRegistry(
 
       # _QuantizeInfo(layers.Conv3D, ['kernel'], ['activation']),
       # _QuantizeInfo(layers.Conv3DTranspose, ['kernel'], ['activation']),
+      _QuantizeInfo(layers.Concatenate, [], [], True),
       _no_quantize(layers.Cropping1D),
       _no_quantize(layers.Cropping2D),
       _no_quantize(layers.Cropping3D),