Searched refs:compute_devices (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | parameter_server_strategy.py | 156 compute_devices = tuple(
159 compute_devices = (worker_device,)
161 self._device_map = values.ReplicaDeviceMap(compute_devices)
163 self._device_map, [(worker_device, compute_devices)])
225 compute_devices = tuple(map("/device:GPU:{}".format, range(num_gpus)))
227 compute_devices = (_LOCAL_CPU,)
229 self._device_map = values.ReplicaDeviceMap(compute_devices)
231 self._device_map, [(worker_device, compute_devices)])
236 assert len(compute_devices) == 1
250 "variable_device = %r", compute_devices, self._variable_device)
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | importer.py | 236 …for new_op in graph._add_new_tf_operations(compute_devices=False): # pylint: disable=protected-ac…
|
| D | ops.py | 3593 def _add_new_tf_operations(self, compute_devices=True): argument
3611 self._create_op_from_tf_operation(c_op, compute_device=compute_devices)
|