Home
last modified time | relevance | path

Searched refs:compute_devices (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dcentral_storage_strategy.py55 def __init__(self, compute_devices=None, parameter_device=None): argument
58 compute_devices=compute_devices,
253 def __init__(self, compute_devices=None, parameter_device=None): argument
257 compute_devices=compute_devices,
Dparameter_server_strategy.py149 compute_devices=None, argument
154 compute_devices=compute_devices,
163 compute_devices=None, argument
169 compute_devices, parameter_device, cluster_resolver=cluster_resolver)
212 compute_devices = tuple(
215 compute_devices = (worker_device,)
218 device_util.canonicalize(d) for d in compute_devices]
220 [(worker_device, compute_devices)])
266 compute_devices, argument
273 if compute_devices is None:
[all …]
/external/OpenCL-CTS/test_conformance/printf/
Dtest_printf.cpp982 uint32_t compute_devices = 0; in InitCL() local
989 …if( (err = clGetDeviceInfo(device, CL_DEVICE_MAX_COMPUTE_UNITS, config_size, &compute_devices, NUL… in InitCL()
991 compute_devices = 1; in InitCL()
1000 log_info( "\tProcessing with %d devices\n", compute_devices ); in InitCL()
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.distribute.experimental.-central-storage-strategy.pbtxt16 …argspec: "args=[\'self\', \'compute_devices\', \'parameter_device\'], varargs=None, keywords=None,…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.distribute.experimental.-central-storage-strategy.pbtxt17 …argspec: "args=[\'self\', \'compute_devices\', \'parameter_device\'], varargs=None, keywords=None,…
/external/tensorflow/tensorflow/python/framework/
Dimporter.py243 …for new_op in graph._add_new_tf_operations(compute_devices=False): # pylint: disable=protected-ac…
Dops.py3453 def _add_new_tf_operations(self, compute_devices=True): argument
3471 self._create_op_from_tf_operation(c_op, compute_device=compute_devices)