An op that demultiplexes a tensor to be sharded by XLA to a list of partitioned
tf.raw_ops.TPUPartitionedOutputV2(
inputs, num_splits, partition_dims, name=None
)
outputs outside the XLA computation. Supports ND sharding.
Returns | |
---|---|
A list of num_splits Tensor objects with the same type as inputs .
|