A list of device names for CPU hosts. Returns: A list of device names for CPU hosts.
(self)
| 631 | |
| 632 | @property |
| 633 | def hosts(self): |
| 634 | """A list of device names for CPU hosts. |
| 635 | |
| 636 | Returns: |
| 637 | A list of device names for CPU hosts. |
| 638 | """ |
| 639 | return copy.copy(self._hosts) |
| 640 | |
| 641 | # TODO(shizhiw): change to num_tensor_cores_per_host to be more explicit and |
| 642 | # to be consistent with `tpu_embedding_configuration.proto`. |