Searched refs:ps_device (Results 1 – 3 of 3) 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
94 self._ps_device = ps_device
122 ps_device = pydev.DeviceSpec.from_string(self._ps_device)
124 current_job, ps_job = current_device.job, ps_device.job
126 ps_device = ps_device.replace(task=self._ps_strategy(op))
128 ps_device = ps_device.make_merged_spec(current_device)
129 return ps_device.to_string()
138 ps_device="/job:ps", argument
208 ps_job_name = pydev.DeviceSpec.from_string(ps_device).job
229 chooser = _ReplicaDeviceChooser(ps_tasks, ps_device, worker_device,
|
| D | device_setter_test.py | 93 device_setter.replica_device_setter(ps_tasks=1, ps_device="/cpu:0")):
156 ps_device="/job:moon",
177 ps_device="/job:moon/cpu:0",
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.train.pbtxt | 429 …argspec: "args=[\'ps_tasks\', \'ps_device\', \'worker_device\', \'merge_devices\', \'cluster\', \'…
|