yamle.quantization package

yamle.quantization.quantizer_factory(quantizer_type=None)

This function is used to create a quantizer instance based on the quantizer type.

Parameters:

quantizer_type (str) – The type of pruner to create.

Return type:

Type[Callable]

Subpackages

• yamle.quantization.models package
  • Subpackages
    • yamle.quantization.models.specific package
      • Submodules
  • Submodules
    • yamle.quantization.models.operations module
      • QuantizableAdd
• yamle.quantization.qat package
  • QATQuantizer
    • QATQuantizer.prepare()
    • QATQuantizer.cleanup()
    • QATQuantizer.get_qconfig()
    • QATQuantizer.add_specific_args()
• yamle.quantization.static package
  • StaticQuantizer
    • StaticQuantizer.prepare()
    • StaticQuantizer.get_qconfig()

Submodules

• yamle.quantization.quantizer module
  • BaseQuantizer
    • BaseQuantizer.prepare()
    • BaseQuantizer.get_qconfig()
    • BaseQuantizer.cleanup()
    • BaseQuantizer.save_original_model()
    • BaseQuantizer.save_quantized_model()
    • BaseQuantizer.recover()
    • BaseQuantizer.add_specific_args()
    • BaseQuantizer.replace_layers_for_quantization()
  • DummyQuantizer
    • DummyQuantizer.prepare()
    • DummyQuantizer.get_qconfig()