tf.raw_ops.TPUReplicatedInput
Connects N inputs to an N-way replicated TPU computation.
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. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/raw_ops/TPUReplicatedInput