Method wrapper

caffe2/python/context.py:95–97

(*args, **kwargs)

    def __call__(self, func):
        @functools.wraps(func)
        def wrapper(*args, **kwargs):
            with self:
                return func(*args, **kwargs)
        return wrapper