add_quant_dequant

class torch.ao.quantization.add_quant_dequant(module)

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

参数

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

• quantize (that we want to) –

返回

根据 qconfig，子模块被包装在 QuantWrapper 中的就地修改模块，或者包装输入模块的新 QuantWrapper 模块。后一种情况仅发生在输入模块是叶子模块且我们想对其进行量化时。