tf.raw_ops.TPUPartitionedCall

Calls a function placed on a specified TPU device.

View aliases

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.raw_ops.TPUPartitionedCall

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

Args
args	A list of Tensor objects. The arguments to the function.
device_ordinal	A Tensor of type int32. The TPU device ordinal to run the function on.
Tout	A list of tf.DTypes. The types of the outputs of the function.
f	A function decorated with @Defun. The function to call.
autotuner_thresh	An optional int. Defaults to 0.
name	A name for the operation (optional).

Returns
A list of Tensor objects of type Tout.