Searched refs:ps_tasks (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/python/training/ |
| D | device_setter.py | 75 def __init__(self, ps_tasks, ps_device, worker_device, merge_devices, ps_ops, argument
90 self._ps_tasks = ps_tasks
134 def replica_device_setter(ps_tasks=0, ps_device="/job:ps", argument
203 ps_tasks = len(cluster_spec[ps_job_name])
205 if ps_tasks == 0:
218 ps_strategy = _RoundRobinStrategy(ps_tasks)
222 ps_tasks, ps_device, worker_device, merge_devices, ps_ops, ps_strategy)
|
| D | localhost_cluster_performance_test.py | 95 with ops.device(device_setter.replica_device_setter(ps_tasks=100)):
|
| D | device_setter_test.py | 93 device_setter.replica_device_setter(ps_tasks=1, ps_device="/cpu:0")):
|
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | elastic_average_optimizer_test.py | 83 ps_tasks=1)):
96 ps_tasks=num_ps)):
|
| D | model_average_optimizer_test.py | 77 ps_tasks=1)):
|
| D | agn_optimizer_test.py | 76 ps_tasks=1)
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/ |
| D | gbdt_batch.py | 1281 ps_tasks = self._num_ps_replicas
1288 ps_strategy = _OpRoundRobinStrategy(ps_ops, ps_tasks)
1291 ps_tasks=ps_tasks,
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | parameter_server_strategy.py | 177 ps_tasks=num_ps_replicas,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.train.pbtxt | 421 …argspec: "args=[\'ps_tasks\', \'ps_device\', \'worker_device\', \'merge_devices\', \'cluster\', \'…
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | estimator.py | 232 ps_tasks=config.num_ps_replicas,
|
| /external/tensorflow/tensorflow/contrib/framework/python/ops/ |
| D | variables_test.py | 457 with ops.device(device_setter.replica_device_setter(ps_tasks=2)):
|