# tf.raw_ops.TPUPartitionedOutputV2

<br />

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](https://www.tensorflow.org/guide/migrate) for
more details.

[`tf.compat.v1.raw_ops.TPUPartitionedOutputV2`](https://www.tensorflow.org/api_docs/python/tf/raw_ops/TPUPartitionedOutputV2)

<br />

    tf.raw_ops.TPUPartitionedOutputV2(
        inputs, num_splits, partition_dims, name=None
    )

outputs outside the XLA computation. Supports ND sharding.

<br />

<br />

<br />

| ## Args ||
|---|---|
| `inputs` | A `Tensor`. A tensor which represents the full shape of partitioned tensors. |
| `num_splits` | An `int` that is `>= 1`. |
| `partition_dims` | A list of `ints`. A list of integers describing how each dimension is partitioned. Emptiness indicates the inputs are replicated. |
| `name` | A name for the operation (optional). |

<br />

<br />

<br />

<br />

| ## Returns ||
|---|---|
| A list of `num_splits` `Tensor` objects with the same type as `inputs`. ||

<br />