Calls a function placed on a specified TPU device.
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.
|
