tf.raw_ops.TPUPartitionedOutput

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

View aliases

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.raw_ops.TPUPartitionedOutput

tf.raw_ops.TPUPartitionedOutput(
    inputs, num_splits, partition_dim=0, name=None
)

outputs outside the XLA computation.

Args
inputs	A Tensor. A tensor which represents the full shape of partitioned tensors.
num_splits	An int that is >= 1.
partition_dim	An optional int. Defaults to 0. An integer describles which dimension is partitioned.
name	A name for the operation (optional).

Returns
A list of num_splits Tensor objects with the same type as inputs.