tf.tpu.experimental.DeviceAssignment

Сопоставление логических ядер в вычислении с физической топологией TPU.

Просмотр псевдонимов

Псевдонимы для миграции

См. Руководство по миграции для получения дополнительных сведений.

tf.compat.v1.tpu.experimental.DeviceAssignment

tf.tpu.experimental.DeviceAssignment(
    topology: tf.tpu.experimental.Topology,
    core_assignment: np.ndarray
)

Предпочтительнее использовать вспомогательную функцию DeviceAssignment.build() для построения DeviceAssignment; это проще, хотя и менее гибко, чем построение DeviceAssignment напрямую.

Аргументы
`topology`	Объект `Topology`, описывающий физическую топологию TPU.
`core_assignment`	Сопоставление логического ядра с физическим, представленное массивом NumPy ранга 3. Подробнее см. описание свойства `core_assignment`.

Исключения
`ValueError`	Если `topology` не является объектом `Topology`.
`ValueError`	Если `core_assignment` не является массивом NumPy ранга 3.

Атрибуты
`core_assignment`	Сопоставление логического ядра с физическим.
`num_cores_per_replica`	Количество ядер на реплику.
`num_replicas`	Количество реплик вычисления.
`topology`	Объект `Topology`, описывающий топологию TPU.

Методы

`build`

Просмотреть исходный код

@staticmethod
build(
    topology: tf.tpu.experimental.Topology,
    computation_shape: Optional[np.ndarray] = None,
    computation_stride: Optional[np.ndarray] = None,
    num_replicas: int = 1
) -> 'DeviceAssignment'

`coordinates`

Просмотреть исходный код

coordinates(
    replica: int, logical_core: int
) -> Tuple

Возвращает физические координаты топологии ядра.

`host_device`

Просмотреть исходный код

host_device(
    replica: int = 0, logical_core: int = 0, job: Optional[Text] = None
) -> Text

Возвращает устройство CPU, подключенное к логическому ядру.

`lookup_replicas`

Просмотреть исходный код

lookup_replicas(
    task_id: int, logical_core: int
) -> List[int]

Найти идентификаторы реплик по номеру задачи и логическому ядру.

Аргументы
`task_id`	Номер задачи TensorFlow.
`logical_core`	Целое число, определяющее логическое ядро.

Возвращает
Отсортированный список реплик, подключенных к этой задаче и логическому ядру.

Исключения
`ValueError`	Если в задаче не существует реплики, содержащей логическое ядро.

`tpu_device`

Просмотреть исходный код

tpu_device(
    replica: int = 0, logical_core: int = 0, job: Optional[Text] = None
) -> Text

Возвращает имя устройства TPU, назначенного логическому ядру.

`tpu_ordinal`

Просмотреть исходный код

tpu_ordinal(
    replica: int = 0, logical_core: int = 0
) -> int

Возвращает порядковый номер устройства TPU, назначенного логическому ядру.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/tpu/experimental/DeviceAssignment