tf.raw_ops.TPUPartitionedOutputV2

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

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.raw_ops.TPUPartitionedOutputV2

outputs outside the XLA computation. Supports ND sharding.

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).

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