Searched refs:ps_tasks (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/training/ |
| D | device_setter.py | 78 def __init__(self, ps_tasks, ps_device, worker_device, merge_devices, ps_ops, argument
93 self._ps_tasks = ps_tasks
137 def replica_device_setter(ps_tasks=0, argument
211 ps_tasks = len(cluster_spec[ps_job_name])
213 if ps_tasks == 0:
226 ps_strategy = _RoundRobinStrategy(ps_tasks)
229 chooser = _ReplicaDeviceChooser(ps_tasks, ps_device, worker_device,
|
| 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/python/distribute/coordinator/ |
| D | cluster_coordinator.py | 912 ps_tasks = _extract_failed_ps_instances(str(e))
914 for t in ps_tasks:
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | parameter_server_strategy.py | 252 ps_tasks=num_ps_replicas,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.train.pbtxt | 429 …argspec: "args=[\'ps_tasks\', \'ps_device\', \'worker_device\', \'merge_devices\', \'cluster\', \'…
|