tf.raw_ops.TPUReplicatedInput

Connects N inputs to an N-way replicated TPU computation.

View aliases

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.raw_ops.TPUReplicatedInput

tf.raw_ops.TPUReplicatedInput(
    inputs, is_mirrored_variable=False, index=-1, is_packed=False, name=None
)

This operation holds a replicated input to a tpu.replicate() computation subgraph. Each replicated input has the same shape and type alongside the output.

For example:

%a = "tf.opA"()
%b = "tf.opB"()
%replicated_input = "tf.TPUReplicatedInput"(%a, %b)
%computation = "tf.Computation"(%replicated_input)

The above computation has a replicated input of two replicas.

Args
inputs	A list of at least 1 Tensor objects with the same type.
is_mirrored_variable	An optional bool. Defaults to False.
index	An optional int. Defaults to -1.
is_packed	An optional bool. Defaults to False.
name	A name for the operation (optional).

Returns
A Tensor. Has the same type as inputs.