| /external/tensorflow/tensorflow/python/training/ |
| D | device_setter.py | 75 def __init__(self, ps_tasks, ps_device, worker_device, merge_devices, ps_ops, argument
92 self._worker_device = worker_device
128 worker_device = pydev.DeviceSpec.from_string(self._worker_device or "")
129 worker_device.merge_from(current_device)
130 return worker_device.to_string()
135 worker_device="/job:worker", merge_devices=True, argument
222 ps_tasks, ps_device, worker_device, merge_devices, ps_ops, ps_strategy)
|
| D | device_setter_test.py | 157 worker_device="/job:sun",
178 worker_device="/job:sun",
|
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | model_average_optimizer_test.py | 69 worker_device = "/job:worker/task:%d/cpu:0" % (worker_id)
71 worker_device=worker_device)
75 worker_device=worker_device,
180 worker_device = "/job:worker/task:0"
182 worker_device=worker_device)
186 worker_device=worker_device,
|
| D | elastic_average_optimizer_test.py | 76 worker_device = "/job:worker/task:%d/cpu:0" % (worker_id)
77 ea_custom = ElasticAverageCustomGetter(worker_device=worker_device)
81 worker_device=worker_device,
94 worker_device=worker_device,
280 ea_custom = ElasticAverageCustomGetter(worker_device="/job:worker/task:0")
284 worker_device="/job:worker/task:0",
|
| D | agn_optimizer_test.py | 72 worker_device = "/job:worker/task:%d/cpu:0" % (worker_id)
74 worker_device=worker_device,
77 agn_getter = agn_optimizer.AGNCustomGetter(worker_device=worker_device)
265 worker_device="/job:worker/task:0")
268 worker_device="/job:worker/task:0",
|
| D | model_average_optimizer.py | 64 def __init__(self, worker_device): argument
70 self._worker_device = worker_device
|
| D | agn_optimizer.py | 46 def __init__(self, worker_device): argument
51 self._worker_device = worker_device
|
| D | elastic_average_optimizer.py | 85 def __init__(self, worker_device): argument
91 self._worker_device = worker_device
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | parameter_server_strategy.py | 149 worker_device = "/job:%s/task:%d" % (task_type, task_id)
150 self._input_host_device = numpy_dataset.SingleDevice(worker_device)
157 "%s/device:GPU:%d" % (worker_device, i) for i in range(num_gpus))
159 compute_devices = (worker_device,)
163 self._device_map, [(worker_device, compute_devices)])
178 worker_device=worker_device,
190 self._default_device = worker_device
208 worker_device = device_util.canonicalize("/device:CPU:0")
209 self._input_host_device = numpy_dataset.SingleDevice(worker_device)
231 self._device_map, [(worker_device, compute_devices)])
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/ops/ |
| D | batch_ops_utils.py | 104 def run_handler_scheduled_ops(per_handler_ops, stamp, worker_device): argument
120 _move_tensors(result, worker_device) for result in results)
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/ |
| D | gbdt_batch.py | 647 worker_device = input_deps[0].device
728 with ops.device(self._get_replica_device_setter(worker_device)):
862 handler_reads, ensemble_stamp, worker_device)
931 per_handler_updates, ensemble_stamp, worker_device)
1279 def _get_replica_device_setter(self, worker_device): argument
1290 worker_device=worker_device,
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | parameter_server_strategy_test.py | 152 worker_device = '/job:%s/replica:0/task:%d' % (task_type, task_id)
174 self.assertEqual(a.device, worker_device + '/' + last_part_device)
175 self.assertEqual(b.device, worker_device + '/' + last_part_device)
176 self.assertEqual(c.device, worker_device + '/' + last_part_device)
216 self.assertEqual(f.device, worker_device + '/' + last_part_device)
221 self.assertEqual(g.device, worker_device + '/device:CPU:1')
|
| D | cross_device_ops_test.py | 542 worker_device = None
546 worker_device = "/job:%s/task:%d" % (task_type, task_id)
569 ops.device(worker_device), \
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | estimator.py | 226 worker_device = '/job:%s/task:%d' % (config.task_type, config.task_id)
228 worker_device = '/job:worker'
233 worker_device=worker_device,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.train.pbtxt | 421 …argspec: "args=[\'ps_tasks\', \'ps_device\', \'worker_device\', \'merge_devices\', \'cluster\', \'…
|