def devices(self): distribute_lib.require_replica_context(self) return [self._distribution_strategy.worker_devices[self._replica_id]]
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]]
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]]
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]]