Searched refs:num_replicas_per_host (Results 1 – 3 of 3) sorted by relevance
1541 num_replicas_per_host=None, argument1573 self._num_replicas_per_host = num_replicas_per_host
880 num_replicas_per_host = ctx.num_of_replicas_per_host885 for _ in range(num_replicas_per_host):971 num_replicas_per_host = ctx.num_of_replicas_per_host977 return replica_id % num_replicas_per_host
535 def num_replicas_per_host(self): member in TPUExtended