Searched refs:is_device_in_replica (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | values.py | 91 def is_device_in_replica(self, device, replica_id): member in DeviceMap
153 def is_device_in_replica(self, device, replica_id): member in SingleDeviceMap
217 def is_device_in_replica(self, device, replica_id): member in ReplicaDeviceMap
1370 assert device_map.is_device_in_replica(v0.device, 0), (
1375 assert device_map.is_device_in_replica(v.device, r + 1), (
|