Home
last modified time | relevance | path

Searched refs:num_ps (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/contrib/opt/python/training/
Delastic_average_optimizer_test.py41 def create_local_cluster(num_workers, num_ps, protocol="grpc"): argument
44 ps_ports = [portpicker.pick_unused_port() for _ in range(num_ps)]
59 for ix in range(num_ps)
67 def _get_workers(num_workers, period, workers, moving_rate, num_ps=1): argument
87 if num_ps > 1:
91 num_ps, axis=0),
96 ps_tasks=num_ps)):
118 if num_ps == 1:
153 num_ps = 1
155 num_workers=num_workers, num_ps=num_ps)
[all …]
Dagn_optimizer_test.py38 def create_local_cluster(num_workers, num_ps, protocol="grpc"): argument
41 ps_ports = [portpicker.pick_unused_port() for _ in range(num_ps)]
56 for ix in range(num_ps)
64 def _get_workers(num_workers, period, workers, num_ps=1): argument
83 if num_ps > 1:
87 num_ps, axis=0),
110 if num_ps == 1:
141 num_ps = 1
142 _, workers, _ = create_local_cluster(num_workers=num_workers, num_ps=num_ps)
202 num_ps = 2
[all …]
Dmodel_average_optimizer_test.py35 def create_local_cluster(num_workers, num_ps, protocol="grpc"): argument
38 ps_ports = [portpicker.pick_unused_port() for _ in range(num_ps)]
53 for ix in range(num_ps)
120 num_ps = 1
122 num_workers=num_workers, num_ps=num_ps)
/external/tensorflow/tensorflow/contrib/distribute/python/
Dmulti_worker_test_base.py70 num_ps, argument
80 ps_ports = [pick_unused_port() for _ in range(num_ps)]
85 if num_ps > 0:
103 for i in range(num_ps):
134 num_ps, argument
167 num_ps=num_ps,
177 num_ps=0, argument
190 if num_ps:
192 'localhost:%s' % pick_unused_port() for _ in range(num_ps)
205 cls._cluster_spec = create_in_process_cluster(num_workers=2, num_ps=0)
Destimator_training_test.py81 num_workers=3, num_ps=2, has_eval=True)
395 num_workers=3, num_ps=2, has_eval=True)
400 num_workers=3, num_ps=0, has_eval=True)
441 num_workers=3, num_ps=0, has_eval=True)
Dkeras_multi_worker_test.py305 num_workers=2, num_ps=1, has_eval=False)
402 num_workers=num_workers, num_ps=2)
Dcollective_all_reduce_strategy_test.py357 num_workers=3, num_ps=0)
510 num_workers=3, num_ps=0, has_chief=True)
Dparameter_server_strategy_test.py617 num_workers=3, num_ps=2)
852 num_workers=3, num_ps=2, has_chief=True)
Dcross_device_ops_test.py445 num_workers=NUM_WORKERS, num_ps=0)
Dmirrored_strategy_multigpu_test.py1535 num_workers=2, num_ps=0, has_chief=True)
/external/grpc-grpc/test/core/iomgr/
Dpollset_set_test.cc209 const int num_ps = GPR_ARRAY_SIZE(pollsets); in pollset_set_test_basic() local
213 init_test_pollsets(pollsets, num_ps); in pollset_set_test_basic()
251 for (int i = 0; i < num_ps; i++) { in pollset_set_test_basic()
282 cleanup_test_pollsets(pollsets, num_ps); in pollset_set_test_basic()
311 const int num_ps = 1; in pollset_set_test_dup_fds() local
315 init_test_pollsets(&pollset, num_ps); in pollset_set_test_dup_fds()
354 cleanup_test_pollsets(&pollset, num_ps); in pollset_set_test_dup_fds()
381 const int num_ps = GPR_ARRAY_SIZE(pollsets); in pollset_set_test_empty_pollset() local
385 init_test_pollsets(pollsets, num_ps); in pollset_set_test_empty_pollset()
418 cleanup_test_pollsets(pollsets, num_ps); in pollset_set_test_empty_pollset()
/external/tensorflow/tensorflow/python/training/
Dsync_replicas_optimizer_test.py96 num_ps = 2
97 workers, _ = create_local_cluster(num_workers=num_workers, num_ps=num_ps)
187 num_ps = 2
188 workers, _ = create_local_cluster(num_workers=num_workers, num_ps=num_ps)
Dlocalhost_cluster_performance_test.py40 workers, _ = test.create_local_cluster(num_workers=2, num_ps=2)
67 test.create_local_cluster(num_workers=1, num_ps=10)
83 workers, _ = test.create_local_cluster(num_workers=1, num_ps=100)
/external/tensorflow/tensorflow/python/distribute/
Ddistribute_coordinator_test.py169 NUM_WORKERS, num_ps=NUM_PS)
196 num_ps=0, argument
209 if num_ps:
211 "localhost:%s" % portpicker.pick_unused_port() for _ in range(num_ps)
592 num_workers=NUM_WORKERS, num_ps=NUM_PS)
607 num_workers=NUM_WORKERS, num_ps=NUM_PS)
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.test.pbtxt25 …argspec: "args=[\'num_workers\', \'num_ps\', \'protocol\', \'worker_config\', \'ps_config\'], vara…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.test.pbtxt37 …argspec: "args=[\'num_workers\', \'num_ps\', \'protocol\', \'worker_config\', \'ps_config\'], vara…
/external/tensorflow/tensorflow/python/kernel_tests/
Dlist_ops_test.py796 worker = test_util.create_local_cluster(num_workers=1, num_ps=1)[0][0]
810 worker = test_util.create_local_cluster(num_workers=1, num_ps=1)[0][0]
825 worker = test_util.create_local_cluster(num_workers=1, num_ps=1)[0][0]
842 worker = test_util.create_local_cluster(num_workers=1, num_ps=1)[0][0]
Dpy_func_test.py494 workers, _ = test_util.create_local_cluster(num_workers=1, num_ps=0)
Dtensor_array_ops_test.py74 cls._workers, _ = test.create_local_cluster(num_workers=3, num_ps=0)
/external/tensorflow/tensorflow/python/framework/
Dtest_util.py2636 num_ps, argument
2680 ps_ports = [portpicker.pick_unused_port() for _ in range(num_ps)]
2703 start=True) for ix in range(num_ps)
/external/mesa3d/src/gallium/state_trackers/nine/
Ddevice9.h110 unsigned num_ps; member
Dnine_ff.c1852 device->ff.num_ps++; in nine_ff_get_ps()
2153 if (device->ff.num_ps > 100) { in nine_ff_prune_ps()
2158 device->ff.num_ps = 0; in nine_ff_prune_ps()