tf.compat.v1.tpu.rewrite

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

tf.compat.v1.tpu.rewrite(
    computation, inputs=None, infeed_queue=None, device_assignment=None, name=None,
    xla_options=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`	(Устарело) Не делает ничего.
`xla_options`	Экземпляр `tpu.XLAOptions`, который указывает на параметры, передаваемые компилятору XLA. Используйте `None` для использования стандартных параметров.

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

Возвращаемое значение

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

1) Отсутствие выхода: будет возвращён 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/r2.4/api_docs/python/tf/compat/v1/tpu/rewrite