add_quant_dequant

class torch.ao.quantization.add_quant_dequant(module) [исходный код]

Оборачивает листовой дочерний модуль в QuantWrapper, если для него задан допустимый qconfig. Обратите внимание, что эта функция изменяет дочерние модули на месте и также может возвращать новый модуль, оборачивающий входной модуль.

Параметры:

• module – входной модуль с атрибутами qconfig для всех листовых модулей
• quantize (который мы хотим квантовать) –

Возвращает:

Либо изменённый на месте модуль с подмодулями, обёрнутыми в QuantWrapper на основе qconfig, либо новый модуль QuantWrapper, оборачивающий входной модуль. Последний вариант возникает только в том случае, если входной модуль является листовым модулем и мы хотим его квантовать.