Searched refs:shard_index_in_host (Results 1 – 1 of 1) sorted by relevance
568 def _tpu_ordinal_function(shard_index_in_host): argument582 0)[shard_index_in_host]585 return shard_index_in_host % self.num_of_cores_per_host