Searched refs:compute_devices (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | central_storage_strategy.py | 55 def __init__(self, compute_devices=None, parameter_device=None): argument
58 compute_devices=compute_devices,
221 def __init__(self, compute_devices=None, parameter_device=None): argument
225 compute_devices=compute_devices,
|
| D | parameter_server_strategy.py | 168 compute_devices=None, argument
173 compute_devices=compute_devices,
182 compute_devices=None, argument
188 compute_devices, parameter_device, cluster_resolver=cluster_resolver)
231 compute_devices = tuple(
235 compute_devices = (self._worker_device,)
238 device_util.canonicalize(d) for d in compute_devices]
284 compute_devices, argument
291 if compute_devices is None:
300 compute_devices = device_util.local_devices_from_num_gpus(num_gpus)
[all …]
|
| D | mirrored_strategy.py | 458 (host_device, (host_device,) * len(compute_devices))
459 for host_device, compute_devices in self._input_workers_devices
|
| D | input_lib.py | 826 for i, (input_device, compute_devices) in enumerate(worker_device_pairs):
830 _SingleWorkerDatasetIteratorSpec(input_device, compute_devices,
|
| /external/OpenCL-CTS/test_conformance/printf/ |
| D | test_printf.cpp | 993 uint32_t compute_devices = 0; in InitCL() local
1005 …if( (err = clGetDeviceInfo(device, CL_DEVICE_MAX_COMPUTE_UNITS, config_size, &compute_devices, NUL… in InitCL()
1007 compute_devices = 1; in InitCL()
1016 log_info( "\tProcessing with %d devices\n", compute_devices ); in InitCL()
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.experimental.-central-storage-strategy.pbtxt | 21 …argspec: "args=[\'self\', \'compute_devices\', \'parameter_device\'], varargs=None, keywords=None,…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.experimental.-central-storage-strategy.pbtxt | 21 …argspec: "args=[\'self\', \'compute_devices\', \'parameter_device\'], varargs=None, keywords=None,…
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | importer.py | 243 …for new_op in graph._add_new_tf_operations(compute_devices=False): # pylint: disable=protected-ac…
|
| D | ops.py | 3676 def _add_new_tf_operations(self, compute_devices=True): argument
3696 self._create_op_from_tf_operation(c_op, compute_device=compute_devices)
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | keras_test.py | 112 compute_devices = ['cpu:0', 'gpu:0'] if (
115 compute_devices, parameter_device='cpu:0')
|