Function fn

imperative/python/megengine/quantization/quantize.py:173–175

(mod: Module)

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)