add_quant_dequant

class torch.ao.quantization.add_quant_dequant(module)

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

参数

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

• quantize (that we want to) –

返回值

要么是就地修改的模块，其子模块基于 qconfig 包装在 QuantWrapper 中，要么是新的 QuantWrapper 模块，它包装了输入模块，后一种情况仅发生在输入模块是叶子模块且我们想要量化它时。