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`	Сопоставление логических ядер с физическими, представленное в виде рангового 3D-массива numpy. Подробнее см. описание свойства `core_assignment`.

Вызывает исключения
`ValueError`	Если `topology` не является объектом `Topology`.
`ValueError`	Если `core_assignment` не является ранговым 3D-массивом numpy.

Атрибуты
`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`	Целое число, идентифицирующее логическое ядро.

Возвращает
Отсортированный список реплик, подключенных к этой задаче и 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