0145cd504585e25b776bef83688d10ff0ca44082,python/tvm/relay/frontend/pytorch.py,,from_pytorch,#Any#Any#Any#,1677

Before Change

    ret = convert_operators(_get_operator_nodes(graph.nodes()),
                            outputs, ret_name)

    if isinstance(ret[0], list):
        ret[0] = _expr.Tuple(ret[0])

    func = tvm.relay.Function(_analysis.free_vars(ret[0]), ret[0])

    return _module.IRModule.from_expr(func), tvm_params

After Change

    params : dict of str to tvm.runtime.NDArray
        Dict of converted parameters stored in tvm.runtime.ndarray format
    
    mod = tvm.IRModule()
    prelude = Prelude(mod)

    convert_map = _get_convert_map(prelude)

    graph = script_module.graph.copy()
    _run_jit_passes(graph)

Instances