Ejemplos de run_distributed_scope en Python

Ejemplo n.º 1

Archivo: keras_parameterized.py Proyecto: zzted/tensorflow

def _v2_distributed_test(f, test_or_class, *args, **kwargs):
    with context.eager_mode():
        with testing_utils.run_eagerly_scope(False):
            with testing_utils.run_distributed_scope(True):
                f(test_or_class, *args, **kwargs)

Ejemplo n.º 2

Archivo: keras_parameterized.py Proyecto: zzted/tensorflow

def _v1_graph_test(f, test_or_class, config, *args, **kwargs):
    with context.graph_mode(), testing_utils.run_eagerly_scope(False):
        with testing_utils.run_distributed_scope(False):
            with test_or_class.test_session(use_gpu=True, config=config):
                f(test_or_class, *args, **kwargs)