Function Func

python/oneflow/framework/session_context.py:57–59

(*args, **kwargs)

def try_init_default_session(func):
    @functools.wraps(func)
    def Func(*args, **kwargs):
        GetDefaultSession().TryInit()
        return func(*args, **kwargs)

    return Func