Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/distribute/
Dcross_device_ops.py158 destinations.logical_device)
167 return destinations.device_map, destinations.logical_device
190 device_map, logical_device = get_device_map_from(destinations)
191 devices = device_map.logical_to_actual_devices(logical_device)
201 return value_lib.Mirrored(device_map, value_updates, logical_device)
452 device_map, logical_device = get_device_map_from(destinations)
461 return [value_lib.Mirrored(device_map, v, logical_device) for v in index]
980 device_map, logical_device = get_device_map_from(destinations)
982 all_reduced.logical_device == logical_device):
984 devices = device_map.logical_to_actual_devices(logical_device)
[all …]
Dinput_lib.py44 def __init__(self, device_map, worker_device_pairs=None, logical_device=0): argument
55 self._logical_device = logical_device
59 device_map.logical_to_actual_devices(logical_device)),)
65 device_map.logical_to_actual_devices(logical_device)), (
67 (flattened, logical_device,
68 device_map.logical_to_actual_devices(logical_device)))
75 def logical_device(self): member in InputWorkers
Dvalues.py235 def __init__(self, device_map, values, logical_device=None): argument
239 if logical_device is None:
240 logical_device = device_map.logical_device_from_values(self._values)
241 self._logical_device = logical_device
268 def logical_device(self): member in DistributedValues
444 def __init__(self, strategy, device_map, values, logical_device=None): argument
447 device_map, values, logical_device=logical_device)
652 self, strategy, device_map, values, aggregation, logical_device=None): argument
654 strategy, device_map, values, logical_device=logical_device)
774 self, strategy, device_map, values, aggregation, logical_device=None): argument
[all …]
Dmirrored_strategy.py198 def _create_mirrored_variable(strategy, device_map, logical_device, # pylint: disable=missing-docs… argument
248 devices = device_map.logical_to_actual_devices(logical_device)
254 logical_device=logical_device)
258 logical_device=logical_device)
524 logical_device = 0 # TODO(josh11b): Get logical device from scope here.
530 logical_device = colocate_with.logical_device
563 self._container_strategy(), device_map, logical_device,
660 device_map=self._device_map, logical_device=0)
Dtpu_strategy.py71 strategy, device_map, logical_device, real_mirrored_creator, argument
102 devices = device_map.logical_to_actual_devices(logical_device)
106 logical_device=logical_device)
402 logical_device = 0 # TODO(josh11b): Get logical device from scope here.
408 logical_device = colocate_with.logical_device
438 self._container_strategy(), device_map, logical_device,
Dcollective_all_reduce_strategy.py241 logical_device = 0 # TODO(josh11b): Get logical device from scope here.
247 logical_device = colocate_with.logical_device
328 self._container_strategy(), device_map, logical_device,
Dparameter_server_strategy.py294 device_map=self._device_map, logical_device=0)