tf.raw_ops.StatefulPartitionedCall | TensorFlow v2.16.1 (original) (raw)

tf.raw_ops.StatefulPartitionedCall

Stay organized with collections Save and categorize content based on your preferences.

returns f(inputs), where f's body is placed and partitioned.

Compat aliases for migration

tf.compat.v1.raw_ops.StatefulPartitionedCall

tf.raw_ops.StatefulPartitionedCall(
    args,
    Tout,
    f,
    config='',
    config_proto='',
    executor_type='',
    name=None
)

Args
args	A list of Tensor objects. A list of input tensors.
Tout	A list of tf.DTypes. A list of output types.
f	A function decorated with @Defun. A function that takes 'args', a list of tensors, and returns 'output', another list of tensors. Input and output types are specified by 'Tin' and 'Tout'. The function body of f will be placed and partitioned across devices, setting this op apart from the regular Call op. This op is stateful.
config	An optional string. Defaults to "".
config_proto	An optional string. Defaults to "".
executor_type	An optional string. Defaults to "".
name	A name for the operation (optional).

Returns
A list of Tensor objects of type Tout.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.

Last updated 2024-04-26 UTC.