TPUReplicatedOutput

public final class TPUReplicatedOutput

Connects N outputs from an N-way replicated TPU computation.

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

For example:

%computation = "tf.Computation"()
 %replicated_output:2 = "tf.TPUReplicatedOutput"(%computation)
 

The above computation has a replicated output of two replicas.

Public Methods

static <T> TPUReplicatedOutput<T>	create(Scope scope, Operand<T> input, Long numReplicas) Factory method to create a class wrapping a new TPUReplicatedOutput operation.
Iterator<Operand<T>>	iterator()
List<Output<T>>	outputs()

Inherited Methods

From class org.tensorflow.op.PrimitiveOp

final boolean	equals(Object obj)
final int	hashCode()
Operation	op() Returns the underlying Operation
final String	toString()

From class java.lang.Object

boolean	equals(Object arg0)
final Class<?>	getClass()
int	hashCode()
final void	notify()
final void	notifyAll()
String	toString()
final void	wait(long arg0, int arg1)
final void	wait(long arg0)
final void	wait()

From interface java.lang.Iterable

void	forEach(Consumer<? super T> arg0)
abstract Iterator<Operand<T>>	iterator()
Spliterator<Operand<T>>	spliterator()

Public Methods

public static TPUReplicatedOutput<T> create (Scope scope, Operand<T> input, Long numReplicas)

Factory method to create a class wrapping a new TPUReplicatedOutput operation.

Parameters

scope	current scope

Returns

• a new instance of TPUReplicatedOutput

public Iterator<Operand<T>> iterator ()

public List<Output<T>> outputs ()