swap_module# class torch.ao.quantization.swap_module(mod, mapping, custom_module_class_mapping, use_precomputed_fake_quant=False)[source]# Swaps the module if it has a quantized counterpart and it has an observer attached. Parameters mod – input module mapping – a dictionary that maps from nn module to nnq module Returns The corresponding quantized module of mod
