add_quant_dequant

class torch.ao.quantization.add_quant_dequant(module)

如果叶子子模块具有有效的 qconfig，则将其包装在 QuantWrapper 中。请注意，此函数将就地修改模块的子模块，并且它可以返回一个新的模块，该模块也包装了输入模块。

参数

• module – 输入模块，所有叶子模块都具有 qconfig 属性

• 量化 (我们想要) –

返回值

根据 qconfig，就地修改的模块（子模块包装在 QuantWrapper 中）或新的 QuantWrapper 模块（包装输入模块）。后一种情况仅在输入模块是叶子模块且我们想要量化它时发生。