tf.contrib.tpu.device_assignment

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

tf.contrib.tpu.device_assignment(
    topology, computation_shape=None, computation_stride=None, num_replicas=1
)

Попытка выбрать компактную сетку ядер для локализации.

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

computation_shape и computation_stride значения должны быть степенями двойки для оптимальной укладки.

Аргументы
`topology`	Объект `Topology`, описывающий топологию кластера TPU. Чтобы получить топологию TPU, оцените `Tensor`, возвращаемое `initialize_system` с помощью `Session.run`. Можно передать сериализованный `TopologyProto` или объект `Topology`. Примечание: вы должны оценить `Tensor` сначала; вы не можете передать неоценённый `Tensor` сюда.
`computation_shape`	Массив numpy типа int32 ранга 1 размером, равным рангу топологии, описывающий форму блока ядер вычисления. Если None, `computation_shape` - `[1] * topology_rank`.
`computation_stride`	Массив numpy типа int32 ранга 1 размером `topology_rank`, описывающий межъядерное расстояние ядер `computation_shape` в топологии TPU. Если None, `computation_stride` - `[1] * topology_rank`.
`num_replicas`	Количество реплик вычисления для запуска. Реплики будут укладываться в свободные области топологии.

Возвращает
Объект DeviceAssignment, описывающий отображение между логическими ядрами в каждой реплике вычисления и физическими ядрами в топологии TPU.

Исключения
`ValueError`	Если `topology` не является допустимым объектом `Topology`.
`ValueError`	Если `computation_shape` или `computation_stride` не являются массивами numpy типа int32 ранга 1 размером [3], где все значения положительны.
`ValueError`	Если реплики вычисления не помещаются в топологию 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/r1.15/api_docs/python/tf/contrib/tpu/device_assignment