def device(self): distribute_lib.require_tower_context(self) return self._distribution_strategy.worker_devices[self._tower_id]
def device(self): distribute_lib.require_tower_context(self) return self._distribution_strategy.worker_devices[self._tower_id]