torch.ao.quantization.fuse_modules.fuse_known_modules#
- torch.ao.quantization.fuse_modules.fuse_known_modules(mod_list, is_qat, additional_fuser_method_mapping=None)[源代码]#
- Returns a list of modules that fuses the operations specified
in the input module list.
Fuses only the following sequence of modules: conv, bn conv, bn, relu conv, relu linear, bn linear, relu For these sequences, the first element in the output module list performs the fused operation. The rest of the elements are set to nn.Identity()