Function _propagate

imperative/python/megengine/quantization/quantize.py:172–177

(module: Module, func_str: str, *args, **kargs)

def _propagate(module: Module, func_str: str, *args, **kargs):
    def fn(mod: Module):
        if isinstance(mod, QATModule):
            getattr(mod, func_str)(*args, **kargs)

    module.apply(fn)


def propagate_qconfig(module: QATModule, qconfig: QConfig):