tf.raw_ops.StatefulPartitionedCall
returns f(inputs)
, where f
's body is placed and partitioned.
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-01-23 UTC.
[null,null,["Last updated 2024-01-23 UTC."],[],[]]