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

org.tensorflow.op.PrimitiveOp
final boolean
equals(Object obj)
final int
hashCode()
Operation
op()
Returns the underlying Operation
final String
toString()
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()
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.