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
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 static TPUPartitionedOutput.Options partitionDim (Long partitionDim)
Parameters
partitionDim | An integer describles which dimension is partitioned. |
---|