添加量化与去量化的操作
- 类 torch.ao.quantization.add_quant_dequant(module) [源代码]
-
如果叶级子模块具有有效的qconfig,则使用QuantWrapper将其包裹起来。需要注意的是,此函数会直接修改模块的子模块,并可能返回一个包装了输入模块的新模块。
- 参数
-
-
module - 输入模块,包含所有叶子模块的 qconfig 属性
-
量化(我们希望实现的)–
-
- 返回值
-
要么是对现有模块进行就地修改,并使用基于 qconfig 的 QuantWrapper 包装其子模块,要么是创建一个新的 QuantWrapper 模块来封装输入模块。后一种情况仅在输入模块为叶模块且需要对其进行量化时发生。