Searched refs:num_of_cores_per_host (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_system_metadata.py | 65 def __new__(cls, num_cores, num_hosts, num_of_cores_per_host, topology, argument
68 cls).__new__(cls, num_cores, num_hosts, num_of_cores_per_host,
122 num_of_cores_per_host = 0
130 num_of_cores_per_host = num_cores_per_host_set.pop()
154 num_of_cores_per_host=num_of_cores_per_host,
163 metadata.num_of_cores_per_host)
|
| D | tpu_embedding.py | 1338 self._num_cores_per_host = tpu_system_metadata.num_of_cores_per_host
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.tpu.experimental.-t-p-u-system-metadata.pbtxt | 19 name: "num_of_cores_per_host"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.tpu.experimental.-t-p-u-system-metadata.pbtxt | 19 name: "num_of_cores_per_host"
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | tpu_strategy.py | 1346 return self._tpu_metadata.num_of_cores_per_host
1353 max_models_per_host = (self._tpu_metadata.num_of_cores_per_host //
|