Ejemplo n.º 1
0
 def device(self):
   distribute_lib.require_tower_context(self)
   return self._distribution_strategy.worker_devices[self._tower_id]
Ejemplo n.º 2
0
 def device(self):
   distribute_lib.require_tower_context(self)
   return self._distribution_strategy.worker_devices[self._tower_id]