tf.tpu.rewrite

Переписывает computation для выполнения на системе TPU.

Просмотр псевдонимов

Основные псевдонимы

`tf.contrib.tpu.rewrite`

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

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

tf.compat.v1.tpu.rewrite

tf.tpu.rewrite(
    computation, inputs=None, infeed_queue=None, device_assignment=None, name=None
)

Аргументы
`computation`	Функция Python, которая строит вычисление для применения к входным данным. Если функция принимает n входных данных, 'inputs' должно быть списком из n тензоров. `computation` может возвращать список операций и тензоров. Тензоры должны предшествовать операциям в возвращаемом списке. Значение, возвращаемое `rewrite`, представляет собой список тензоров, соответствующих тензорам из выходных данных `computation`. Все `Operation`s, созданные во время `computation` , будут выполнены при оценке любого из возвращаемых выходных тензоров, а не только тех, которые возвращаются.
`inputs`	Список входных тензоров или `None` (эквивалентно пустому списку). Каждый вход может быть вложенной структурой, содержащей значения, преобразуемые в тензоры. Обратите внимание, что передача списка N-измерений совместимых значений приведет к списку N-измерений скалярных тензоров, а не к одному тензору ранга N. Если вам требуется другое поведение, преобразуйте часть входов в тензоры с помощью `tf.convert_to_tensor`.
`infeed_queue`	Если не `None`, то `InfeedQueue` , из которого добавить кортеж аргументов в качестве входных данных в `computation`.
`device_assignment`	Если не `None`, то `DeviceAssignment` описывающий отображение между логическими ядрами в вычислении и физическими ядрами в топологии TPU. Может быть опущено для вычисления с одним ядром, в этом случае используется ядро, подключенное к задаче 0, устройству TPU 0.
`name`	(Устарело) Ничего не делает.

Возвращает
Такая же структура данных, как если бы вызов computation(*inputs) был выполнен напрямую, с некоторыми исключениями для обеспечения корректности. Исключение включает: 1) Выход None: возвращается NoOp, который контролирует зависимость от вычисления. 2) Выход с единственным значением: возвращается кортеж, содержащий значение. 3) Выходы только с операциями: возвращается NoOp, который контролирует зависимость от вычисления.

Возвращает

Такая же структура данных, как если бы вызов computation(*inputs) был выполнен напрямую, с некоторыми исключениями для обеспечения корректности. Исключение включает:

1) Выход None: возвращается NoOp, который контролирует зависимость от вычисления. 2) Выход с единственным значением: возвращается кортеж, содержащий значение. 3) Выходы только с операциями: возвращается NoOp, который контролирует зависимость от вычисления.

© 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/tpu/rewrite