Searched refs:num_of_cores_per_host (Results 1 – 5 of 5) sorted by relevance
111 num_of_cores_per_host = 0119 num_of_cores_per_host = num_cores_per_host_set.pop()143 num_of_cores_per_host=num_of_cores_per_host,152 metadata.num_of_cores_per_host)
343 def num_of_cores_per_host(self): member in _InternalTPUContext345 return metadata.num_of_cores_per_host358 return self.num_of_cores_per_host542 host_id = replica_id / self.num_of_cores_per_host558 num_of_cores_per_host = self.num_of_cores_per_host559 host_id = i / num_of_cores_per_host560 ordinal_id = i % num_of_cores_per_host585 return shard_index_in_host % self.num_of_cores_per_host625 num_cores_per_host = self._get_tpu_system_metadata().num_of_cores_per_host723 num_of_cores_per_host=1,
352 self._num_cores_per_host = self._tpu_system_metadata.num_of_cores_per_host
739 num_cores_per_host = ctx.num_of_cores_per_host746 invocation_index=host_id * ctx.num_of_cores_per_host + core_ordinal)
537 return self._tpu_metadata.num_of_cores_per_host544 max_models_per_host = (self._tpu_metadata.num_of_cores_per_host //