add_quant_dequant¶
- class torch.ao.quantization.add_quant_dequant(module)[source][source]¶
Wrap the leaf child module in QuantWrapper if it has a valid qconfig Note that this function will modify the children of module inplace and it can return a new module which wraps the input module as well.
- Parameters
module – input module with qconfig attributes for all the leaf modules
quantize (that we want to) –
- Returns
Either the inplace modified module with submodules wrapped in QuantWrapper based on qconfig or a new QuantWrapper module which wraps the input module, the latter case only happens when the input module is a leaf module and we want to quantize it.
