tensorplay.ao.quantization.fuse_modules_qat

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

Quantization-aware training variant of fuse_modules().