tensorplay.ao.quantization.fuse_modules

tensorplay.ao.quantization.fuse_modules(model, modules_to_fuse, inplace=False, fuser_func=<function fuse_known_modules>, fuse_custom_config_dict=None)[source]

Fuse a list of modules into a single module.

Supports conv-bn, conv-bn-relu, conv-relu, linear-relu and bn-relu sequences. The first listed module is replaced by the fused module and the remaining entries become identity modules.