tf.raw_ops.TPUPartitionedCall

Вызывает функцию, размещенную на указанном устройстве TPU.

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

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

tf.raw_ops.TPUPartitionedCall(
    args, device_ordinal, Tout, f, autotuner_thresh=0, name=None
)

Аргументы
`args`	Список объектов `Tensor`. Аргументы функции.
`device_ordinal`	`Tensor` типа `int32`. Порядковый номер устройства TPU для выполнения функции.
`Tout`	Список `tf.DTypes`. Типы выходов функции.
`f`	Функция, декорированная @Defun. Вызываемая функция.
`autotuner_thresh`	Необязательный `int`. По умолчанию `0`.
`name`	Имя операции (необязательно).

Возвращаемые значения
Список объектов `Tensor` типа `Tout`.

© 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/raw_ops/TPUPartitionedCall