tf.raw_ops.TPUPartitionedOutput

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

outputs outside the XLA computation.

inputsA Tensor. A tensor which represents the full shape of partitioned tensors.
num_splitsAn int that is >= 1.
partition_dimAn optional int. Defaults to 0. An integer describles which dimension is partitioned.
nameA name for the operation (optional).

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