Searched refs:num_replicas_per_host (Results 1 – 5 of 5) sorted by relevance
143 self.num_replicas_per_host = None268 report.config.num_cores_per_host = tt_config.num_replicas_per_host355 tt_config.num_replicas_per_host))
1942 num_replicas_per_host=None, argument1981 self._tt_config.num_replicas_per_host = num_replicas_per_host1984 if self._tt_config.num_replicas_per_host is None:1985 self._tt_config.num_replicas_per_host = 81988 num_replicas // self._tt_config.num_replicas_per_host +1989 (num_replicas % self._tt_config.num_replicas_per_host > 0))2038 self._tt_config.num_replicas_per_host = 1
340 "num_replicas_per_worker").set(self.extended.num_replicas_per_host)646 "num_replicas_per_worker").set(self.extended.num_replicas_per_host)708 "num_replicas_per_worker").set(self.extended.num_replicas_per_host)1344 def num_replicas_per_host(self): member in TPUExtended
1228 self.assertEqual(strategy.extended.num_replicas_per_host, 1) # pylint: disable=protected-access
2788 num_replicas_per_host = strategy.extended.num_replicas_per_host2790 for replica_id in range(num_replicas_per_host):2791 ordered_replicas += replicas[replica_id::num_replicas_per_host]