Searched refs:num_of_replicas_per_host (Results 1 – 2 of 2) sorted by relevance
93 * self._internal_ctx.num_of_replicas_per_host)100 replicas_consumed = self._internal_ctx.num_of_replicas_per_host127 def num_of_replicas_per_host(self): member in TPUContext132 return self._internal_ctx.num_of_replicas_per_host353 def num_of_replicas_per_host(self): member in _InternalTPUContext694 num_of_replicas_per_host = self.num_of_replicas_per_host695 host_id = replica_id / num_of_replicas_per_host696 ordinal_id = replica_id % num_of_replicas_per_host
815 num_of_replicas_per_host = ctx.num_of_replicas_per_host832 infeed_queue.set_number_of_shards(num_of_replicas_per_host)870 num_invocations_per_step=ctx.num_of_replicas_per_host)880 num_replicas_per_host = ctx.num_of_replicas_per_host971 num_replicas_per_host = ctx.num_of_replicas_per_host991 for _ in xrange(ctx.num_of_replicas_per_host):