def devices(self):
     distribute_lib.require_replica_context(self)
     return [self._distribution_strategy.worker_devices[self._replica_id]]
Ejemplo n.º 2
0
 def devices(self):
     distribute_lib.require_replica_context(self)
     replica_id = tensor_util.constant_value(self._replica_id_in_sync_group)
     return [self._distribution_strategy.worker_devices[replica_id]]
Ejemplo n.º 3
0
 def devices(self):
   distribute_lib.require_replica_context(self)
   replica_id = tensor_util.constant_value(self._replica_id_in_sync_group)
   return [self._distribution_strategy.extended.worker_devices[replica_id]]
Ejemplo n.º 4
0
 def devices(self):
   distribute_lib.require_replica_context(self)
   ds = self._distribution_strategy
   return [ds.worker_devices[self._replica_id_in_sync_group]]
 def devices(self):
   distribute_lib.require_replica_context(self)
   return [self._distribution_strategy.worker_devices[self._replica_id]]