tf.tpu.experimental.Topology

Описывает набор устройств TPU.

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

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

tf.tpu.experimental.Topology(
    serialized=None, mesh_shape=None, device_coordinates=None
)

Представляет как форму физической сетки, так и отображение между устройствами TensorFlow TPU и координатами физической сетки.

Аргументы
`serialized`	Последовательность байтов, представляющая сериализованную `TopologyProto` или `None`. Если это не `None`, сериализованный протокол разбирается для определения топологии.
`mesh_shape`	Последовательность из 4 положительных целых чисел, или `None`. Если это не `None`, форма топологии TPU в количестве ядер. Игнорируется, если `serialized` не является `None`.
`device_coordinates`	Массив NumPy ранга 3, описывающий отображение от устройств TensorFlow TPU к координатам TPU-матрицы, или `None`. Если указан, массив имеет ранг 3 и тип int32, с формой `[tasks, devices, axis]`. `tasks` - количество задач в кластере TPU, `devices` - количество устройств TPU на задачу, а `axis` - количество осей в топологии кластера TPU. Каждая запись указывает `axis`-ю координату в топологии пары задача/устройство. Топологии TPU являются 4-мерными, с измерениями `(x, y, z, core number)`. Этот аргумент игнорируется, если `serialized is not`None`.

Возможные исключения
`ValueError`	Если `serialized` не описывает корректную топологию.
`ValueError`	Если `serialized` является `None` и `mesh_shape` не является последовательностью из 4 положительных целых чисел.
`ValueError`	Если `serialized` является `None` и `device_coordinates` не является массивом NumPy ранга 3 типа int32, описывающим корректное отображение координат.

Атрибуты
`device_coordinates`	Описывает отображение от устройств TPU к координатам топологии.
`mesh_rank`	Возвращает количество измерений в сетке.
`mesh_shape`	Массив int32 ранга 1, описывающий форму топологии TPU.
`missing_devices`	Массив индексов отсутствующих устройств.
`num_tasks`	Возвращает количество задач TensorFlow в фрагменте TPU.
`num_tpus_per_task`	Возвращает количество устройств TPU на задачу во фрагменте TPU.

Методы

cpu_device_name_at_coordinates(
    device_coordinates, job=None
)

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

serialized()

Возвращает сериализованную форму топологии.

task_ordinal_at_coordinates(
    device_coordinates
)

Возвращает номер задачи TensorFlow, подключённой к device_coordinates.

Аргументы
`device_coordinates`	Последовательность целых чисел, описывающая физические координаты устройства в ткани TPU.

Возвращаемое значение
Возвращает номер задачи TensorFlow, содержащей устройство TPU с указанными физическими координатами.

tpu_device_name_at_coordinates(
    device_coordinates, job=None
)

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

tpu_device_ordinal_at_coordinates(
    device_coordinates
)

Возвращает номер устройства TensorFlow по device_coordinates.

Аргументы
`device_coordinates`	Последовательность целых чисел, описывающая физические координаты устройства в ткани TPU.

Возвращаемое значение
Возвращает номер устройства TensorFlow внутри задачи, соответствующей присоединённому устройству с указанными физическими координатами.

© 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/Topology