Function deconv_loader

imperative/python/megengine/traced_module/compat.py:318–342

(expr)

@register_functional_loader(("megengine.functional.nn", "conv_transpose2d"))
def deconv_loader(expr):
    args, kwargs = expr.args, expr.kwargs
    if not hasattr(expr, "version"):

        def orig_conv_transpose2d_signature(
            inp,
            weight,
            bias=None,
            stride=1,
            padding=0,
            dilation=1,
            groups=1,
            conv_mode="cross_correlation",
            compute_mode="default",
        ):
            pass

        args, kwargs = _convert_kwargs_to_args(
            orig_conv_transpose2d_signature, expr.args, expr.kwargs
        )
        expr.set_args_kwargs(*args, **kwargs)
    if len(args) == 9:
        args = list(args)
        args.insert(4, 0)  # output padding = 0
        expr.set_args_kwargs(*args, **kwargs)


@register_functional_loader(("megengine.functional.quantized", "conv_transpose2d"))