Searched refs:parameter_device (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | central_storage_strategy.py | 55 def __init__(self, compute_devices=None, parameter_device=None): argument
59 parameter_device=parameter_device)
221 def __init__(self, compute_devices=None, parameter_device=None): argument
226 parameter_device=parameter_device))
|
| D | parameter_server_strategy.py | 169 parameter_device=None): argument
174 parameter_device=parameter_device)
183 parameter_device=None): argument
188 compute_devices, parameter_device, cluster_resolver=cluster_resolver)
285 parameter_device, argument
304 if parameter_device is None:
308 parameter_device = compute_devices[0]
310 parameter_device = _LOCAL_CPU
312 self._variable_device = parameter_device
314 self._parameter_devices = (parameter_device,)
|
| /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/keras/mixed_precision/ |
| D | keras_test.py | 115 compute_devices, parameter_device='cpu:0')
|