Пример #1
0
 def device(self):
   distribute_lib.require_tower_context(self)
   return self._distribution_strategy.worker_devices[self._tower_id]
Пример #2
0
 def device(self):
   distribute_lib.require_tower_context(self)
   return self._distribution_strategy.worker_devices[self._tower_id]