Searched defs:device_for_replica (Results 1 – 1 of 1) sorted by relevance
142 def device_for_replica(self, replica_id): member in TPUContext674 def device_for_replica(self, replica_id): member in _InternalTPUContext