Home
last modified time | relevance | path

Searched refs:logical_to_actual_devices (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dinput_lib.py59 device_map.logical_to_actual_devices(logical_device)),)
65 device_map.logical_to_actual_devices(logical_device)), (
68 device_map.logical_to_actual_devices(logical_device)))
Dcross_device_ops.py157 return destinations.device_map.logical_to_actual_devices(
191 devices = device_map.logical_to_actual_devices(logical_device)
984 devices = device_map.logical_to_actual_devices(logical_device)
Dmirrored_strategy.py248 devices = device_map.logical_to_actual_devices(logical_device)
787 return self._device_map.logical_to_actual_devices(0)
878 ops.device(self.device_map.logical_to_actual_devices(0)[
Dvalues.py75 def logical_to_actual_devices(self, logical_device_id): member in DeviceMap
134 def logical_to_actual_devices(self, logical_device_id): member in SingleDeviceMap
195 def logical_to_actual_devices(self, logical_device_id): member in ReplicaDeviceMap
265 return self._device_map.logical_to_actual_devices(self._logical_device)
818 return self._device_map.logical_to_actual_devices(self._logical_device)
Dtpu_strategy.py102 devices = device_map.logical_to_actual_devices(logical_device)