tf.tpu.experimental.DeviceAssignment

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

Псевдонимы

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

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

tf.compat.v1.tpu.experimental.DeviceAssignment

tf.tpu.experimental.DeviceAssignment(
    topology, core_assignment
)

Предпочтительно использовать вспомогательную функцию 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, computation_shape=None, computation_stride=None, num_replicas=1
)

`coordinates`

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

coordinates(
    replica, logical_core
)

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

`host_device`

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

host_device(
    replica=0, logical_core=0, job=None
)

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

`lookup_replicas`

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

lookup_replicas(
    task_id, logical_core
)

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

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

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

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

`tpu_device`

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

tpu_device(
    replica=0, logical_core=0, job=None
)

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

`tpu_ordinal`

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

tpu_ordinal(
    replica=0, logical_core=0
)

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

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