Function pad_func_loader

imperative/python/megengine/traced_module/compat.py:161–165

(expr)

@register_functional_loader(("megengine.functional.nn", "pad"))
def pad_func_loader(expr):
    if "pad_witdth" in expr.kwargs:
        kwargs = expr.kwargs
        kwargs["pad_width"] = kwargs.pop("pad_witdth")
        expr.set_args_kwargs(*expr.args, **kwargs)


@register_functional_loader(("megengine.functional.nn", "batch_norm"))