tensorplay.ao.quantization.fuse_modules

tensorplay.ao.quantization.fuse_modules(model, modules_to_fuse, inplace=False, fuser_func=None)[source]

Fuse supported modules in-place within a model.

Parameters:

• model – the model to fuse.

• modules_to_fuse – an iterable of module-name groups, each a list of fully-qualified attribute paths, e.g. [["conv1", "bn1", "relu1"], ["fc"]]. Groups of length one are ignored (kept for call-site symmetry).

• inplace – mutate the model instead of returning a copy.

• fuser_func – replacement for fuse_known_modules().

Returns:

The fused model (the same object when inplace).