tf.raw_ops.TPUReplicatedInput

Подключает N входов к N-кратному реплицированному TPU вычислению.

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

Данная операция содержит реплицированный вход в tpu.replicate() вычислительный подграф. Каждый реплицированный вход имеет такую же форму и тип, что и выход.

Например:

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

Вышеприведенное вычисление имеет реплицированный вход из двух реплик.