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`

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

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

`coordinates`

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

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

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

`host_device`

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

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

Возвращает устройство 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[str] = None
) -> str

Возвращает имя устройства 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/api_docs/python/tf/tpu/experimental/DeviceAssignment