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