"""Module containing Quantization abstraction and quantizers."""

# quantize with custom quantization parameterization or implementation, or
# handle custom Keras layers.
from tensorflow_model_optimization.python.core.quantization.keras.quantizers import LastValueQuantizer
from tensorflow_model_optimization.python.core.quantization.keras.quantizers import MovingAverageQuantizer
from tensorflow_model_optimization.python.core.quantization.keras.quantizers import Quantizer

"""Module containing Quantization abstraction and quantizers."""

# quantize with custom quantization parameterization or implementation, or
# handle custom Keras layers.
from tensorflow_model_optimization.python.core.quantization.keras.quantizers import AllValuesQuantizer
from tensorflow_model_optimization.python.core.quantization.keras.quantizers import LastValueQuantizer
from tensorflow_model_optimization.python.core.quantization.keras.quantizers import MovingAverageQuantizer
from tensorflow_model_optimization.python.core.quantization.keras.quantizers import Quantizer

Include AllValuesQuantizer in external APIs
Include AllValuesQuantizer in external APIs

PiperOrigin-RevId: 320104499

  """Module containing Quantization abstraction and quantizers."""
  
  # quantize with custom quantization parameterization or implementation, or
  # handle custom Keras layers.
+ from tensorflow_model_optimization.python.core.quantization.keras.quantizers import AllValuesQuantizer
  from tensorflow_model_optimization.python.core.quantization.keras.quantizers import LastValueQuantizer
  from tensorflow_model_optimization.python.core.quantization.keras.quantizers import MovingAverageQuantizer
  from tensorflow_model_optimization.python.core.quantization.keras.quantizers import Quantizer
  
Include AllValuesQuantizer in external APIs
## Code Before:
"""Module containing Quantization abstraction and quantizers."""

# quantize with custom quantization parameterization or implementation, or
# handle custom Keras layers.
from tensorflow_model_optimization.python.core.quantization.keras.quantizers import LastValueQuantizer
from tensorflow_model_optimization.python.core.quantization.keras.quantizers import MovingAverageQuantizer
from tensorflow_model_optimization.python.core.quantization.keras.quantizers import Quantizer

## Instruction:
Include AllValuesQuantizer in external APIs
## Code After:
"""Module containing Quantization abstraction and quantizers."""

# quantize with custom quantization parameterization or implementation, or
# handle custom Keras layers.
from tensorflow_model_optimization.python.core.quantization.keras.quantizers import AllValuesQuantizer
from tensorflow_model_optimization.python.core.quantization.keras.quantizers import LastValueQuantizer
from tensorflow_model_optimization.python.core.quantization.keras.quantizers import MovingAverageQuantizer
from tensorflow_model_optimization.python.core.quantization.keras.quantizers import Quantizer
