Function scope

tensorflow/python/distribute/hvd_strategy.py:1426–1428

(**kwargs)

@contextlib.contextmanager
def scope(**kwargs):
    with GraphRewriting.scope(**kwargs) as ctx:
        yield ctx


@contextlib.contextmanager