PartitionedOutput

public final class PartitionedOutput

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

outputs outside the XLA computation.

class PartitionedOutput.Options Optional attributes for PartitionedOutput

String OP_NAME The name of this op, as known by TensorFlow core engine

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

From class org.tensorflow.op.RawOp

final boolean	equals(Object obj)
final int	hashCode()
Operation	op() Return this unit of computation as a single `Operation`.
final String	toString()

From class java.lang.Object

From interface org.tensorflow.op.Op

abstract ExecutionEnvironment	env() Return the execution environment this op was created in.
abstract Operation	op() Return this unit of computation as a single `Operation`.

From interface java.lang.Iterable

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

Constants

The name of this op, as known by TensorFlow core engine

Constant Value: "TPUPartitionedOutput"

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

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

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

partitionDim	An integer describles which dimension is partitioned.