tf.compat.v1.tpu.rewrite

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

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

Аргументы
`computation`	Функция Python, которая создаёт вычисление для применения к входным данным. Если функция принимает n входных данных, 'inputs' должна быть списком из n тензоров. `computation` может возвращать список операций и тензоров. Тензоры должны предшествовать операциям в возвращаемом списке. Значение возвращаемое `rewrite` — это список тензоров, соответствующих тензорам из результата `computation`. Все `Operation` , созданные во время `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) Отсутствует выходной параметр: возвращается NoOp, который управляется computation. 2) Выходной параметр — единственное значение: возвращается кортеж, содержащий значение. 3) Только выходные параметры — операции: возвращается NoOp, который управляется computation.

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

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

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

© 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.3/api_docs/python/tf/compat/v1/tpu/rewrite