TPUPartitionedOutputV2

public final class TPUPartitionedOutputV2

An op that demultiplexes a tensor to be sharded by XLA to a list of partitioned

outputs outside the XLA computation. Supports ND sharding.

Public Methods

static <T> TPUPartitionedOutputV2<T>	create(Scope scope, Operand<T> inputs, Long numSplits, List<Long> partitionDims) Factory method to create a class wrapping a new TPUPartitionedOutputV2 operation.
Iterator<Operand<T>>	iterator()
List<Output<T>>	output() A list of partitioned outputs which have the same shape.

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 TPUPartitionedOutputV2<T> create (Scope scope, Operand<T> inputs, Long numSplits, List<Long> partitionDims)

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

Parameters

scope	current scope
inputs	A tensor which represents the full shape of partitioned tensors.
partitionDims	A list of integers describing how each dimension is partitioned. Emptiness indicates the inputs are replicated.

Returns

• a new instance of TPUPartitionedOutputV2

public Iterator<Operand<T>> iterator ()

public List<Output<T>> output ()

A list of partitioned outputs which have the same shape.