Home
last modified time | relevance | path

Searched refs:ps_device (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/training/
Ddevice_setter.py75 def __init__(self, ps_tasks, ps_device, worker_device, merge_devices, ps_ops, argument
91 self._ps_device = ps_device
119 ps_device = pydev.DeviceSpec.from_string(self._ps_device)
121 current_job, ps_job = current_device.job, ps_device.job
123 ps_device.task = self._ps_strategy(op)
125 ps_device.merge_from(current_device)
126 return ps_device.to_string()
134 def replica_device_setter(ps_tasks=0, ps_device="/job:ps", argument
200 ps_job_name = pydev.DeviceSpec.from_string(ps_device).job
222 ps_tasks, ps_device, worker_device, merge_devices, ps_ops, ps_strategy)
Ddevice_setter_test.py93 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/contrib/opt/python/training/
Dagn_optimizer_test.py73 ps_device = device_setter.replica_device_setter(
75 ps_device="/job:ps/task:0/cpu:0",
79 "", custom_getter=agn_getter), ops.device(ps_device):
88 custom_getter=agn_getter), ops.device(ps_device):
269 ps_device="/job:ps")), \
Delastic_average_optimizer_test.py82 ps_device="/job:ps/task:0/cpu:0",
95 ps_device="/job:ps/task:0/cpu:0",
285 ps_device="/job:ps")), \
Dmodel_average_optimizer_test.py76 ps_device="/job:ps/task:0/cpu:0",
187 ps_device="/job:ps")), \
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.train.pbtxt421 …argspec: "args=[\'ps_tasks\', \'ps_device\', \'worker_device\', \'merge_devices\', \'cluster\', \'…