tf.raw_ops.PartitionedCall
возвращает f(inputs), тело f размещается и разбивается на разделы.
tf.raw_ops.PartitionedCall(
args,
Tout,
f,
config='',
config_proto='',
executor_type='',
name=None
)
Асинхронно выполняет функцию, потенциально на нескольких устройствах, но в рамках одного процесса. Ядро размещает и разбивает базовый граф заданной функции, а затем выполняет каждый из разделенных подграфов как функцию.
| Аргументы | |
|---|---|
args | Список объектов Tensor. Список входных тензоров. |
Tout | Список tf.DTypes. Список выходных типов. |
f | Функция, декорированная с помощью @Defun. Функция, которая принимает «args» — список тензоров — и возвращает «output» — другой список тензоров. Типы входных и выходных данных задаются параметрами «Tin» и «Tout». Тело функции f будет размещено и разбито на разделы на нескольких устройствах, что отличает эту операцию от обычной операции Call. |
config | Необязательный string. Значение по умолчанию: "". |
config_proto | Необязательный string. Значение по умолчанию: "". |
executor_type | Необязательный string. Значение по умолчанию: "". |
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/PartitionedCall