Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/training/
Ddevice_setter.py78 def __init__(self, ps_tasks, ps_device, worker_device, merge_devices, ps_ops, argument
95 self._worker_device = worker_device
131 worker_device = pydev.DeviceSpec.from_string(self._worker_device or "")
132 worker_device = worker_device.make_merged_spec(current_device)
133 return worker_device.to_string()
139 worker_device="/job:worker", argument
229 chooser = _ReplicaDeviceChooser(ps_tasks, ps_device, worker_device,
Ddevice_setter_test.py157 worker_device="/job:sun",
178 worker_device="/job:sun",
/external/tensorflow/tensorflow/python/distribute/v1/
Dcross_device_ops_test.py545 worker_device = None
549 worker_device = "/job:%s/task:%d" % (task_type, task_id)
570 ops.device(worker_device), \
683 worker_device = None
687 worker_device = "/job:%s/task:%d" % (task_type, task_id)
689 ops.device(worker_device), \
713 devices_cpu = [(worker_device or "") + "/device:CPU:0"] * (
/external/tensorflow/tensorflow/python/distribute/
Dparameter_server_strategy_test.py124 worker_device = '/job:%s/replica:0/task:%d' % (task_type, task_id)
145 self.assertEqual(a.device, worker_device + '/' + last_part_device)
146 self.assertEqual(b.device, worker_device + '/' + last_part_device)
147 self.assertEqual(c.device, worker_device + '/' + last_part_device)
187 self.assertEqual(f.device, worker_device + '/' + last_part_device)
192 self.assertEqual(g.device, worker_device + '/device:CPU:1')
Dparameter_server_strategy.py253 worker_device=self._worker_device,
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.train.pbtxt429 …argspec: "args=[\'ps_tasks\', \'ps_device\', \'worker_device\', \'merge_devices\', \'cluster\', \'…