TPUPartitionedOutput

public final class TPUPartitionedOutput

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

outputs outside the XLA computation.

Nested Classes

class

TPUPartitionedOutput.Options

Optional attributes for TPUPartitionedOutput

Public Methods

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

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 TPUPartitionedOutput<T> create (Scope scope, Operand<T> inputs, Long numSplits, Options... options)

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

Parameters

scope	current scope
inputs	A tensor which represents the full shape of partitioned tensors.
options	carries optional attributes values

Returns

• a new instance of TPUPartitionedOutput

public Iterator<Operand<T>> iterator ()

public List<Output<T>> output ()

A list of partitioned inputs which must have the same shape.

public static TPUPartitionedOutput.Options partitionDim (Long partitionDim)

Parameters

partitionDim	An integer describles which dimension is partitioned.